Update src/cunumeric/index/select.cu

Co-authored-by: Manolis Papadakis <[email protected]>
nv-legate · Dec 3, 2023 · 348bf7d · 348bf7d
1 parent 02ef7bd
commit 348bf7d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cunumeric/index/select.cu b/src/cunumeric/index/select.cu
@@ -97,7 +97,7 @@ struct SelectImplBody<VariantKind::GPU, CODE, DIM> {
 
       VAL* outptr = out.ptr(rect);
       select_kernel_dense<VAL><<<blocks, THREADS_PER_BLOCK, 0, stream>>>(
-        outptr, narrays, cond_arr, choice_arr, default_val, out_size);
+        outptr, narrays, cond_arr, choice_arr, default_val, rect.volume());
 
     } else {  // not dense
       auto cond_arr =