Add end-to-end tests for iree_linalg_ext.custom_op.
#18764
+261
−37
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
MaheshRavishankar
force-pushed
the
custom_tilable_op_e2e
branch
8 times, most recently
from
a3e1227
to
691e8c2
Compare
This PR plumbs through e2e tests for `iree_linalg_ext.custom_op` on CPU and AMDGPU backends. The end-to-end test does a matmul + bias add with and without the use of `custom_op` and checks that the same value is obtained. Signed-off-by: MaheshRavishankar <[email protected]>
Signed-off-by: MaheshRavishankar <[email protected]>
Signed-off-by: MaheshRavishankar <[email protected]>
MaheshRavishankar
force-pushed
the
custom_tilable_op_e2e
branch
from
691e8c2
to
b7f1c89
Compare
|
Needs upstream patch to submit this, but apart from that this is ready for review |
MaheshRavishankar
requested review from
hanhanW,
IanWood1,
qedawkins,
kuhar and
Groverkss
as code owners
MaheshRavishankar
changed the title
iree_linalg_ext.custom_op.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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.
This PR plumbs through e2e tests for
iree_linalg_ext.custom_oponCPU and AMDGPU backends. The end-to-end test does a matmul + bias add
with and without the use of
custom_opand checks that the same valueis obtained.
Signed-off-by: MaheshRavishankar [email protected]