From 0f51b22f75a7250559c4b3ff48afbe18ec858a91 Mon Sep 17 00:00:00 2001
From: Icey <1790571317@qq.com>
Date: Mon, 14 Jul 2025 06:30:01 +0000
Subject: [PATCH] remove nccl check

---
 tools/build_pytorch_libs.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tools/build_pytorch_libs.py b/tools/build_pytorch_libs.py
index 5dd5a221975..ca315302fc3 100644
--- a/tools/build_pytorch_libs.py
+++ b/tools/build_pytorch_libs.py
@@ -119,7 +119,8 @@ def build_pytorch(
     cmake: CMake,
 ) -> None:
     my_env = _create_build_env()
-    checkout_nccl()
+    if os.getenv("USE_NCCL", "1") != "0":
+        checkout_nccl()
     build_test = not check_negative_env_flag("BUILD_TEST")
     cmake.generate(
         version, cmake_python_library, build_python, build_test, my_env, rerun_cmake
-- 
2.34.1