[test] Add conv correctness test suite

nod-ai / SHARK-Turbine

Unified compiler/runtime for interfacing with PyTorch Dynamo.

Apache License 2.0

82 stars 41 forks source link

Open antiagainst opened 2 months ago

antiagainst commented 2 months ago

https://github.com/openxla/iree/pull/16849 landed for this. @pashu123 is there more work planned?

pashu123 commented 2 months ago

openxla/iree#16849 landed for this. @pashu123 is there more work planned?

Yes, there's a minor work to add attributes to enable this for GPU.

pashu123 commented 2 months ago

@antiagainst If this is a priority, I can add the feature.