Skip to content

[Metal] Add Metal GEMM support with simdgroup_matrix MMA #1869

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
LeiWang1999 merged 16 commits into from
May 22, 2026
Merged

[Metal] Add Metal GEMM support with simdgroup_matrix MMA #1869

Changes from 1 commit
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
4b015bf
[Metal] Add Metal GEMM support with simdgroup_matrix MMA
oraluben Apr 26, 2026
9e45e47
Merge branch 'main' of https://github.com/tile-ai/tilelang into metal…
LeiWang1999 May 7, 2026
a3fdb11
Move Metal buffer helpers to Metal backend
LeiWang1999 May 7, 2026
950d009
fix(metal): remove layout_inference bypass, fix build & clarify docs
oraluben May 11, 2026
ce6e4c2
Merge upstream/main - fix backend import path for metal
oraluben May 12, 2026
eaa9569
Merge main into metal-gemm
oraluben May 21, 2026
0eb2865
fix: tir->tirx migration build fixes for metal backend
oraluben May 21, 2026
ab88cae
fix: improve metal_fragment_to_simdgroup buffer remapping
oraluben May 21, 2026
8e8e83b
lint
oraluben May 21, 2026
5bff0ee
fix: add full-body var subst in MetalFragmentToSimdgroup to reach gem…
oraluben May 21, 2026
cd48d6a
fix: add missing <cmath> include for std::isinf/isnan in codegen_metal
oraluben May 21, 2026
4707ede
Merge branch 'main' into metal-gemm
oraluben May 21, 2026
7f7adfd
[Metal] Fix review findings: dedup warp partition, kernel_only, barri…
oraluben May 21, 2026
db6c2f7
lint
oraluben May 21, 2026
147f37c
Refactor MetalFragmentToSimdgroup import path in LowerAndLegalize fun…
LeiWang1999 May 22, 2026
b7fcad6
Remove metal_fragment_to_simdgroup.py file and update import in metal…
LeiWang1999 May 22, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
lint
  • Loading branch information
@oraluben
oraluben committed May 21, 2026
commit db6c2f75e835909e1ba866a589bf9a18227235ba
8 changes: 5 additions & 3 deletions tilelang/transform/metal_fragment_to_simdgroup.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,11 @@

@lru_cache(maxsize=1)
def _get_gemm_ops():
return frozenset({
Op.get("tl.tileop.gemm"),
})
return frozenset(
{
Op.get("tl.tileop.gemm"),
}
)


def _extract_buffer_var_from_region(region_call):
Expand Down
Loading