Skip to content

Commit 00d0e7b

Browse files
committed
fix ir_version
1 parent 67eb1c2 commit 00d0e7b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

_unittests/ut_translate_api/test_translate_builder.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,7 @@ def test_local_function(self):
227227
graph,
228228
opset_imports=[oh.make_opsetid("", 14), oh.make_opsetid(new_domain, 1)],
229229
functions=[linear_regression],
230+
ir_version=10,
230231
)
231232
tr = Translater(onnx_model, emitter=BuilderEmitter("mm"))
232233
code = tr.export(as_str=True)
@@ -260,7 +261,7 @@ def make_custom_LinearRegression(g: "GraphBuilder"):
260261
261262
262263
def mm() -> "ModelProto":
263-
g = GraphBuilder({'': 14, 'custom': 1}, ir_version=11)
264+
g = GraphBuilder({'': 14, 'custom': 1}, ir_version=10)
264265
g.make_tensor_input("X", TensorProto.FLOAT, ('', ''))
265266
g.make_tensor_input("A", TensorProto.FLOAT, ('', ''))
266267
g.make_tensor_input("B", TensorProto.FLOAT, ('', ''))

0 commit comments

Comments
 (0)