diff options
Diffstat (limited to 'third_party/gpus/cuda/platform.bzl.tpl')
-rw-r--r-- | third_party/gpus/cuda/platform.bzl.tpl | 59 |
1 files changed, 0 insertions, 59 deletions
diff --git a/third_party/gpus/cuda/platform.bzl.tpl b/third_party/gpus/cuda/platform.bzl.tpl index 539ed58d2c..01ef24b94e 100644 --- a/third_party/gpus/cuda/platform.bzl.tpl +++ b/third_party/gpus/cuda/platform.bzl.tpl @@ -8,65 +8,6 @@ def cuda_sdk_version(): def cudnn_sdk_version(): return CUDNN_VERSION -def cuda_library_path(name, version = cuda_sdk_version()): - if PLATFORM == "Darwin": - if not version: - return "lib/lib{}.dylib".format(name) - else: - return "lib/lib{}.{}.dylib".format(name, version) - elif PLATFORM == "Windows": - if not version: - return "lib/{}.dll".format(name) - else: - return "lib/{}{}.dll".format(name, version) - else: - if not version: - return "lib64/lib{}.so".format(name) - else: - return "lib64/lib{}.so.{}".format(name, version) - -def cuda_static_library_path(name): - if PLATFORM == "Darwin": - return "lib/lib{}_static.a".format(name) - elif PLATFORM == "Windows": - return "lib/{}_static.lib".format(name) - else: - return "lib64/lib{}_static.a".format(name) - -def cudnn_library_path(version = cudnn_sdk_version()): - if PLATFORM == "Darwin": - if not version: - return "lib/libcudnn.dylib" - else: - return "lib/libcudnn.{}.dylib".format(version) - elif PLATFORM == "Windows": - if not version: - return "lib/cudnn.dll" - else: - return "lib/cudnn{}.dll".format(version) - else: - if not version: - return "lib64/libcudnn.so" - else: - return "lib64/libcudnn.so.{}".format(version) - -def cupti_library_path(version = cuda_sdk_version()): - if PLATFORM == "Darwin": - if not version: - return "extras/CUPTI/lib/libcupti.dylib" - else: - return "extras/CUPTI/lib/libcupti.{}.dylib".format(version) - elif PLATFORM == "Windows": - if not version: - return "extras/CUPTI/lib/cupti.dll" - else: - return "extras/CUPTI/lib/cupti{}.dll".format(version) - else: - if not version: - return "extras/CUPTI/lib64/libcupti.so" - else: - return "extras/CUPTI/lib64/libcupti.so.{}".format(version) - def readlink_command(): if PLATFORM == "Darwin": return "greadlink" |