|
|
|
@@ -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() |
|
|
|
|
|
|
|
|
Megvii Engine Team
3 years ago