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
此提交包含在:
corey-derochie-amd
2025-01-29 08:54:09 -07:00
提交者 GitHub
父節點 7ac82248de
當前提交 bd0f5cccbe
+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
#==================================================================================================