From 3c61e934f2adcebab88a289fa06d25e012ea679d Mon Sep 17 00:00:00 2001 From: Nilesh M Negi Date: Mon, 23 Sep 2024 19:05:49 -0500 Subject: [PATCH] [BUILD] Enable MSCCL++ for gfx942 variants (#1344) Signed-off-by: nileshnegi --- CMakeLists.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 57dc43b115..4496c361d0 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -287,8 +287,9 @@ if (HAVE_KERNARG_PRELOAD) endif() ## Disable building MSCCL++ if the build environment is invalid +## Currently MSCCL++ is supported only on gfx942 if (ENABLE_MSCCLPP) - if(NOT "gfx942" IN_LIST GPU_TARGETS) + if(NOT ("gfx942" IN_LIST GPU_TARGETS OR "gfx942:xnack-" IN_LIST GPU_TARGETS OR "gfx942:xnack+" IN_LIST GPU_TARGETS)) set(ENABLE_MSCCLPP OFF) message(WARNING "Can only build MSCCL++ for gfx942; disabling MSCCL++ build") endif()