Add tags parameter support to @register_benchmark decorator #663
+66
−5
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Implements the feature requested in PR #647 comments - allows users to specify tags directly in the decorator
Tags from the decorator are merged with auto-detected tags (from AST analysis and name-based heuristics). Added
REGISTERED_BACKEND_TAGSglobal dict to store decorator tags, and updated the tagging system inrun.pyto merge all tag sources.Apply a patch locally to test the new functionality
in
tritonbench/operators/gemm/operator.py, add a dummy tag to triton backendrun the analyzer to see it appears
we can see dummy tag appears in
triton_tutorial_matmul