We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5ae30cc commit be6282eCopy full SHA for be6282e
cpp/tests/CMakeLists.txt
@@ -512,13 +512,11 @@ if(BUILD_CUGRAPH_MG_TESTS)
512
# - find MPI - only enabled if MG tests are to be built
513
find_package(MPI REQUIRED)
514
515
- execute_process(
516
- COMMAND nvidia-smi -L
517
- COMMAND wc -l
518
- OUTPUT_VARIABLE GPU_COUNT)
519
-
520
- string(REGEX REPLACE "\n$" "" GPU_COUNT ${GPU_COUNT})
521
- MESSAGE(STATUS "GPU_COUNT: " ${GPU_COUNT})
+ # Set the GPU count to 1. If the caller wants to execute MG tests using
+ # more than 1, override from the command line using -DGPU_COUNT=<gpucount>
+ if (NOT DEFINED GPU_COUNT)
+ set(GPU_COUNT "1")
+ endif()
522
523
if(MPI_CXX_FOUND)
524
###########################################################################################
0 commit comments