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}"
|
||
|
|
""",
|
||
|
|
)
|