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.8%
Cuda
31.6%
Python
22.2%
C
2.1%
CMake
1.1%
Other
1.1%