iree-org / iree-torch

Torch Frontend for IREE
Apache License 2.0
25 stars 11 forks source link

Add benchmark functionality and correctness checks #15

Closed ramiro050 closed 2 years ago

ramiro050 commented 2 years ago

This commit adds support for timing the iterations of a model being run by torchbench.py. For multiple iterations, the script also outputs the mean time and standard deviation. This commit also adds a simple correctness check of the output from the torch-mlir compiled module against the eager equivalent in pytorch.