cf7824313fe2218a8de7d77e87975c574f425b33
Profiling OOM: flash_attn_varlen_func allocates large temp buffers during profile_run with 8192 dummy tokens. patch_worker_profile_override.py already has skip logic when num_gpu_blocks_override is set. 4000 blocks × 16 tokens/block = 64K token KV capacity. With max-model-len=80000 and prefix caching, this is sufficient. Sub 168 reference: ran 262K context on 0.95 util without override because base image profiling doesn't use flash_attn_varlen (uses native xformers).
project_6
Description
Languages
C++
41.8%
Cuda
31.6%
Python
22.2%
C
2.1%
CMake
1.1%
Other
1.1%