Skip to content

Commit bb7ee2e

Browse files
minor fix
1 parent 8fbb2a3 commit bb7ee2e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/onnx/quantization/test_weights_compression.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -372,7 +372,7 @@ def get_sequential_matmul_model() -> onnx.ModelProto:
372372
weights_data = np.arange(0, 16).reshape(4, 4).astype(np.float32)
373373
weights_data[-1, -1] = main_value
374374
weights_data = weights_data.T
375-
mb.add_matmul(x, shape=weights_data.shape, output=output if i == 4 else None, data=weights_data)
375+
x = mb.add_matmul(x, shape=weights_data.shape, output=output if i == 4 else None, data=weights_data)
376376

377377
return mb.build()
378378

0 commit comments

Comments
 (0)