add more test cases for coverage

rapidsai · Apr 1, 2024 · ae70762 · ae70762
1 parent 5abbd4e
commit ae70762
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/cpp/test/matrix/select_k_csr.cu b/cpp/test/matrix/select_k_csr.cu
@@ -343,21 +343,25 @@ template <typename index_t>
 const std::vector<SelectKCsrInputs<index_t>> selectk_inputs = {
   {10, 32, 10, 0.0, true, false},
   {10, 32, 10, 0.0, true, true},
-  {10, 32, 10, 0.01, true, false},
+  {10, 32, 10, 0.01, true, false},  // kWarpImmediate
   {10, 32, 10, 0.1, true, true},
-  {10, 32, 251, 0.1, true, false},
+  {10, 32, 251, 0.1, true, false},  // kWarpImmediate
   {10, 32, 251, 0.6, true, true},
-  {1024, 1024, 258, 0.3, true, false},
+  {1000, 1024 * 100, 1, 0.1, true, false},  // kWarpImmediate
+  {1000, 1024 * 100, 1, 0.2, true, true},
+  {1024, 1024, 258, 0.3, true, false},  // kRadix11bitsExtraPass
   {1024, 1024, 600, 0.2, true, true},
-  {1024, 1024, 1024, 0.3, true, false},
+  {1024, 1024, 1024, 0.3, true, false},  // kRadix11bitsExtraPass
   {1024, 1024, 1024, 0.2, true, true},
-  {100, 1024 * 1000, 251, 0.1, true, false},
+  {100, 1024 * 1000, 251, 0.1, true, false},  // kWarpDistributedShm
   {100, 1024 * 1000, 251, 0.2, true, true},
-  {1024, 1024 * 10, 251, 0.3, true, false},
+  {1024, 1024 * 10, 251, 0.3, true, false},  // kWarpImmediate
   {1024, 1024 * 10, 251, 0.2, true, true},
-  {2048, 1024 * 10, 1000, 0.2, true, false},
+  {1000, 1024 * 20, 1000, 0.2, true, false},  // kRadix11bits
+  {1000, 1024 * 20, 1000, 0.3, true, true},
+  {2048, 1024 * 10, 1000, 0.2, true, false},  // kRadix11bitsExtraPass
   {2048, 1024 * 10, 1000, 0.3, true, true},
-  {2048, 1024 * 10, 2100, 0.1, true, false},
+  {2048, 1024 * 10, 2100, 0.1, true, false},  // kRadix11bitsExtraPass
   {2048, 1024 * 10, 2100, 0.2, true, true}};
 
 INSTANTIATE_TEST_CASE_P(SelectKCsrTest,