diff --git a/cpp/tests/CMakeLists.txt b/cpp/tests/CMakeLists.txt index 9f96b93e7a..7411056fd3 100644 --- a/cpp/tests/CMakeLists.txt +++ b/cpp/tests/CMakeLists.txt @@ -187,6 +187,10 @@ if(BUILD_TESTS) linalg/ternary_op.cu linalg/transpose.cu linalg/unary_op.cu + GPUS + 1 + PERCENT + 50 ) ConfigureTest( @@ -213,7 +217,9 @@ if(BUILD_TESTS) EXPLICIT_INSTANTIATE_ONLY ) - ConfigureTest(NAME MATRIX_SELECT_TEST PATH matrix/select_k.cu LIB EXPLICIT_INSTANTIATE_ONLY) + ConfigureTest( + NAME MATRIX_SELECT_TEST PATH matrix/select_k.cu LIB EXPLICIT_INSTANTIATE_ONLY GPUS 1 PERCENT 50 + ) ConfigureTest( NAME MATRIX_SELECT_LARGE_TEST PATH matrix/select_large_k.cu LIB EXPLICIT_INSTANTIATE_ONLY