From 1e499d939ef13ba3caf9453741e07ebf71e38000 Mon Sep 17 00:00:00 2001
From: Yan Xu <yancey.yx@alibaba-inc.com>
Date: Tue, 13 Aug 2024 10:45:40 +0800
Subject: [PATCH] fix build failed with NCCL (#5)

* fix build failed on nccl

* using nccl hdrs
---
 torch_xla/csrc/runtime/disc/BUILD | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/torch_xla/csrc/runtime/disc/BUILD b/torch_xla/csrc/runtime/disc/BUILD
index d6d25ced337..999aa85ea64 100755
--- 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",