torch-mlir/frontends/pytorch/test/acap_export
Stella Laurenzo 9e52f6235b More progress on PyTorch acap device capture.
* Now gets far enough to capture batch_norm.
* Has some issues still with in-place ops.
* Can materialize constants.
* Includes an upgrade to PyTorch nightly, which has important bug fixes for fallback and boxed kernel dispatch.
* Fixes #78, #79, #80.
* Will do more testing in a follow-up once further bugs are fixed that facilitate getting at the other features.
2020-10-15 21:43:21 -07:00
..
test_export_ResA.py Port prior acap export tests to new dispatcher based versions. 2020-10-13 16:37:46 -07:00
test_export_add3.py Port prior acap export tests to new dispatcher based versions. 2020-10-13 16:37:46 -07:00
test_export_batchnorm.py More progress on PyTorch acap device capture. 2020-10-15 21:43:21 -07:00
test_export_conv2d_back.py Port prior acap export tests to new dispatcher based versions. 2020-10-13 16:37:46 -07:00
test_export_multi_out.py Port prior acap export tests to new dispatcher based versions. 2020-10-13 16:37:46 -07:00
test_export_resnet18.py Port prior acap export tests to new dispatcher based versions. 2020-10-13 16:37:46 -07:00
test_export_vgg11.py Port prior acap export tests to new dispatcher based versions. 2020-10-13 16:37:46 -07:00