fix bug

HandH1998 · Jan 24, 2025 · 8c3dc13 · 8c3dc13
1 parent 4a98c75
commit 8c3dc13
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py
@@ -68,6 +68,7 @@ def _get_version():
     "src/sgl-kernel/csrc/trt_reduce_kernel.cu",
     "src/sgl-kernel/csrc/moe_align_kernel.cu",
     "src/sgl-kernel/csrc/int8_gemm_kernel.cu",
+    "src/sgl-kernel/csrc/fp8_gemm_kernel.cu",
     "src/sgl-kernel/csrc/sampling_scaling_penalties.cu",
     "src/sgl-kernel/csrc/lightning_attention_decode_kernel.cu",
     "src/sgl-kernel/csrc/sgl_kernel_ops.cu",