Increase memory requirements for failing tests.

rapidsai · Jan 31, 2025 · d08d565 · d08d565
1 parent 605368d
commit d08d565
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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