Skip to content

Commit

Permalink
[Tests] Runtime writeable weights = 0 for RTL thresholding in tfc test
Browse files Browse the repository at this point in the history
  • Loading branch information
auphelia committed Mar 29, 2024
1 parent 038d306 commit b318693
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/end2end/test_end2end_bnn_pynq.py
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,8 @@ def fold_tfc(model):
inp_qnt_node = model.get_nodes_by_op_type("Thresholding_rtl")[0]
inp_qnt = getCustomOp(inp_qnt_node)
inp_qnt.set_nodeattr("PE", 49)
inp_qnt.set_nodeattr("runtime_writeable_weights", 1)
# TODO: update PYNQ driver to support runtime writeable weights for RTL Thresholding
# inp_qnt.set_nodeattr("runtime_writeable_weights", 1)
return model


Expand Down

0 comments on commit b318693

Please sign in to comment.