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

[Test] end2end bnn-pynq test bugfix

parent 1d37f8fd
No related branches found
No related tags found
No related merge requests found
......@@ -450,9 +450,7 @@ class TestEnd2End:
oname = parent_model.graph.output[0].name
sdp_node = parent_model.get_nodes_by_op_type("StreamingDataflowPartition")[0]
sdp_node = getCustomOp(sdp_node)
sdp_chkpt = get_checkpoint_name(topology, wbits, abits, "deploy")
load_test_checkpoint_or_skip(sdp_chkpt)
sdp_node.set_nodeattr("model", sdp_chkpt)
sdp_node.set_nodeattr("model", prev_chkpt_name)
ret = execute_onnx(parent_model, {iname: input_tensor_npy}, True)
y = ret[oname]
assert np.isclose(y, output_tensor_npy).all()
......
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