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

[Test] use smaller PE/SIMD values in cnv-w1a1 conversion test

parent 338c651d
No related branches found
No related tags found
No related merge requests found
......@@ -89,8 +89,16 @@ def test_convert_to_hls_layers_cnv_w1a1():
inst.set_nodeattr("mem_mode", "decoupled")
mw = inst.get_nodeattr("MW")
mh = inst.get_nodeattr("MH")
inst.set_nodeattr("PE", mh)
inst.set_nodeattr("SIMD", mw)
if mh % 4 == 0:
pe = mh // 4
else:
pe = mh
inst.set_nodeattr("PE", pe)
if mw % 16 == 0:
simd = mw // 16
else:
simd = mw
inst.set_nodeattr("SIMD", simd)
model.save("cnv-pre-compile.onnx")
model = model.transform(CodeGen_npysim())
model = model.transform(Compile())
......
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