dylanyunlon e9eaad0592 perf: increase prefix attention tile budget 96MB→256MB
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.
2026-08-07 04:44:18 +00:00
2026-07-30 17:03:23 +08:00

project_6

Description
No description provided
Readme 427 MiB
Languages
C++ 41.8%
Cuda 31.6%
Python 22.2%
C 2.1%
CMake 1.1%
Other 1.1%