iree
iree copied to clipboard
Add end-to-end tests for `iree_linalg_ext.custom_op`.
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]
Needs upstream patch to submit this, but apart from that this is ready for review
tests are always good; @MaheshRavishankar, is this something you can rebase or is it stale?
Its on my back log to rebase and land this. Have been having unrelated failures.