[cpu/arm64] support run sglang on arm64 cpu (#14867)

2026-01-11 04:27:19 -08:00
parent f9fc50acd6
commit 2f4a6addf3
5 changed files with 17 additions and 26 deletions
@@ -116,7 +116,7 @@ class RotaryEmbedding(MultiPlatformOp):

        if (
            (not (_is_cuda or _is_npu) or self.head_size not in [64, 128, 256, 512])
-            and not (_is_cpu and _is_cpu_amx_available)
+            and not (_is_cpu)
            and not (_is_xpu)
        ):
            if _is_cuda or _is_hip: