Add unit tests to torchdynamo torch-mlir compiler

iree-org / iree-torch

Torch Frontend for IREE

Apache License 2.0

25 stars 11 forks source link

Add unit tests to torchdynamo torch-mlir compiler #14

Closed ramiro050 closed 2 years ago

ramiro050 commented 2 years ago

This commit adds some unit tests for the torch-MLIR compiler used by torchdynamo, and fixes a few errors in the way that tuple returns were being handled.