Skip to content

Commit 86abdd2

Browse files
committed
update qonnx sepconv test
1 parent da4f9e5 commit 86abdd2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/pytest/test_qonnx.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,12 +102,12 @@ def test_sep_conv(sep_conv_model, backend):
102102
model = sep_conv_model
103103
ishape = tuple(model.get_tensor_shape(model.graph.input[0].name))
104104
X = np.random.uniform(low=0, high=1, size=np.prod(ishape)).reshape(ishape)
105-
# X = (np.round(X * 2**16) * 2**-16).astype(np.float32)
105+
X = (np.round(X * 2**16) * 2**-16).astype(np.float32)
106106
idict = {model.graph.input[0].name: X}
107107
y_qonnx = oxe.execute_onnx(model, idict)[model.graph.output[0].name]
108108

109109
config = hls4ml.utils.config.config_from_onnx_model(
110-
model, granularity='name', backend=backend, default_precision='fixed<16,6>'
110+
model, granularity='name', backend=backend, default_precision='fixed<32,16>'
111111
)
112112

113113
hls_model = hls4ml.converters.convert_from_onnx_model(

0 commit comments

Comments
 (0)