One more check to make sure.

dmlc · Jul 12, 2024 · 82f1013 · 82f1013
1 parent 0fb9a5b
commit 82f1013
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/graphbolt/src/cnumpy.cc b/graphbolt/src/cnumpy.cc
@@ -220,6 +220,7 @@ void OnDiskNpyArray::IndexSelectIOUringImpl(
         read_queue.Clear();
         // Wait for the reads.
         struct io_uring_cqe *cqe;
+        TORCH_CHECK(num_submitted - num_completed <= 2 * kGroupSize);
         TORCH_CHECK(
             ::io_uring_wait_cqe_nr(
                 &my_io_uring_queue, &cqe, num_submitted - num_completed) == 0);