mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
950dc1b457
Fixes https://github.com/pytorch/pytorch/issues/69674 The fix is Back Compatible with any Caffe2 build. It simply tries to use `onnxptimizer` module when `onnx.optimizer` is not available. `onnx.optimizer` does not exist since ONNX 1.9 (April 2021) as the code was moved to a different [repo](https://github.com/onnx/onnxoptimizer) If both `onnx<1.9` and `onnxoptimizer` are not found, the current fallback behavior is maintained (no ONNX optimization happens). Otherwise, the ONNX optimization pass will run from whatever module it is found. This PR does not require or enforce a direct package dependency to work Pull Request resolved: https://github.com/pytorch/pytorch/pull/75718 Approved by: https://github.com/BowenBao, https://github.com/malfet |
||
|---|---|---|
| .. | ||
| bin | ||
| tests | ||
| __init__.py | ||
| backend.py | ||
| backend_cpp_rep.py | ||
| backend_rep.py | ||
| error.py | ||
| frontend.py | ||
| helper.py | ||
| onnxifi.py | ||
| ONNXOpCoverage.md | ||
| README.md | ||
| test_onnxifi.py | ||
| workspace.py | ||
Caffe2 implementation of Open Neural Network Exchange (ONNX)
Usage
Installation
onnx-caffe2 is installed as a part of Caffe2. Please follow the instructions to install Caffe2.
Folder Structure
- ./: the main folder that all code lies under
- frontend.py: translate from caffe2 model to onnx model
- backend.py: execution engine that runs onnx on caffe2
- tests/: test files
Testing
onnx-caffe2 uses pytest as test driver. In order to run tests, first you need to install pytest:
pip install pytest-cov
After installing pytest, do
pytest
to run tests.
Testing coverage issues/status: https://github.com/caffe2/caffe2/blob/master/caffe2/python/onnx/ONNXOpCoverage.md
Development
During development it's convenient to install caffe2 in development mode:
cd /path/to/caffe2
pip install -e caffe2/