Skip to content

Commit f10be07

Browse files
committed
cleaning and writeup
1 parent f406775 commit f10be07

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

README.md

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
# onnx2code
22

3-
Generate plain C/ASM code for inference of ONNX models without dependencies
3+
Generate plain C++ code for inference of ONNX models without dependencies
4+
5+
This project was made as an alternative to a final exam for the assignment "Computer Organization II". You can read the writeup in [docs/TP Final onnx2code.pdf](docs/TP%20Final%20onnx2code.pdf) (in Spanish).
46

57
## Model support
68

@@ -44,16 +46,16 @@ Only `float` data type is supported.
4446
We provide a ready to use [Docker image](https://hub.docker.com/r/mlomb/onnx2code):
4547

4648
```sh
47-
docker run --rm -it -v $pwd/mnist.onnx:/app/input.onnx:ro -v $pwd/output:/app/output:rw mlomb/onnx2code:latest --variations=asm,c --checks=3
49+
docker run --rm -it -v $pwd/mnist.onnx:/app/input.onnx:ro -v $pwd/output:/app/output:rw mlomb/onnx2code:latest --variations=im2col,loop-tiling --checks=3
4850
```
4951

50-
The command above will generate C and ASM code for the `mnist.onnx` model in the `output` folder.
52+
The command above will generate C++ code for the `mnist.onnx` model in the `output` folder.
5153

5254
## Setting up locally
5355

5456
### Prerequisites
5557

56-
* gcc, nasm (required if checking models)
58+
* gcc (required if checking models)
5759
* Python 3.10
5860
* [pipenv](https://pypi.org/project/pipenv/)
5961

@@ -64,5 +66,5 @@ Clone and install dependencies with `pipenv install`.
6466
To generate code from an ONNX model, run the following command inside a pipenv shell:
6567

6668
```sh
67-
python -m onnx2code --variation=asm,c mnist.onnx output_folder --checks=3
69+
python -m onnx2code --variation=im2col,loop-tiling mnist.onnx output_folder --checks=3
6870
```

docs/TP Final onnx2code.pdf

2.22 MB
Binary file not shown.

0 commit comments

Comments
 (0)