From ea3730cb5077f7b9138ff506c49ac1e742c97201 Mon Sep 17 00:00:00 2001
From: Dmitry Kurtaev <dmitry.kurtaev@gmail.com>
Date: Wed, 22 Nov 2023 20:04:27 +0300
Subject: [PATCH] Test data for graph simplifier

---
 testdata/dnn/onnx/data/input_bias_gelu.npy       | Bin 0 -> 152 bytes
 testdata/dnn/onnx/data/output_bias_gelu.npy      | Bin 0 -> 152 bytes
 .../onnx/generate_onnx_models_with_onnxscript.py |  15 +++++++++++++++
 testdata/dnn/onnx/models/bias_gelu.onnx          | Bin 0 -> 668 bytes
 4 files changed, 15 insertions(+)
 create mode 100644 testdata/dnn/onnx/data/input_bias_gelu.npy
 create mode 100644 testdata/dnn/onnx/data/output_bias_gelu.npy
 create mode 100644 testdata/dnn/onnx/models/bias_gelu.onnx

diff --git a/testdata/dnn/onnx/data/input_bias_gelu.npy b/testdata/dnn/onnx/data/input_bias_gelu.npy
new file mode 100644
index 0000000000000000000000000000000000000000..7d60fe1d6c9500b1455622e3f17e17dd2931bf40
GIT binary patch
literal 152
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I$-ItoTQ3dWi`3bhL40WR)(9{Xit=JuQYr0mOUx$Qk(-mnXdRJ8{HP8lP8

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/output_bias_gelu.npy b/testdata/dnn/onnx/data/output_bias_gelu.npy
new file mode 100644
index 0000000000000000000000000000000000000000..89762b94c7eea8158f52041ef771af9aecf26265
GIT binary patch
literal 152
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I$-ItoTQ3dWi`3bhL40WO8iZ*~)AMccb8DcM)mf3cf!lGi??Ue_J~YT_jS

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/generate_onnx_models_with_onnxscript.py b/testdata/dnn/onnx/generate_onnx_models_with_onnxscript.py
index c8c9e6122..8fadd6896 100644
--- a/testdata/dnn/onnx/generate_onnx_models_with_onnxscript.py
+++ b/testdata/dnn/onnx/generate_onnx_models_with_onnxscript.py
@@ -124,3 +124,18 @@ def two_resizes_with_shared_subgraphs(x: ost.FLOAT["batch", 1, "height", "width"
     return opset11.Add(resized_y, resized_z)
 
 make_model_and_data(two_resizes_with_shared_subgraphs, np.random.rand(1, 1, 4, 5).astype(np.float32), np.random.rand(1, 1, 3, 2).astype(np.float32), np.random.rand(1, 1, 2, 1).astype(np.float32))
+
+@ost.script()
+def bias_gelu(x: ost.FLOAT[1, 2, 3]) -> ost.FLOAT[1, 2, 3]:
+    bias = op.Constant(value=onnx.helper.make_tensor("", onnx.TensorProto.FLOAT, [3], np.array([0.1, 0.3, 0.2], dtype=np.float32)))
+    add1 = op.Add(x, bias)
+    tmp = op.Constant(value=onnx.helper.make_tensor("", onnx.TensorProto.FLOAT, [], np.array([np.sqrt(2)], dtype=np.float32)))
+    div = op.Div(add1, tmp)
+    erf = op.Erf(div)
+    tmp_0 = op.Constant(value=onnx.helper.make_tensor("", onnx.TensorProto.FLOAT, [], np.array([1], dtype=np.float32)))
+    add2 = op.Add(erf, tmp_0)
+    mul = op.Mul(add1, add2)
+    tmp_1 = op.Constant(value=onnx.helper.make_tensor("", onnx.TensorProto.FLOAT, [], np.array([0.5], dtype=np.float32)))
+    return op.Mul(mul, tmp_1)
+
+make_model_and_data(bias_gelu, np.random.rand(1, 2, 3).astype(np.float32))
diff --git a/testdata/dnn/onnx/models/bias_gelu.onnx b/testdata/dnn/onnx/models/bias_gelu.onnx
new file mode 100644
index 0000000000000000000000000000000000000000..2d23e8f495589731317a7f8c19ccf86884b11e41
GIT binary patch
literal 668
zcmaKp%TB^j5Qa}V2U8{(oADN_CZz62jI`d6M2x6AeE~O3NsED$YD!C7`U);_<xBV$
z<!O8Y=k$Uss*AIkng9D|{-f49?^S4{=ntKVUI|);dh;AiA}5HN4X92XKlYj%D%TXF
zK2MU%kJ+p{Pwu-t_Nj;t109rCNW~}|*EI=Y8vNRIsdtDx8oyA*T7qsBP#~oE7GGBd
zY8O%hND*>(I3?w=!Ec9CVr_z~Aos!nl}`+Q7Y?X=gjJfhTU0y!(}c0Ng~>ih1Z1jo
zWs9_D#mam64ssI<`8f8edv5TD*#C8tN4EcQV+*$fkciC9L<qeo4g;G)CBj8c#hQ%X
ze)9bIahuXa1Wi@24i#0QSE0xI4OkMFO^h)A1pt-QLaIi3c1`L0YL@c~2pVUjoh;uH
hZ`q>AN+D}SkOd}lv`UAGLt&J*a*l~Prze1E>pvsyero^#

literal 0
HcmV?d00001