ch

xadupre · xadupre · commit 9351aca03147 · 2025-02-18T12:16:37.000+01:00
diff --git a/CHANGELOGS.rst b/CHANGELOGS.rst
@@ -4,7 +4,7 @@ Change Logs
 0.3.1
 +++++
 
-* :pr:`94`: improves translation to GraphBuilder
+* :pr:`95`: improves translation to GraphBuilder
 
 0.3.0
 +++++
diff --git a/_unittests/ut_translate_api/test_translate_builder.py b/_unittests/ut_translate_api/test_translate_builder.py
@@ -8,7 +8,8 @@
 from onnx_array_api.ext_test_case import ExtTestCase
 from onnx_array_api.light_api import start
 from onnx_array_api.graph_api import GraphBuilder
-from onnx_array_api.translate_api import translate
+from onnx_array_api.translate_api import translate, Translater
+from onnx_array_api.translate_api.builder_emitter import BuilderEmitter
 
 
 OPSET_API = min(19, onnx_opset_version() - 1)
@@ -38,7 +39,7 @@ def light_api(
             op.Identity(Y, outputs=["Y"])
             return Y
 
-        g = GraphBuilder({'': 19})
+        g = GraphBuilder({'': 19}, ir_version=11)
         g.make_tensor_input("X", TensorProto.FLOAT, ())
         light_api(g.op, "X")
         g.make_tensor_output("Y", TensorProto.FLOAT, ())
@@ -89,7 +90,7 @@ def light_api(
                 op.Identity(Y, outputs=["Y"])
                 return Y
 
-            g = GraphBuilder({'': 19})
+            g = GraphBuilder({'': 19}, ir_version=11)
             g.make_tensor_input("X", TensorProto.FLOAT, ())
             light_api(g.op, "X")
             g.make_tensor_output("Y", TensorProto.FLOAT, ())
@@ -117,6 +118,62 @@ def light_api(
         self.assertNotEmpty(model)
         check_model(model)
 
+    def test_exp_f(self):
+        onx = start(opset=19).vin("X").Exp().rename("Y").vout().to_onnx()
+        self.assertIsInstance(onx, ModelProto)
+        self.assertIn("Exp", str(onx))
+        ref = ReferenceEvaluator(onx)
+        a = np.arange(10).astype(np.float32)
+        got = ref.run(None, {"X": a})[0]
+        self.assertEqualArray(np.exp(a), got)
+
+        tr = Translater(onx, emitter=BuilderEmitter("mm"))
+        code = tr.export(as_str=True)
+
+        expected = dedent(
+            """
+        def light_api(
+            op: "GraphBuilder",
+            X: "FLOAT[]",
+        ):
+            Y = op.Exp(X)
+            op.Identity(Y, outputs=["Y"])
+            return Y
+
+
+        def mm() -> "ModelProto":
+            g = GraphBuilder({'': 19}, ir_version=11)
+            g.make_tensor_input("X", TensorProto.FLOAT, ())
+            light_api(g.op, "X")
+            g.make_tensor_output("Y", TensorProto.FLOAT, ())
+            model = g.to_onnx()
+            return model
+
+
+        model = mm()
+        """
+        ).strip("\n")
+        self.assertEqual(expected, code.strip("\n"))
+
+        def light_api(
+            op: "GraphBuilder",
+            X: "FLOAT[]",  # noqa: F722
+        ):
+            Y = op.Exp(X)
+            op.Identity(Y, outputs=["Y"])
+            return Y
+
+        g2 = GraphBuilder({"": 19})
+        g2.make_tensor_input("X", TensorProto.FLOAT, ("A",))
+        light_api(g2.op, "X")
+        g2.make_tensor_output("Y", TensorProto.FLOAT, ("A",))
+        onx2 = g2.to_onnx()
+
+        ref = ReferenceEvaluator(onx2)
+        a = np.arange(10).astype(np.float32)
+        got = ref.run(None, {"X": a})[0]
+        self.assertEqualArray(np.exp(a), got)
+
 
 if __name__ == "__main__":
     unittest.main(verbosity=2)
