@@ -56,11 +56,12 @@ if(ENABLE_MSCCLPP)
56
56
#if(NOT mscclpp_nccl_FOUND)
57
57
# Ensure the source code is checked out
58
58
set (MSCCLPP_SOURCE ${CMAKE_CURRENT_SOURCE_DIR} /ext -src/mscclpp CACHE PATH "" )
59
- if (NOT EXISTS ${MSCCLPP_SOURCE} /CMakeLists.txt)
60
- message (STATUS "Checking out microsoft/mscclpp" )
59
+ set (JSON_SOURCE ${CMAKE_CURRENT_SOURCE_DIR} /ext -src/json CACHE PATH "" )
60
+ if ((NOT EXISTS ${MSCCLPP_SOURCE} /CMakeLists.txt) OR (NOT EXISTS ${JSON_SOURCE} /CMakeLists.txt))
61
+ message (STATUS "Checking out external code" )
61
62
execute_process (
62
63
COMMAND git submodule update --init --recursive
63
- WORKING_DIRECTORY ${MSCCLPP_SOURCE }
64
+ WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR }
64
65
)
65
66
endif ()
66
67
@@ -69,7 +70,7 @@ if(ENABLE_MSCCLPP)
69
70
WORKING_DIRECTORY ${MSCCLPP_SOURCE}
70
71
)
71
72
72
- execute_process (
73
+ execute_process (
73
74
COMMAND git apply ${CMAKE_CURRENT_SOURCE_DIR} /ext -src/read-allred.patch
74
75
WORKING_DIRECTORY ${MSCCLPP_SOURCE}
75
76
)
@@ -98,7 +99,7 @@ if(ENABLE_MSCCLPP)
98
99
#GIT_REPOSITORY https://github.com/microsoft/mscclpp.git
99
100
#GIT_TAG 4ee15b7ad085daaf74349d4c49c9b8480d28f0dc
100
101
INSTALL_DIR ${MSCCLPP_ROOT}
101
- CMAKE_ARGS -DAMDGPU_TARGETS=${GFX942_VARIANT} -DGPU_TARGETS=${GFX942_VARIANT} -DMSCCLPP_BYPASS_GPU_CHECK=ON -DMSCCLPP_USE_ROCM=ON -DCMAKE_BUILD_TYPE=${CMAKE_BUILD_TYPE} -DMSCCLPP_BUILD_APPS_NCCL=ON -DMSCCLPP_BUILD_PYTHON_BINDINGS=OFF -DMSCCLPP_BUILD_TESTS=OFF -DCMAKE_INSTALL_PREFIX=<INSTALL_DIR> "${CMAKE_PREFIX_PATH_ARG} " -DCMAKE_VERBOSE_MAKEFILE=1 "${CMAKE_INSTALL_RPATH_USE_LINK_PATH_ARG} " "${HIP_COMPILER_ARG} " -DFETCHCONTENT_SOURCE_DIR_JSON=${CMAKE_CURRENT_SOURCE_DIR} / ext -src/json
102
+ CMAKE_ARGS -DAMDGPU_TARGETS=${GFX942_VARIANT} -DGPU_TARGETS=${GFX942_VARIANT} -DMSCCLPP_BYPASS_GPU_CHECK=ON -DMSCCLPP_USE_ROCM=ON -DCMAKE_BUILD_TYPE=${CMAKE_BUILD_TYPE} -DMSCCLPP_BUILD_APPS_NCCL=ON -DMSCCLPP_BUILD_PYTHON_BINDINGS=OFF -DMSCCLPP_BUILD_TESTS=OFF -DCMAKE_INSTALL_PREFIX=<INSTALL_DIR> "${CMAKE_PREFIX_PATH_ARG} " -DCMAKE_VERBOSE_MAKEFILE=1 "${CMAKE_INSTALL_RPATH_USE_LINK_PATH_ARG} " "${HIP_COMPILER_ARG} " -DFETCHCONTENT_SOURCE_DIR_JSON=${JSON_SOURCE}
102
103
LOG_DOWNLOAD FALSE
103
104
LOG_CONFIGURE FALSE
104
105
LOG_BUILD FALSE
@@ -109,7 +110,7 @@ if(ENABLE_MSCCLPP)
109
110
110
111
111
112
find_package (mscclpp_nccl REQUIRED)
112
- execute_process (
113
+ execute_process (
113
114
COMMAND git apply --reverse ${CMAKE_CURRENT_SOURCE_DIR} /ext -src/cpx.patch
114
115
WORKING_DIRECTORY ${MSCCLPP_SOURCE}
115
116
)
0 commit comments