ir

Author: xadupre

_unittests/ut_translate_api/test_translate_builder.py

@@ -20,7 +20,7 @@ def setUp(self):
         self.maxDiff = None
 
     def test_exp(self):
-        onx = start(opset=19, ir_version=11).vin("X").Exp().rename("Y").vout().to_onnx()
+        onx = start(opset=19, ir_version=10).vin("X").Exp().rename("Y").vout().to_onnx()
         self.assertIsInstance(onx, ModelProto)
         self.assertIn("Exp", str(onx))
         ref = ReferenceEvaluator(onx)
@@ -39,7 +39,7 @@ def light_api(
             op.Identity(Y, outputs=["Y"])
             return Y
 
-        g = GraphBuilder({'': 19}, ir_version=11)
+        g = GraphBuilder({'': 19}, ir_version=10)
         g.make_tensor_input("X", TensorProto.FLOAT, ())
         light_api(g.op, "X")
         g.make_tensor_output("Y", TensorProto.FLOAT, ())
@@ -69,7 +69,7 @@ def light_api(
 
     def test_zdoc(self):
         onx = (
-            start(opset=19, ir_version=11)
+            start(opset=19, ir_version=10)
             .vin("X")
             .reshape((-1, 1))
             .Transpose(perm=[1, 0])
@@ -90,7 +90,7 @@ def light_api(
                 op.Identity(Y, outputs=["Y"])
                 return Y
 
-            g = GraphBuilder({'': 19}, ir_version=11)
+            g = GraphBuilder({'': 19}, ir_version=10)
             g.make_tensor_input("X", TensorProto.FLOAT, ())
             light_api(g.op, "X")
             g.make_tensor_output("Y", TensorProto.FLOAT, ())
@@ -119,7 +119,7 @@ def light_api(
         check_model(model)
 
     def test_exp_f(self):
-        onx = start(opset=19, ir_version=11).vin("X").Exp().rename("Y").vout().to_onnx()
+        onx = start(opset=19, ir_version=10).vin("X").Exp().rename("Y").vout().to_onnx()
         self.assertIsInstance(onx, ModelProto)
         self.assertIn("Exp", str(onx))
         ref = ReferenceEvaluator(onx)
@@ -142,7 +142,7 @@ def light_api(
 
 
         def mm() -> "ModelProto":
-            g = GraphBuilder({'': 19}, ir_version=11)
+            g = GraphBuilder({'': 19}, ir_version=10)
             g.make_tensor_input("X", TensorProto.FLOAT, ())
             light_api(g.op, "X")
             g.make_tensor_output("Y", TensorProto.FLOAT, ())

onnx_array_api/translate_api/builder_emitter.py

@@ -148,6 +148,8 @@ def _emit_node(self, **kwargs: Dict[str, Any]) -> List[str]:
         if kwargs.get("domain", "") != "":
             domain = kwargs["domain"]
             op_type = f"{domain}.{op_type}"
+        else:
+            domain = ""
         atts = kwargs.get("atts", {})
         args = []
         for k, v in atts.items():
@@ -158,9 +160,14 @@ def _emit_node(self, **kwargs: Dict[str, Any]) -> List[str]:
 
         outs = ", ".join(outputs)
         inps = ", ".join(inputs)
+        op_type = self._emit_node_type(op_type, domain)
+        sdomain = "" if not domain else f", domain={domain!r}"
         if args:
             sargs = ", ".join(args)
-            row = f"    {outs} = op.{op_type}({inps}, {sargs})"
+            row = f"    {outs} = op.{op_type}({inps}, {sargs}{sdomain})"
         else:
-            row = f"    {outs} = op.{op_type}({inps})"
+            row = f"    {outs} = op.{op_type}({inps}{sdomain})"
         return [row]
+
+    def _emit_node_type(self, op_type, domain):
+        return op_type