update norm cu (#3048)

This commit is contained in:
Yineng Zhang
2025-01-22 18:58:44 +08:00
committed by GitHub
parent 6fc37bd8ee
commit 9f8f2c7f74
2 changed files with 1 additions and 29 deletions

View File

@@ -91,7 +91,7 @@ ext_modules = [
"src/sgl-kernel/csrc/sampling_scaling_penalties.cu",
"src/sgl-kernel/csrc/sgl_kernel_ops.cu",
"src/sgl-kernel/csrc/rotary_embedding.cu",
"src/sgl-kernel/csrc/norm.cu",
"3rdparty/flashinfer/csrc/norm.cu",
],
include_dirs=include_dirs,
extra_compile_args={