From fdf6c70e3a86f111ae167f7d5b12cdda15b4ce25 Mon Sep 17 00:00:00 2001
From: Ma Mingfei <mingfei.ma@intel.com>
Date: Thu, 9 Jul 2026 14:36:45 +0800
Subject: [PATCH] fix arm test_norm.py error (#30597)

---
 sgl-kernel/csrc/cpu/norm.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/sgl-kernel/csrc/cpu/norm.cpp b/sgl-kernel/csrc/cpu/norm.cpp
index 9c54c0c46..76cce6a85 100644
--- a/sgl-kernel/csrc/cpu/norm.cpp
+++ b/sgl-kernel/csrc/cpu/norm.cpp
@@ -397,6 +397,7 @@ void norm4d_kernel_impl(
     const scalar_t* __restrict__ input,
     const NormParams& p,
     const scalar_t* __restrict__ gate = nullptr) {
+#if defined(CPU_CAPABILITY_AVX512)
   // fast path only applies to bfloat16 when D in {32, 64, 128, 256, 512}
   if constexpr (std::is_same_v<scalar_t, at::BFloat16>) {
     switch (p.D) {
@@ -409,6 +410,7 @@ void norm4d_kernel_impl(
         break;
     }
   }
+#endif
 
   // generic path
   LAUNCH_PARALLEL_LOOP(