AMD CI Test - unskip moe_sum test and moe_align_block_size tests (#32039)

Signed-off-by: Hongxia Yang <hongxia.yang@amd.com>
This commit is contained in:
Hongxia Yang
2026-01-14 02:25:10 -05:00
committed by GitHub
parent 7933638051
commit 048bb59728
3 changed files with 11 additions and 4 deletions

View File

@@ -187,6 +187,17 @@ class RocmPlatform(Platform):
if not on_gfx9():
supported_quantization += ["bitsandbytes"]
@classmethod
def import_kernels(cls) -> None:
"""Import ROCm-specific kernels."""
super().import_kernels()
import contextlib
# Import ROCm-specific extension
with contextlib.suppress(ImportError):
import vllm._rocm_C # noqa: F401
@classmethod
def get_attn_backend_cls(
cls,