From b75658c88fb86ca1ad4f355f0afdf38a5669080b Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 5 Nov 2021 19:27:58 +0800 Subject: [PATCH] fix(build): use official build script to build nccl GitOrigin-RevId: 99e2951a6e48488e4841b497496b0b8147852b5d --- CMakeLists.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/CMakeLists.txt b/CMakeLists.txt index 3bd7d1ee1..5464eda9c 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -1141,6 +1141,7 @@ if(MGE_WITH_DISTRIBUTED) set(MEGRAY_WITH_NCCL ${MGE_WITH_CUDA} CACHE BOOL "Override MegRay option" FORCE) set(MEGRAY_WITH_SHM ${MGE_WITH_CUDA} CACHE BOOL "Override MegRay option" FORCE) set(MEGRAY_WITH_RCCL ${MGE_WITH_ROCM} CACHE BOOL "Override MegRay option" FORCE) + set(MEGRAY_CUDA_GENCODE ${MGE_CUDA_GENCODE} CACHE STRING "Overwrite MegRay CUDA -gencode specifications" FORCE) add_subdirectory(${PROJECT_SOURCE_DIR}/third_party/MegRay) endif() -- GitLab