Increase memory requirements for failing tests.

rapidsai · Jan 31, 2025 · 7c7162c · 7c7162c
1 parent 605368d
commit 7c7162c
Showing 1 changed file with 13 additions and 2 deletions.
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(
@@ -211,12 +215,19 @@ if(BUILD_TESTS)
     sparse/spectral_matrix.cu
     LIB
     EXPLICIT_INSTANTIATE_ONLY
+    GPUS
+    1
+    PERCENT
+    50
   )
 
-  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
+    NAME MATRIX_SELECT_LARGE_TEST PATH matrix/select_large_k.cu LIB EXPLICIT_INSTANTIATE_ONLY GPUS
+    1 PERCENT 50
   )
 
   ConfigureTest(