e9eaad0592d9455332db76dd266a8bacc7c36d30
CCCL dispatch_transform.cuh spread_out_items_per_thread pattern: reduce tile count = reduce Python loop iterations = faster prefill. At 256K context with q_len=4096: old 96MB budget → 219 KV tokens/tile → ~1200 tiles per layer → 16 min per chunk. New 256MB budget → ~580 KV tokens/tile → ~450 tiles per layer → ~6 min per chunk. BI-V100 has 32 GB HBM; 256 MB temporary tensor is safe.
project_6
Description
Languages
C++
41.5%
Cuda
31.3%
Python
22.8%
C
2.1%
CMake
1.1%
Other
1.1%