1*b6fb3261SAndroid Build Coastguard Workerfilegroup( 2*b6fb3261SAndroid Build Coastguard Worker name = "LICENSE", 3*b6fb3261SAndroid Build Coastguard Worker visibility = ["//visibility:public"], 4*b6fb3261SAndroid Build Coastguard Worker) 5*b6fb3261SAndroid Build Coastguard Worker 6*b6fb3261SAndroid Build Coastguard Workercc_library( 7*b6fb3261SAndroid Build Coastguard Worker name = "nccl", 8*b6fb3261SAndroid Build Coastguard Worker srcs = ["libnccl.so.%{nccl_version}"], 9*b6fb3261SAndroid Build Coastguard Worker hdrs = ["nccl.h"], 10*b6fb3261SAndroid Build Coastguard Worker include_prefix = "third_party/nccl", 11*b6fb3261SAndroid Build Coastguard Worker visibility = ["//visibility:public"], 12*b6fb3261SAndroid Build Coastguard Worker deps = [ 13*b6fb3261SAndroid Build Coastguard Worker "@local_config_cuda//cuda:cuda_headers", 14*b6fb3261SAndroid Build Coastguard Worker ], 15*b6fb3261SAndroid Build Coastguard Worker) 16*b6fb3261SAndroid Build Coastguard Worker 17*b6fb3261SAndroid Build Coastguard Workergenrule( 18*b6fb3261SAndroid Build Coastguard Worker name = "nccl-files", 19*b6fb3261SAndroid Build Coastguard Worker outs = [ 20*b6fb3261SAndroid Build Coastguard Worker "libnccl.so.%{nccl_version}", 21*b6fb3261SAndroid Build Coastguard Worker "nccl.h", 22*b6fb3261SAndroid Build Coastguard Worker ], 23*b6fb3261SAndroid Build Coastguard Worker cmd = """ 24*b6fb3261SAndroid Build Coastguard Workercp "%{nccl_header_dir}/nccl.h" "$(@D)/nccl.h" && 25*b6fb3261SAndroid Build Coastguard Workercp "%{nccl_library_dir}/libnccl.so.%{nccl_version}" \ 26*b6fb3261SAndroid Build Coastguard Worker "$(@D)/libnccl.so.%{nccl_version}" 27*b6fb3261SAndroid Build Coastguard Worker""", 28*b6fb3261SAndroid Build Coastguard Worker) 29