Skip to content

Commit

Permalink
Merge branch 'master' of https://github.com/comp-physics/roseNNa
Browse files Browse the repository at this point in the history
  • Loading branch information
ajaybati committed Jun 9, 2023
2 parents cd9a256 + bc8560a commit 04a833d
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/CI.yml
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,6 @@ jobs:
- name: Running Test Cases
run: |
mkdir fLibrary/objFiles/
mkdir -p fLibrary/objFiles/
cd test/
./run.sh
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ python3 ../goldenFiles/gemm_small/gemm_small.py
```
``` bash
#read and interpret the correspoding output files from last step
python3 modelParserONNX.py -w ../goldenFiles/gemm_samll/gemm_small.onnx -f ../goldenFiles/gemm_samll/gemm_small_weights.onnx
python3 modelParserONNX.py -w ../goldenFiles/gemm_small/gemm_small.onnx -f ../goldenFiles/gemm_small/gemm_small_weights.onnx
```
``` bash
#compile the library
Expand All @@ -71,7 +71,7 @@ gfortran -o flibrary libcorelib.a capiTester.o
```
``` bash
#check whether python output from PyTorch model = roseNNa's output
python3 ../test/testChecker.py
python3 ../test/testChecker.py gemm_small
```

## Compiling roseNNa
Expand Down Expand Up @@ -115,7 +115,7 @@ torch.onnx.export(model, # model being run

3. **Preprocess the model**

`fLibrary/` holds the library files that recreate the model and run inference on it. It contains a `Makefile` that first pre-processes the model. First, run `make preprocess args="path/to/model/structure path/to/weights/file"`. From the example above, the command would be `make preprocess args=lstm_gemm.onnx lstm_gemm_weights.onnx`.
`fLibrary/` holds the library files that recreate the model and run inference on it. Run `python3 modelParserONNX.py -f path/to/model/structure -w path/to/weights/file` to reconstruct the model.

4. **Compiling the library**

Expand Down

0 comments on commit 04a833d

Please sign in to comment.