util/gpu_t.cuh: make launch_params_t more forgiving.

supranational · Feb 29, 2024 · 2e56622 · 2e56622
1 parent d3d17ad
commit 2e56622
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/util/gpu_t.cuh b/util/gpu_t.cuh
@@ -50,7 +50,9 @@ struct launch_params_t {
     size_t shared;
 
     launch_params_t(dim3 g, dim3 b, size_t sz = 0) : gridDim(g), blockDim(b), shared(sz) {}
-    launch_params_t(int g, int b, size_t sz = 0) : gridDim(g), blockDim(b), shared(sz) {}
+    launch_params_t(int g, int b, size_t sz = 0) : gridDim(g>0 ? g : 0), blockDim(b), shared(sz) {}
+    launch_params_t(unsigned int g, unsigned int b, size_t sz = 0) : gridDim(g), blockDim(b), shared(sz) {}
+
 };
 
 class stream_t {