e31bd697794c67a6b34e43a08b0eb1bec8815790
flash_attn skip worked but Q-tiling fallback still OOMs at 4096 tokens. K tensor: [28_heads, 4096, 256] float32 = 112MB per layer slice. At 256 tokens: [28, 256, 256] = 7MB — safe for profiling. This only affects profiling dummy batch size. Real inference chunked prefill still processes up to max_model_len tokens.
project_6
Description
Languages
C++
41.8%
Cuda
31.6%
Python
22.2%
C
2.1%
CMake
1.1%
Other
1.1%