Merge pull request #424 from senior-zero/fix-main/github/radix_sort_test

Fix segfault in device radix sort tests
NVIDIA · Jan 20, 2022 · d5d009a · d5d009a
2 parents 82be33c + c923dc9
commit d5d009a
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/test/test_device_radix_sort.cu b/test/test_device_radix_sort.cu
@@ -1242,9 +1242,19 @@ void TestSegments(
     }
 
     // Test single segment
-    if (num_items < 128 * 1000 || pre_sorted) {
-        // Right now we assign a single thread block to each segment, so lets keep it to under 128K items per segment
-        TestSegmentIterators(h_keys, num_items, 1, pre_sorted, h_segment_offsets, d_segment_offsets);
+    if (num_items > 0)
+    {
+      if (num_items < 128 * 1000 || pre_sorted)
+      {
+        // Right now we assign a single thread block to each segment, so lets
+        // keep it to under 128K items per segment
+        TestSegmentIterators(h_keys,
+                             num_items,
+                             1,
+                             pre_sorted,
+                             h_segment_offsets,
+                             d_segment_offsets);
+      }
     }
 
     if (h_segment_offsets) delete[] h_segment_offsets;