From fb2947ba1dc2d62d08e0d46347ab3e70e98d8f4f Mon Sep 17 00:00:00 2001
From: Shenghang Tsai <jackalcooper@gmail.com>
Date: Fri, 29 Oct 2021 16:04:10 +0800
Subject: [PATCH] rm NCCL_NVCC_GENCODE (#6651)

---
 cmake/caches/cn/fast/mlir-cuda-75.cmake | 1 -
 cmake/third_party/nccl.cmake            | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/cmake/caches/cn/fast/mlir-cuda-75.cmake b/cmake/caches/cn/fast/mlir-cuda-75.cmake
index 257b65d5de..0e3d453b91 100644
--- a/cmake/caches/cn/fast/mlir-cuda-75.cmake
+++ b/cmake/caches/cn/fast/mlir-cuda-75.cmake
@@ -4,7 +4,6 @@ set(BUILD_GIT_VERSION NO CACHE BOOL "")
 set(TREAT_WARNINGS_AS_ERRORS YES CACHE BOOL "")
 set(BUILD_HWLOC NO CACHE BOOL "")
 set(BUILD_TESTING OFF CACHE BOOL "")
-set(NCCL_NVCC_GENCODE "-gencode=arch=compute_75,code=sm_75" CACHE STRING "")
 set(WITH_MLIR YES CACHE BOOL "")
 set(THIRD_PARTY_MIRROR aliyun CACHE STRING "")
 set(PIP_INDEX_MIRROR "https://pypi.tuna.tsinghua.edu.cn/simple" CACHE STRING "")
diff --git a/cmake/third_party/nccl.cmake b/cmake/third_party/nccl.cmake
index f3622f1a03..bce192ef7b 100644
--- a/cmake/third_party/nccl.cmake
+++ b/cmake/third_party/nccl.cmake
@@ -52,7 +52,7 @@ else()
             UPDATE_COMMAND ""
             CONFIGURE_COMMAND ""
             BUILD_IN_SOURCE 1
-            BUILD_COMMAND make -j${PROC_NUM} src.build CUDA_HOME=${CUDATOOLKIT_BIN_ROOT} NVCC_GENCODE=${NCCL_NVCC_GENCODE}
+            BUILD_COMMAND make -j${PROC_NUM} src.build CUDA_HOME=${CUDATOOLKIT_BIN_ROOT}
             INSTALL_COMMAND make src.install PREFIX=${NCCL_INSTALL_DIR}
             BUILD_BYPRODUCTS ${NCCL_LIBRARIES}
         )
-- 
GitLab