feat: Integrate CuTe DSL FMHA prefill kernels by loading cubin#3039

Merged

aleozlx merged 30 commits into

flashinfer-ai:mainfrom

limin2021:integrate_dsl_cubin_fmha

Apr 24, 2026

Commits on Apr 12, 2026

feat: Integrate CuTe DSL FMHA cubin kernels into prefill backend

limin2021
and
claude
committed
feat: Add FP8 support and fix TMA padding for DSL FMHA prefill

limin2021
and
claude
committed
feat: Add TVM-FFI support for DSL FMHA prefill kernels

limin2021
and
claude
committed
refactor: Consolidate DSL FMHA prefill tests, add ragged shape coverage

limin2021
and
claude
committed
fix: Use front-padding for ragged varlen tensors (match DSL example)

limin2021
and
claude
committed
feat: Add cute-dsl to benchmark framework, fix CUDA graph compatibility

limin2021
and
claude
committed

Commits on Apr 13, 2026

Commits on Apr 15, 2026

Commits on Apr 16, 2026

Commits on Apr 20, 2026

Commits on Apr 21, 2026

Commits on Apr 22, 2026

Merge branch 'main' into integrate_dsl_cubin_fmha
yzh119
authored

Commits on Apr 23, 2026

Merge branch 'main' into integrate_dsl_cubin_fmha

limin2021
and
claude
committed