minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

fix arm test_norm.py error (#30597)

Browse Source
This commit is contained in:
Ma Mingfei
2026-07-09 14:36:45 +08:00
committed by GitHub
parent 0562ccb1a8
commit fdf6c70e3a
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sgl-kernel/csrc/cpu/norm.cpp
+2
View File
@@ -397,6 +397,7 @@ void norm4d_kernel_impl(
const scalar_t* __restrict__ input, const scalar_t* __restrict__ input,
const NormParams& p, const NormParams& p,
const scalar_t* __restrict__ gate = nullptr) { 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} // fast path only applies to bfloat16 when D in {32, 64, 128, 256, 512}
if constexpr (std::is_same_v<scalar_t, at::BFloat16>) { if constexpr (std::is_same_v<scalar_t, at::BFloat16>) {
switch (p.D) { switch (p.D) {
@@ -409,6 +410,7 @@ void norm4d_kernel_impl(
break; break;
} }
} }
#endif
// generic path // generic path
LAUNCH_PARALLEL_LOOP( LAUNCH_PARALLEL_LOOP(