[UT][intel GPU] use current_platform instead of device hardcode in v1 tests (#20169)

Signed-off-by: Ma, Liangliang <liangliang.ma@intel.com>
This commit is contained in:
Liangliang Ma
2025-07-02 09:06:04 +08:00
committed by GitHub
parent 3be8d312a2
commit a0389e0554
10 changed files with 44 additions and 26 deletions

View File

@@ -217,7 +217,8 @@ try:
is_vllm_fa = True
except ImportError:
# For rocm use upstream flash attention
from flash_attn import flash_attn_varlen_func
if current_platform.is_rocm():
from flash_attn import flash_attn_varlen_func
is_vllm_fa = False
if TYPE_CHECKING: