Enable fa3 PDL by compiling it with corresponding flags (#18756)

2026-02-18 01:12:05 -08:00
parent 390c154306
commit 90d5e27f79
1 changed files with 2 additions and 0 deletions
--- a/sgl-kernel/CMakeLists.txt
+++ b/sgl-kernel/CMakeLists.txt
@@ -444,6 +444,8 @@ if (SGL_KERNEL_ENABLE_FA3)
        "-DCUTLASS_TEST_LEVEL=0"
        "-DCUTLASS_TEST_ENABLE_CACHED_RESULTS=1"
        "-DCUTLASS_DEBUG_TRACE_LEVEL=0"
+        "-DCUTLASS_ENABLE_GDC_FOR_SM90"  # For PDL
+        "-DCUTE_SM90_EXTENDED_MMA_SHAPES_ENABLED"  # Necessary for the WGMMA shapes that we use
        "--expt-relaxed-constexpr"
        "--expt-extended-lambda"
        "--use_fast_math"