[sgl-kernel/cpu]: exclude amx gemm source from arm build (#29286)

2026-06-26 08:23:43 +08:00
parent ed71fb8f95
commit 4b0b0ea55f
3 changed files with 18 additions and 2 deletions
@@ -473,7 +473,6 @@ at::Tensor int8_scaled_mm_cpu(
  return out;
 }

-#ifndef __aarch64__
 // fused `per_token_quant_int8_cpu` and `int8_scaled_mm_cpu`
 at::Tensor int8_scaled_mm_with_quant(
    at::Tensor& mat1,
@@ -540,4 +539,3 @@ at::Tensor int8_scaled_mm_with_quant(
  });
  return out;
 }
-#endif  // #ifndef __aarch64__