fix build failed with NCCL (#5)

* fix build failed on nccl * using nccl hdrs
AlibabaPAI · Aug 13, 2024 · 1e499d9 · 1e499d9
1 parent 41baac7
commit 1e499d9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torch_xla/csrc/runtime/disc/BUILD b/torch_xla/csrc/runtime/disc/BUILD
@@ -26,6 +26,7 @@ ptxla_cc_library(
         "@disc_compiler//:disc_custom_op",
         "@disc_compiler//:headers",
         "@local_config_cuda//cuda:cuda_headers",
+        "@nccl_archive//:nccl_headers",
         "@torch//:libc10",
         "@torch//:libc10_cuda",
         "@torch//:libtorch_cuda",
@@ -58,6 +59,8 @@ ptxla_cc_library(
         ":disc_ral",
         ":disc_utils",
         "//torch_xla/csrc/runtime:tf_logging",
+        "//torch_xla/csrc/runtime:sys_util",
+        "//torch_xla/csrc/runtime:env_vars",
         "@llvm-project//mlir:FuncDialect",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:Parser",