fix(build): use official build script to build nccl

GitOrigin-RevId: 99e2951a6e48488e4841b497496b0b8147852b5d

fix(build): use official build script to build nccl
GitOrigin-RevId: 99e2951a6e48488e4841b497496b0b8147852b5d
b75658c8 · Megvii Engine Team · ec59a5bc · b75658c8
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

CMakeLists.txt CMakeLists.txt +1 -0

未找到文件。
--- 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()