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

[Test] fix failing test_layer_streaming_maxpool_batch

parent 0b2d96fb
No related branches found
No related tags found
No related merge requests found
......@@ -7,6 +7,7 @@ from finn.core.modelwrapper import ModelWrapper
from finn.transformation.fpgadataflow.cleanup import CleanUp
from finn.transformation.fpgadataflow.codegen_npysim import CodeGen_npysim
from finn.transformation.fpgadataflow.compile import Compile
from finn.transformation.fpgadataflow.set_sim_mode import SetSimMode
def test_layer_streaming_maxpool_batch():
......@@ -112,6 +113,7 @@ def test_layer_streaming_maxpool_batch():
).reshape(2, 2, 4, 4)
print(input_tensor)
model = model.transform(SetSimMode("npysim"))
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