[CI/Build] Fix CPU platform pre-commit formatting (#52981)
Signed-off-by: mgoin <mgoin64@gmail.com> Co-authored-by: OpenAI Codex <codex@openai.com>
Misha Goin committed
Aug 19, 2026 at 13:32 UTC
f76d71d7ceac040d555323f36b2f6afeab0e8df2
1 file changed
+5
-2
vllm/platforms/cpu.py
+5
-2
@@ -181,13 +181,16 @@ class CpuPlatform(Platform):
181
"otherwise the performance is not optimized."
182
)
183
184
- # Accelerated GDN (AMX tiles or AVX-512BF16 VDPBF16PS) requires float32 SSM state.
184
+ # Accelerated GDN (AMX tiles or AVX-512BF16 VDPBF16PS) requires
185
+ # float32 SSM state.
186
if (
187
torch.cpu._is_avx512_bf16_supported()
188
and cache_config.mamba_ssm_cache_dtype != "float32"
189
):
190
cache_config.mamba_ssm_cache_dtype = "float32"
190
- logger.warning("Reset SSM cache type to float32 for accelerated GDN mamba attention.")
191
+ logger.warning(
192
+ "Reset SSM cache type to float32 for accelerated GDN mamba attention."
193
+ )
194
195
# Lagecy setting
196
env_key = "VLLM_CPU_KVCACHE_SPACE"