blob: 5ad8cbc731dbc646331f3526eaee21e60d4263ec (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
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
|
