From b318693ff1c11e00b63434b11c276c66554bfd6e Mon Sep 17 00:00:00 2001 From: auphelia Date: Fri, 29 Mar 2024 14:11:26 +0000 Subject: [PATCH] [Tests] Runtime writeable weights = 0 for RTL thresholding in tfc test --- tests/end2end/test_end2end_bnn_pynq.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/end2end/test_end2end_bnn_pynq.py b/tests/end2end/test_end2end_bnn_pynq.py index 94134967fa..556ba1d187 100644 --- a/tests/end2end/test_end2end_bnn_pynq.py +++ b/tests/end2end/test_end2end_bnn_pynq.py @@ -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