Skip to content
Snippets Groups Projects
Commit 30e618ff authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] remove/don't generate tmp files in main folder

parent e653b671
No related branches found
No related tags found
No related merge requests found
...@@ -57,9 +57,9 @@ def test_absorb_transp_into_flatten(perm, shape, ishape, data_layout): ...@@ -57,9 +57,9 @@ def test_absorb_transp_into_flatten(perm, shape, ishape, data_layout):
model = model.transform(InferDataLayouts()) model = model.transform(InferDataLayouts())
model = model.transform(GiveUniqueNodeNames()) model = model.transform(GiveUniqueNodeNames())
model = model.transform(GiveReadableTensorNames()) model = model.transform(GiveReadableTensorNames())
model.save("test.onnx") # model.save("test.onnx")
model_transformed = model.transform(AbsorbTransposeIntoFlatten()) model_transformed = model.transform(AbsorbTransposeIntoFlatten())
model_transformed.save("test2.onnx") # model_transformed.save("test2.onnx")
# verify transformation # verify transformation
inp_values = np.random.uniform(low=-1, high=1, size=tuple(ishape)).astype( inp_values = np.random.uniform(low=-1, high=1, size=tuple(ishape)).astype(
......
# import os import os
import onnx import onnx
from finn.util.test import get_test_model_trained from finn.util.test import get_test_model_trained
import brevitas.onnx as bo import brevitas.onnx as bo
...@@ -57,4 +57,4 @@ def test_topk_insert(k): ...@@ -57,4 +57,4 @@ def test_topk_insert(k):
output_pysim_topk = output_pysim_topk.astype(np.int).flatten() output_pysim_topk = output_pysim_topk.astype(np.int).flatten()
assert np.array_equal(output_golden_topk, output_pysim_topk) assert np.array_equal(output_golden_topk, output_pysim_topk)
# os.remove(export_onnx_path) os.remove(export_onnx_path)
...@@ -61,4 +61,4 @@ def test_hls_random_mlp_maker(bitwidth): ...@@ -61,4 +61,4 @@ def test_hls_random_mlp_maker(bitwidth):
ret = create.hls_random_mlp_maker(layer_spec) ret = create.hls_random_mlp_maker(layer_spec)
assert len(ret.graph.node) == 5 assert len(ret.graph.node) == 5
ret.save("mlp-%s.onnx" % str(bitwidth)) # ret.save("mlp-%s.onnx" % str(bitwidth))
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