The assert fix for block_radix_sort (#714)

NB4444 · web-flow · commit 55aaa6708b21 · 2025-04-16T09:21:48.000-04:00
diff --git a/rocprim/include/rocprim/block/block_radix_sort.hpp b/rocprim/include/rocprim/block/block_radix_sort.hpp
@@ -164,7 +164,7 @@ class block_radix_sort
         ROCPRIM_DEVICE ROCPRIM_FORCE_INLINE
     block_radix_sort()
     {
-        assert(BlockSize % ::rocprim::arch::wavefront::size() == 0);
+        assert(!warp_striped || BlockSize % ::rocprim::arch::wavefront::size() == 0);
     }
 
     /// \brief Performs ascending radix sort over keys partitioned across threads in a block.

Original file line number	Diff line number	Diff line change
`@@ -164,7 +164,7 @@ class block_radix_sort`
`164`	`164`	`ROCPRIM_DEVICE ROCPRIM_FORCE_INLINE`
`165`	`165`	`block_radix_sort()`
`166`	`166`	`{`
`167`		`- assert(BlockSize % ::rocprim::arch::wavefront::size() == 0);`
	`167`	`+ assert(!warp_striped \|\| BlockSize % ::rocprim::arch::wavefront::size() == 0);`
`168`	`168`	`}`
`169`	`169`
`170`	`170`	`/// \brief Performs ascending radix sort over keys partitioned across threads in a block.`