8e9ca6617b
Use a single python script (third_party/gpus/find_cuda_config.py) from configure.py and the different *_configure.bzl scripts to find the different CUDA library and header paths based on a set of environment variables. PiperOrigin-RevId: 243669844
29 lines
601 B
Smarty
29 lines
601 B
Smarty
filegroup(
|
|
name = "LICENSE",
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "nccl",
|
|
srcs = ["libnccl.so.%{nccl_version}"],
|
|
hdrs = ["nccl.h"],
|
|
include_prefix = "third_party/nccl",
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
"@local_config_cuda//cuda:cuda_headers",
|
|
],
|
|
)
|
|
|
|
genrule(
|
|
name = "nccl-files",
|
|
outs = [
|
|
"libnccl.so.%{nccl_version}",
|
|
"nccl.h",
|
|
],
|
|
cmd = """
|
|
cp "%{nccl_header_dir}/nccl.h" "$(@D)/nccl.h" &&
|
|
cp "%{nccl_library_dir}/libnccl.so.%{nccl_version}" \
|
|
"$(@D)/libnccl.so.%{nccl_version}"
|
|
""",
|
|
)
|