diff --git a/examples/CMakeLists.txt b/examples/CMakeLists.txt index 0bafb40e..827151cc 100644 --- a/examples/CMakeLists.txt +++ b/examples/CMakeLists.txt @@ -1,7 +1,9 @@ if(CMAKE_BUILD_TESTS) add_subdirectory(1_SimpleNet) add_subdirectory(2_ResNet18) - # add_subdirectory(3_MultiGPU) + if(ENABLE_CUDA AND ENABLE_MPI) + add_subdirectory(3_MultiGPU) + endif() add_subdirectory(4_MultiIO) # add_subdirectory(5_Looping) add_subdirectory(6_Autograd) diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt index b5ed4c87..c5ad5c9f 100644 --- a/src/CMakeLists.txt +++ b/src/CMakeLists.txt @@ -125,8 +125,10 @@ if(CMAKE_BUILD_TESTS) DESTINATION ${CMAKE_CURRENT_SOURCE_DIR}/test/examples) file(COPY ${CMAKE_CURRENT_SOURCE_DIR}/../examples/2_ResNet18 DESTINATION ${CMAKE_CURRENT_SOURCE_DIR}/test/examples) - # file(COPY ${CMAKE_CURRENT_SOURCE_DIR}/../examples/3_MultiGPU - # DESTINATION ${CMAKE_CURRENT_SOURCE_DIR}/test/examples) + if(ENABLE_CUDA AND ENABLE_MPI) + file(COPY ${CMAKE_CURRENT_SOURCE_DIR}/../examples/3_MultiGPU + DESTINATION ${CMAKE_CURRENT_SOURCE_DIR}/test/examples) + endif() file(COPY ${CMAKE_CURRENT_SOURCE_DIR}/../examples/4_MultiIO DESTINATION ${CMAKE_CURRENT_SOURCE_DIR}/test/examples) # file(COPY ${CMAKE_CURRENT_SOURCE_DIR}/../examples/5_Looping