[feat] add CUTLASS kernel backend for HSTU attention#465

Merged

tiankongdeguiji merged 25 commits intoalibaba:masterfrom

tiankongdeguiji:feat/cutlass-hstu-attn

Apr 8, 2026

Commits on Apr 3, 2026

[feat] add CUTLASS kernel backend for HSTU attention

tiankongdeguiji
and
claude
committed
[fix] handle CUTLASS kernel unsupported param combinations

tiankongdeguiji
and
claude
committed
[feat] add hstu_attn wheel to extra requirements

tiankongdeguiji
and
claude
committed
[feat] add hstu_attn wheels for cp310/cp312

tiankongdeguiji
and
claude
committed
[docs] move CUTLASS usage guide from FAQ to dlrm_hstu.md

tiankongdeguiji
and
claude
committed
[docs] simplify CUTLASS docs into kernel description

tiankongdeguiji
and
claude
committed
[docs] use repo.html style install for hstu_attn like dynamicemb

tiankongdeguiji
and
claude
committed
[docs] add cu126 option for hstu_attn install

tiankongdeguiji
and
claude
committed
[feat] add logging warning when CUTLASS falls back to Triton

tiankongdeguiji
and
claude
committed
[feat] add logging warning in cutlass_cached_hstu_mha fallback

tiankongdeguiji
and
claude
committed
[feat] add CUTLASS provider to hstu_attention_bench.py

tiankongdeguiji
and
claude
committed
[fix] move CUTLASS fallback logic to dispatch layer, add export test

tiankongdeguiji
and
claude
committed
[chore] bump version to 1.1.7

tiankongdeguiji
and
claude
committed

Commits on Apr 8, 2026