Minor style fix for sgl-kernel (#4243)

2025-03-09 20:15:13 -07:00
parent 4a05bdfa86
commit 730d084f2a
5 changed files with 20 additions and 22 deletions
--- a/sgl-kernel/setup.py
+++ b/sgl-kernel/setup.py
@@ -76,6 +76,7 @@ nvcc_flags = [
    "-std=c++17",
    "-use_fast_math",
    "-DFLASHINFER_ENABLE_F16",
+    "-DCUTLASS_ENABLE_TENSOR_CORE_MMA=1",
    "-DCUTLASS_VERSIONS_GENERATED",
    "-DCUTE_USE_PACKED_TUPLE=1",
    "-DCUTLASS_TEST_LEVEL=0",