Fix pre-volta tests

NVIDIA · Nov 14, 2023 · cce16b0 · cce16b0
1 parent d295ecd
commit cce16b0
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/static_map/insert_and_find_test.cu b/tests/static_map/insert_and_find_test.cu
@@ -61,7 +61,14 @@ __global__ void parallel_sum(Ref v)
     }
 #if __CUDA_ARCH__ < 700
     else {
-      v.insert(cuco::pair{i, gridDim.x * blockDim.x});
+      auto constexpr cg_size = Ref::cg_size;
+      if constexpr (cg_size == 1) {
+        v.insert(cuco::pair{i, gridDim.x * blockDim.x});
+      } else {
+        auto const tile =
+          cooperative_groups::tiled_partition<cg_size>(cooperative_groups::this_thread_block());
+        v.insert(tile, cuco::pair{i, gridDim.x * blockDim.x / cg_size});
+      }
     }
 #endif
   }