Skip to content

Commit b6377e0

Browse files
Changed working dir for the submodule command and extended it to the json repo (#1495)
This allows it to work when the sub repos don't exist.
1 parent f77308a commit b6377e0

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

cmake/MSCCLPP.cmake

+7-6
Original file line numberDiff line numberDiff line change
@@ -56,11 +56,12 @@ if(ENABLE_MSCCLPP)
5656
#if(NOT mscclpp_nccl_FOUND)
5757
# Ensure the source code is checked out
5858
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")
6162
execute_process(
6263
COMMAND git submodule update --init --recursive
63-
WORKING_DIRECTORY ${MSCCLPP_SOURCE}
64+
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
6465
)
6566
endif()
6667

@@ -69,7 +70,7 @@ if(ENABLE_MSCCLPP)
6970
WORKING_DIRECTORY ${MSCCLPP_SOURCE}
7071
)
7172

72-
execute_process(
73+
execute_process(
7374
COMMAND git apply ${CMAKE_CURRENT_SOURCE_DIR}/ext-src/read-allred.patch
7475
WORKING_DIRECTORY ${MSCCLPP_SOURCE}
7576
)
@@ -98,7 +99,7 @@ if(ENABLE_MSCCLPP)
9899
#GIT_REPOSITORY https://github.com/microsoft/mscclpp.git
99100
#GIT_TAG 4ee15b7ad085daaf74349d4c49c9b8480d28f0dc
100101
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}
102103
LOG_DOWNLOAD FALSE
103104
LOG_CONFIGURE FALSE
104105
LOG_BUILD FALSE
@@ -109,7 +110,7 @@ if(ENABLE_MSCCLPP)
109110

110111

111112
find_package(mscclpp_nccl REQUIRED)
112-
execute_process(
113+
execute_process(
113114
COMMAND git apply --reverse ${CMAKE_CURRENT_SOURCE_DIR}/ext-src/cpx.patch
114115
WORKING_DIRECTORY ${MSCCLPP_SOURCE}
115116
)

0 commit comments

Comments
 (0)