-
Notifications
You must be signed in to change notification settings - Fork 701
Remove triton.ops, copy necessary bits here #1413
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.
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
Conversation
Summary: Triton upstream removed `triton.ops` and moved it to a semi-unmaintained `kernels` repo. Since all that's needed here is the perf model, just add those bits here.
@@ -0,0 +1,208 @@ | |||
import functools |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This file should have a reference comment to the triton upstream it was copies from, as well as suitable license attribution.
import functools | |
# Adapted from https://github.com/triton-lang/kernels/blob/eeeebdd8be7d13629de22d600621e6234057eed3/kernels/matmul_perf_model.py | |
# https://github.com/triton-lang/kernels is licensed under the MIT License. | |
import functools |
should probably be enough.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done!
Hi @akx, any chance to get this merged? Thanks! |
@bertmaher Sorry, I don't have merge privileges here, just a humble contributor. Cc @matthewdouglas :) |
LGTM, thanks! |
…1413) * Remove triton.ops, copy necessary bits here Summary: Triton upstream removed `triton.ops` and moved it to a semi-unmaintained `kernels` repo. Since all that's needed here is the perf model, just add those bits here. * Add source reference/license comment
Summary: Triton upstream removed
triton.ops
and moved it to a semi-unmaintainedkernels
repo. Since all that's needed here is the perf model, just add those bits here.