Refactor scan_by_key tuning (#3139)

* Densen * Drop unnnecessary comments * Move and rename policy_hub * Drop needless namespace qualifications * Remove uint128 redundancy * Fix spelling of some variable names * Eliminate redundancy * Add TODO comments * Swap sm80 and sm90 tunings * Simplification
NVIDIA · Dec 12, 2024 · 7321a51 · 7321a51
1 parent d95aab7
commit 7321a51
Show file tree

Hide file tree

Showing 2 changed files with 330 additions and 660 deletions.
diff --git a/cub/cub/device/dispatch/dispatch_scan_by_key.cuh b/cub/cub/device/dispatch/dispatch_scan_by_key.cuh
@@ -233,7 +233,7 @@ template <
     cub::detail::value_t<ValuesInputIteratorT>,
     ::cuda::std::_If<std::is_same<InitValueT, NullType>::value, cub::detail::value_t<ValuesInputIteratorT>, InitValueT>>,
   typename SelectedPolicy =
-    DeviceScanByKeyPolicy<KeysInputIteratorT, AccumT, cub::detail::value_t<ValuesInputIteratorT>, ScanOpT>>
+    detail::scan_by_key::policy_hub<KeysInputIteratorT, AccumT, cub::detail::value_t<ValuesInputIteratorT>, ScanOpT>>
 struct DispatchScanByKey : SelectedPolicy
 {
   //---------------------------------------------------------------------