Skip to content

[Triton] GEMM tunning script #1833

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

Jump to bottom
Open
k50112113 wants to merge 26 commits into
base: main
Choose a base branch
from
Open

[Triton] GEMM tunning script #1833

k50112113 wants to merge 26 commits into from

Conversation

@k50112113
Copy link
Contributor

@k50112113 k50112113 commented Jan 13, 2026

This PR adds GEMM tunning scripts that

  1. screen through parameter space and select the most performant config
  2. generate JSON config files
  3. verify performance results

k50112113 added 22 commits January 5, 2026 20:59
@k50112113
add files
036b2b9
@k50112113
update
ab26ea4
@k50112113
Merge remote-tracking branch 'upstream/main' into shaoclee/triton_gem…
a5b858e 
…m_tunning
@k50112113
update
379a2d7
@k50112113
add a8w8 gemm blockscale ut example
e88e773
@k50112113
add a8w8 gemm per token scale example
ceebc42
@k50112113
add ut template
d5b2d64
@k50112113
rename
d3c7482
@k50112113
Merge remote-tracking branch 'upstream/main' into shaoclee/triton_gem…
b8a846e 
…m_tunning
@k50112113
update
aaab52d
@k50112113
Merge remote-tracking branch 'upstream/main' into shaoclee/triton_gem…
ed8832f 
…m_tunning
@k50112113
error message parsing and skipping failed BLOCK shapes
b022b1d
@k50112113
update a16w16
314f627
@k50112113
Merge remote-tracking branch 'upstream/main' into shaoclee/triton_gem…
3adec3c 
…m_tunning
@k50112113
fix JSON output filename and add more example commands
4cf2a56
@k50112113
update .gitignore
f072fd0
@k50112113
update
d09288c
@k50112113
update example command
791fb33
@k50112113
_utils relative import
10b109d
@k50112113
add verify-perf.py and move example command into README.md and clean …
51eb060 
…up code
@k50112113
verbose and add update argparse and update pre-prune rules
0bfb99e
@k50112113
Merge remote-tracking branch 'upstream/main' into shaoclee/triton_gem…
a3b5815 
…m_tunning
@k50112113 k50112113 requested review from a team and azaidy January 13, 2026 22:11
@brunomazzottiamd brunomazzottiamd mentioned this pull request Jan 14, 2026
Merged
1 task
@cagrikymk cagrikymk assigned cagrikymk and unassigned cagrikymk Jan 14, 2026
@cagrikymk cagrikymk self-requested a review January 14, 2026 19:39
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@azaidy azaidy Awaiting requested review from azaidy

@cagrikymk cagrikymk Awaiting requested review from cagrikymk

At least 1 approving review is required to merge this pull request.

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@k50112113 @cagrikymk