Enable onesweep for I8/I16

gevtushenko · May 7, 2023 · d18fa6a · d18fa6a
1 parent f4b02e8
commit d18fa6a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cub/device/dispatch/dispatch_radix_sort.cuh b/cub/device/dispatch/dispatch_radix_sort.cuh
@@ -1005,7 +1005,7 @@ struct DeviceRadixSortPolicy
             PRIMARY_RADIX_BITS     = (sizeof(KeyT) > 1) ? 7 : 5,
             SINGLE_TILE_RADIX_BITS = (sizeof(KeyT) > 1) ? 6 : 5,
             SEGMENTED_RADIX_BITS   = (sizeof(KeyT) > 1) ? 6 : 5,
-            ONESWEEP               = sizeof(KeyT) >= sizeof(uint32_t),
+            ONESWEEP               = true,
             ONESWEEP_RADIX_BITS    = 8,
             OFFSET_64BIT           = sizeof(OffsetT) == 8 ? 1 : 0,
             FLOAT_KEYS             = std::is_same<KeyT, float>::value ? 1 : 0,