Rebase SM90 SBO features#51

Open

b8zhong wants to merge 27 commits into

sgl-project:devfrom

bzhng-development:sbo-rebase-on-dev

Commits on Apr 25, 2026

support tvm ffi interfaces

Fridge003
and
rainj-me
committed

Commits on May 2, 2026

rebase on 0426 upstream

rainj-me
authored and
Fridge003
committed
Relax timeout to 180s
Fridge003
committed
pin tvm to 0.1.9
Fridge003
committed

Commits on May 4, 2026

Support wheel compilation (#26)
Fridge003
authored

Commits on May 14, 2026

update readme
Fridge003
committed

Commits on May 20, 2026

Export the PDL utils of DeepGEMM (#34)
b8zhong
authored

Commits on May 29, 2026

Expose BF16 grouped GEMM wrappers

popsiclexu
authored and
Fridge003
committed
Fix version of dev branch to 0.0.0
Fridge003
committed

Commits on Jun 1, 2026

Fix IMA guard in paged MQA logits scheduler (#38)

nvjullin
and
liji-nv
authored

Commits on Jun 9, 2026

Fix various issues in DeepGEMM tests (#39)

authored

Commits on Jun 10, 2026

Change license in pyproject.toml to avoid build and publish failures with setuptools>=77

b8zhong
and
Brayden Zhong
authored

Commits on Jun 11, 2026

Move run tests script for DeepGEMM (#42)
b8zhong
authored

Commits on Jun 12, 2026

Support num_heads == 16 in MQA logits (#43)
zRzRzRzRzRzRzR
authored

Commits on Jun 15, 2026

Commits on Jun 16, 2026

Add Hopper mega moe test to runner (#46)
Fridge003
authored

Commits on Jun 18, 2026

chore: bump apache-tvm-ffi 0.1.9 -> 0.1.11 (#47)
MartinHua
authored

Commits on Jun 22, 2026

feat: add signal for SBO in SM90 masked gemm
Brayden Zhong
committed
feat: add test for signal GEMM
Brayden Zhong
committed