Update src/cunumeric/index/select.cu

Co-authored-by: Manolis Papadakis <[email protected]>
nv-legate · Dec 3, 2023 · ca3a9d0 · ca3a9d0
1 parent 348bf7d
commit ca3a9d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cunumeric/index/select.cu b/src/cunumeric/index/select.cu
@@ -83,7 +83,7 @@ struct SelectImplBody<VariantKind::GPU, CODE, DIM> {
 #ifdef DEBUG_CUNUMERIC
     assert(narrays == choicelist.size());
 #endif
-    const size_t blocks = (out_size + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK;
+    const size_t blocks = (rect.volume() + THREADS_PER_BLOCK - 1) / THREADS_PER_BLOCK;
 
     auto stream = get_cached_stream();