diff --git a/flashinfer/jit/gemm/core.py b/flashinfer/jit/gemm/core.py index a26d472f7d..45120e6deb 100644 --- a/flashinfer/jit/gemm/core.py +++ b/flashinfer/jit/gemm/core.py @@ -91,8 +91,6 @@ def gen_gemm_sm100_module_cutlass_fp4() -> JitSpec: + [ "-DENABLE_BF16", "-DENABLE_FP4", - "-DCUTLASS_ENABLE_GDC_FOR_SM100=1", - "-DCUTLASS_ENABLE_GDC_FOR_SM90=1", ], extra_cflags=[ "-DFAST_BUILD", @@ -160,8 +158,6 @@ def gen_gemm_sm103_module_cutlass_fp4() -> JitSpec: + [ "-DENABLE_BF16", "-DENABLE_FP4", - "-DCUTLASS_ENABLE_GDC_FOR_SM100=1", - "-DCUTLASS_ENABLE_GDC_FOR_SM90=1", ], extra_cflags=[ "-DFAST_BUILD", @@ -210,8 +206,6 @@ def gen_gemm_sm120_module_cutlass_fp4() -> JitSpec: + [ "-DENABLE_BF16", "-DENABLE_FP4", - "-DCUTLASS_ENABLE_GDC_FOR_SM100=1", - "-DCUTLASS_ENABLE_GDC_FOR_SM90=1", ], extra_cflags=[ "-DFAST_BUILD", @@ -262,8 +256,6 @@ def gen_gemm_sm100_module_cutlass_fp8() -> JitSpec: extra_cuda_cflags=nvcc_flags + [ "-DENABLE_BF16", - "-DCUTLASS_ENABLE_GDC_FOR_SM100=1", - "-DCUTLASS_ENABLE_GDC_FOR_SM90=1", ], extra_cflags=[ "-DFAST_BUILD", @@ -357,8 +349,6 @@ def gen_gemm_sm100_module_cutlass_mxfp8() -> JitSpec: extra_cuda_cflags=nvcc_flags + [ "-DENABLE_BF16", - "-DCUTLASS_ENABLE_GDC_FOR_SM100=1", - "-DCUTLASS_ENABLE_GDC_FOR_SM90=1", ], extra_cflags=[ "-DFAST_BUILD", @@ -526,11 +516,7 @@ def gen_gemm_sm120_module() -> JitSpec: return gen_jit_spec( "gemm_sm120", source_paths, - extra_cuda_cflags=nvcc_flags - + [ - "-DCUTLASS_ENABLE_GDC_FOR_SM100=1", - "-DCUTLASS_ENABLE_GDC_FOR_SM90=1", - ], + extra_cuda_cflags=nvcc_flags, )