mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00

History

Xuehai Pan 8d45f555d7 [BE] [1/3] Rewrite `super()` calls in caffe2 and benchmarks (#94587 ) Rewrite Python built-in class `super()` calls. Only non-semantic changes should be applied. - #94587 - #94588 - #94592 Also, methods with only a `super()` call are removed: ```diff class MyModule(nn.Module): - def __init__(self): - super().__init__() - def forward(self, ...): ... ``` Some cases that change the semantics should be kept unchanged. E.g.: `f152a79be9/caffe2/python/net_printer.py (L184-L190)` `f152a79be9/test/test_jit_fuser_te.py (L2628-L2635)` Pull Request resolved: https://github.com/pytorch/pytorch/pull/94587 Approved by: https://github.com/ezyang		2023-02-11 18:19:48 +00:00
..
bin
tests	Bump to stable ONNX 1.13.0 (#90332 )	2023-02-08 11:49:06 +00:00
__init__.py
backend_cpp_rep.py	[BE] [1/3] Rewrite `super()` calls in caffe2 and benchmarks (#94587 )	2023-02-11 18:19:48 +00:00
backend_rep.py	[BE] [1/3] Rewrite `super()` calls in caffe2 and benchmarks (#94587 )	2023-02-11 18:19:48 +00:00
backend.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

onnx-caffe2 is installed as a part of Caffe2. Please follow the instructions to install Caffe2.

./: 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

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.

During development it's convenient to install caffe2 in development mode:

cd /path/to/caffe2
pip install -e caffe2/