Ean Garvey 0ee515a7be Fix GPU benchmarks on PyTorch model tests. (#218) ...

* Add CUDA_BENCHMARKS option to setup.venv to enable PyTorch benchmarks on CUDA.

* Fix PyTorch GPU benchmarks for tank models.

2022-07-29 15:17:55 -07:00

..

nightly.yml

Divide iree_utils and do module imports on function calls.

2022-06-22 14:17:33 +05:30

test-models.yml

Fix GPU benchmarks on PyTorch model tests. (#218)

2022-07-29 15:17:55 -07:00