Disabled MSCCL++ feature except when building on Ubuntu or CentOS host systems (#1505)
* Added condition for MSCCL++ to only build on an Ubuntu host system. * Added CentOS to the supported OS list
此提交包含在:
+6
-1
@@ -305,7 +305,7 @@ if (HAVE_PARALLEL_JOBS)
|
||||
endif()
|
||||
|
||||
## Disable building MSCCL++ if the build environment is invalid
|
||||
## Currently MSCCL++ is supported only on gfx942
|
||||
## Currently MSCCL++ is supported only on gfx942, and only on Ubuntu and CentOS
|
||||
if (ENABLE_MSCCLPP AND NOT ("gfx942" IN_LIST COMPILING_TARGETS OR "gfx942:xnack-" IN_LIST COMPILING_TARGETS OR "gfx942:xnack+" IN_LIST COMPILING_TARGETS))
|
||||
set(ENABLE_MSCCLPP OFF)
|
||||
message(WARNING "Can only build MSCCL++ for gfx942; disabling MSCCL++ build")
|
||||
@@ -314,6 +314,11 @@ if (ENABLE_MSCCLPP AND ROCM_VERSION VERSION_LESS "60200")
|
||||
set(ENABLE_MSCCLPP OFF)
|
||||
message(WARNING "MSCCL++ integration only supported on ROCm 6.2 or greater; disabling MSCCL++ build")
|
||||
endif()
|
||||
cmake_host_system_information(RESULT HOST_OS_ID QUERY DISTRIB_ID)
|
||||
if (ENABLE_MSCCLPP AND NOT(${HOST_OS_ID} STREQUAL "ubuntu" OR ${HOST_OS_ID} STREQUAL "centos"))
|
||||
set(ENABLE_MSCCLPP OFF)
|
||||
message(WARNING "MSCCL++ integration not supported on this OS (${HOST_OS_ID}); disabling MSCCL++ build")
|
||||
endif()
|
||||
|
||||
# Determine version from makefiles/version.mk and fill in templates
|
||||
#==================================================================================================
|
||||
|
||||
新增問題並參考
封鎖使用者