Check if selected backend is None in get_attn_backend_cls() (vllm-project#12975)

Signed-off-by: Yuan Tang <[email protected]>
Signed-off-by: Louis Ulmer <[email protected]>

Author: terrytangyuan

vllm/platforms/cpu.py

@@ -35,7 +35,7 @@ def get_attn_backend_cls(cls, selected_backend: _Backend, head_size: int,
                              dtype: torch.dtype, kv_cache_dtype: Optional[str],
                              block_size: int, use_v1: bool,
                              use_mla: bool) -> str:
-        if selected_backend != _Backend.TORCH_SDPA:
+        if selected_backend and selected_backend != _Backend.TORCH_SDPA:
             logger.info("Cannot use %s backend on CPU.", selected_backend)
         logger.info("Using Torch SDPA backend.")
         return "vllm.attention.backends.torch_sdpa.TorchSDPABackend"