Closed ramiro050 closed 2 years ago
This commit adds the file utils.py to share the logic used by torchbench.py to compile, time, and check the torchbench models with other models not in torchbench. This commit also adds an script that runs Bert on IREE and PyTorch eager.
utils.py
torchbench.py
Bert
This commit adds the file
utils.pyto share the logic used bytorchbench.pyto compile, time, and check the torchbench models with other models not in torchbench. This commit also adds an script that runsBerton IREE and PyTorch eager.