Skip to content
Snippets Groups Projects
Commit 677a7277 authored by François Bobot's avatar François Bobot
Browse files

[NN_prover] Add $CAISAR_ONNX_OUTPUT_DIR for specifying output directory

  - todo use the onnx-out-dir option
parent 1046b8f6
No related branches found
No related tags found
No related merge requests found
true true
ok ok
ok ok
Input name 0 1 files checked
...@@ -185,7 +185,8 @@ let create_new_nn env input_vars outputs : string = ...@@ -185,7 +185,8 @@ let create_new_nn env input_vars outputs : string =
assert ( assert (
IR.Shape.equal output.shape (IR.Shape.of_array [| List.length outputs |])); IR.Shape.equal output.shape (IR.Shape.of_array [| List.length outputs |]));
let nn = IR.create output in let nn = IR.create output in
let filename = Stdlib.Filename.temp_file "caisar" ".onnx" in let temp_dir = Sys.getenv "CAISAR_ONNX_OUTPUT_DIR" in
let filename = Stdlib.Filename.temp_file ?temp_dir "caisar" ".onnx" in
Onnx.Simple.write nn filename; Onnx.Simple.write nn filename;
filename filename
......
...@@ -4,6 +4,9 @@ Test verify on acasxu ...@@ -4,6 +4,9 @@ Test verify on acasxu
$ bin/pyrat.py --version $ bin/pyrat.py --version
PyRAT 1.1 PyRAT 1.1
$ mkdir out
$ export CAISAR_ONNX_OUTPUT_DIR=out
$ cat > file.mlw <<EOF $ cat > file.mlw <<EOF
> theory T > theory T
> use ieee_float.Float64 > use ieee_float.Float64
...@@ -1179,3 +1182,6 @@ Test verify on acasxu ...@@ -1179,3 +1182,6 @@ Test verify on acasxu
Goal run3'vc: Unknown () Goal run3'vc: Unknown ()
Goal test'vc: Unknown () Goal test'vc: Unknown ()
Goal P3: Unknown () Goal P3: Unknown ()
$ python3 bin/inspect_onnx.py
12 files checked
import onnx import onnx
import os import os
for file in os.listdir("out"): l = os.listdir("out")
for file in l:
m = onnx.load(os.path.join("out", file)) m = onnx.load(os.path.join("out", file))
if len(m.graph.initializer) >= 1:
initi = m.graph.initializer[0]
print(
"Initializer name {name} has data {val:3.3f}".format(
name=initi.name, val=initi.float_data[2]
)
)
if len(m.graph.input) >= 1:
print(
f"Input name {m.graph.input[0].name}"
)
onnx.checker.check_model(m) onnx.checker.check_model(m)
print(len(l),"files checked")
\ No newline at end of file
...@@ -26,4 +26,4 @@ Test verify ...@@ -26,4 +26,4 @@ Test verify
Data should be 0.135 Data should be 0.135
$ python3 bin/inspect_onnx.py $ python3 bin/inspect_onnx.py
Input name 0 1 files checked
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment