diff options
Diffstat (limited to '0001-remove-nccl-check.patch')
-rw-r--r-- | 0001-remove-nccl-check.patch | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/0001-remove-nccl-check.patch b/0001-remove-nccl-check.patch new file mode 100644 index 0000000..5ad8cbc --- /dev/null +++ b/0001-remove-nccl-check.patch @@ -0,0 +1,25 @@ +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
|