Skip to content

Commit b3d8f49

Browse files
blake-sncclaude
andcommitted
nit: update SM120 comment to include attention kernels
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 9976cf6 commit b3d8f49

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

flashinfer/aot.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -531,7 +531,7 @@ def gen_all_modules(
531531
if has_sm121:
532532
jit_specs.append(gen_fp4_quantization_sm121_module())
533533
if has_sm120 or has_sm121:
534-
# SM120 and SM121 share the same CUTLASS kernels for fused MOE and GEMM.
534+
# SM120 and SM121 share the same kernels for fused MOE, GEMM, and attention.
535535
# The SM120 module generators use supported_major_versions=[12] which
536536
# compiles for all SM12x targets.
537537
jit_specs.append(gen_cutlass_fused_moe_sm120_module())

0 commit comments

Comments
 (0)