[EPIC]: Optimize Segmented Scan #7850
Description
Is this a duplicate?
- I confirmed there appear to be no duplicate issues for this request and that I agree to the Code of Conduct
Area
CUB
Is your feature request related to a problem? Please describe.
As of #6022, CUB provides device-level segmented scan primitive. The MVP assigns a single block per CTA, which leads to suboptimal performance.
Describe the solution you'd like
This issue will track various optimizations that we have in mind for segmented scan.
Describe alternatives you've considered
No response
Additional context
No response