Adds Grouped and Batched GEMM kernels with blockscaling matching DeepGEMM API#433

Open

aryaman-gupta wants to merge 58 commits into

mainfrom

aryaman/group-gemm

Commits on Mar 27, 2026

Commits on Mar 31, 2026

corrects test_grouped_gemm
aryaman-gupta
committed
test_grouped_gemm: add argparse CLI entry point

aryaman-gupta
and
claude
committed
test_grouped_gemm: use test_common verify_output and run_perftest

aryaman-gupta
and
claude
committed
test_grouped_gemm: add large_shape marks to expensive test cases

aryaman-gupta
and
claude
committed
test_grouped_gemm: add m_per_group sweep mode (--m_per_group 0)

aryaman-gupta
and
claude
committed
test_grouped_gemm: default to sweep mode (--m_per_group 0)

aryaman-gupta
and
claude
committed
test_grouped_gemm: match perf output format to blockscale test

aryaman-gupta
and
claude
committed
test_grouped_gemm: wire argparse args through to test functions

aryaman-gupta
and
claude
committed
test_grouped_gemm: fix device mismatch with set_default_device

aryaman-gupta
and
claude
committed
grouped_gemm.py: fixes bug in buffer_load offsets
aryaman-gupta
committed
grouped_gemm.py: adds LDS ping-pong
aryaman-gupta
committed

Commits on Apr 1, 2026

Commits on Apr 17, 2026

Commits on Apr 23, 2026

Commits on Apr 24, 2026

Commits on Apr 30, 2026

Commits on May 4, 2026

Commits on May 5, 2026

group gemm: drop redundant arith.index_cast(T.i64, ...) on num_records_bytes

aryaman-gupta
and
claude
committed