From 3de79cfeee3160205e686e910a329b2a8b52061c Mon Sep 17 00:00:00 2001
From: Joel Schlosser <jbschlosser@fb.com>
Date: Sun, 13 Feb 2022 19:32:11 -0800
Subject: [PATCH] Mobile upgrader: add GELU test modules

Test Plan: This is test code

Reviewed By: tugsbayasgalan

Differential Revision: D33858429

fbshipit-source-id: f5bc687ad81125f89cb281f9aa51dc5ac74d6954
(cherry picked from commit 202dff607e72d538c8b3fcd44382d11a401e52c5)
---
 test/jit/fixtures/test_versioned_gelu_out_v9.ptl | Bin 0 -> 3367 bytes
 test/jit/fixtures/test_versioned_gelu_v9.ptl     | Bin 0 -> 3083 bytes
 test/jit/fixtures_srcs/fixtures_src.py           |  15 +++++++++++++++
 test/jit/fixtures_srcs/generate_models.py        |   2 ++
 4 files changed, 17 insertions(+)
 create mode 100644 test/jit/fixtures/test_versioned_gelu_out_v9.ptl
 create mode 100644 test/jit/fixtures/test_versioned_gelu_v9.ptl

diff --git a/test/jit/fixtures/test_versioned_gelu_out_v9.ptl b/test/jit/fixtures/test_versioned_gelu_out_v9.ptl
new file mode 100644
index 0000000000000000000000000000000000000000..208ae5100757245d5935bef5b5b3c69ef5faef35
GIT binary patch
literal 3367
zcmb_e2~ZPP7+%5=2p)hJwT?PrTENBx0WF8x0)lpeBt?WBk6Aa#CRwt{W-l%WK`lY8
zwWHS7YU@Fz%AnHLQf8|50O<wNR%nYlI*La{TXl?#ozCd=y-kP-Y6d&LnVo&_fB*a6
ze|-P{Kdm}22th(akw-=nqC$9r<FOKg<w)8_7%&q-2^cK^QmTq5<5^r~XK90=Cs>TM
z8EIuPN855TUqY4)f2@?7X>f-Pw-WL+xq&Dtu_h)bDsj%7pfuuoLPygUrJlCR6>^<G
zQU-b2O4kIl(gt{hw)2B~7%NZ&#+Tb6-CoX{X&a^^Z8%%*!vQvpIz2d@rpJv&B2mTD
ztlq3L>Qq)*M^c1Ju$x%i0E~i6Cc>8HA&CBf7gHQ?hRq|^PT{<fX07mE_cS3P6>Qsa
zy#+TB7)wwD&OyDrh;GOF65XEgh~`NvfpI+R;u7p&O5j!?u!MwBagn`Ol?pjLBTt(#
zD>*qiDJ3OkmO?Jr$TCde2q>ml$LY-!&B0ZrRxJ%J-xA}j4nPq2X`hIP82@-M;5@Fh
zTd1duf3-x9VPXI<ObOkqEGBuSku2i{mf$ds)pI^8SpncF97N=N07#9%7pBS>Bz|>e
z913v^a7o%on~4cnhmHqpjN~w*Kv9?u1fvvZ4VW2#ijm|AIwnvF9k6dl;_20Z6MoPE
zV?jbZnh{zg!(T!`u&6w?sGuz>V2jGf7N3>UUY?UV=V`L#(Y&&1J}qv0b;LR6xQW&U
z$L~#Dp1w`-f;z5`y0N9PY5&x7uXi^!bs5485eq6bxB3_U*w`9NrfxQ0&UL2OR~)=@
z_e|yOWn%_x(&{p2U)6!=@W@g22WP&}kGzO4KJon|XYizroei7obO+)uUfX}=kB*H+
zSKDu2I<$SS=Fc3eB`G*m@m6<y{e{jwpX3&u*xUPNTd5*zouwgj!?dk)bK9KJD;I^J
zAJ%vNbgHf6>~Z1w*5+R4&Yy43Z`}~D>^XGbA$xz-S8Wrz|N6Z$Jrx}|S_Ol#sC(x0
zE)a-U+-Q7!FqR<Go_R2o20|y8FcHKr4l^u^wR7mT2@y5P%kNB?R-T;yP4qI!wo~;V
zwRdU5>U)D)#`cXaIMuMTs>HG{xp?<YzT)2b{=9q5YkC@IOZ$b%0|)Q+=fC^U;Wuss
zO21qB%ckp>_C)s_qgz&tnmp^m8YXeY=*|N>l>c5X?h#byQ$ASRn8hSDT+}t}uB(Wy
zi@6n>LciC4ykkX7(Ql{!&bqZW!<o^%^{P2aksI1o`dr2O((MtBtdptX(#oifCtfSr
z-8cK>^{dxXHid7w@cGodQB~PdF>TEmO|5$hJ8woDwj7#x|LyGkM~>~S*=CWEwg$Q*
z{p+I->fTt@^6@)!mwx%#l}l6Fe>i(h3d5AY>s#{x2vfG-q7jG8^dBZ0$Ky7hb5)KR
z!*se=8dOW-#T2%x!iqp6L*V!18WxMh`cp?&&J!?=UE0J9N1JMiT8id^OFfxCoGPY#
zd=*oI9s&y#gT$dDfJ_p2hZXcZsPh~X5Ql~U+M)tYes&%sQ3t3cwP-M)%*)qk^O-<E
zKx$EFf37BD9uuUN46YYy3bUDD*IEMhHCiS_jeuQqi0e=wAm(QmFjA4qJJ2vdpzQ>U
zgUZY?p=bnH<2>kuX%z%Za~MTh2qp{#RD_YK18Py3hXKA3m~a$|1TO_<1klpq5~v4O
zh{0w?=G8bv)nTvZm{IW#bR^(s2@}cjpzm^I1;LC~OB}qY3C@}XD*;f#M2Pmm7Rf}4
zQ;wjFOq4^MLt)GqbRLlOEP}y|b!lZFjJQC7HxEt;1T!wrQEQGw0qBu3X1v2ZMhr>;
zVdH3)ncy(PePFLtGzauB5V&Y_q*(?y&F<|Q2VfCFtLhE9lL{gNzbAuMj=by-wC-0#
zPG-_@@amO1;xtJHllj+Oro|8$@YAXT;Lr0W^nM*A!401Q+uT#W5Qkpi-F;(xghv2H
z{9G{j<Zuam*e85Ch>yTX082jt|5pnidb@XFT;g;4E|8qh`u2xCYp7ejD`K3F9@(Id
z_|?O!C=ZqD4cS;9sXhLsde_#WLcQT9_Yt}tR3bmBy0NcShlD8H=OzbE_eg+Q2*=?6
khzM*1iGt+{3~)}|YKO%Z44BngbucWm;EWW3{4U=A0L3aNx&QzG

literal 0
HcmV?d00001

diff --git a/test/jit/fixtures/test_versioned_gelu_v9.ptl b/test/jit/fixtures/test_versioned_gelu_v9.ptl
new file mode 100644
index 0000000000000000000000000000000000000000..5e4ffb20f823518a3730844f81fc515198e0b536
GIT binary patch
literal 3083
zcmb_eeNa?Y6n`wRyMQ90sErD;z$g>!0|9|0G)+)hg_W;f<;?5t?%mx7yZiQitso;}
z>LR8w+N3s3)F_tHSY~P2#Ps1ujfF<0<s>;y)6^d}=9oHSS?9i8a2H)i%{z1F-E;4`
zzjN=ozjMwr86%?zLaQZiYb23JNJx}uFA_wSb0aHlM^2gc7L%w+5*X4Wa8}uZ1e$f*
zIDLc2xy$F0#54ETT67K$Y`7U0(iwDC<n_8{<j>GEq9aFdV=TzbagBNl=h9{A%rfh=
z>I^OZ0qx?fcnjx|hSq37b|PA8_TYI>v*h61w3&4?LUV`)EZWQ#*ln;dHXE8jN}OPE
zkTx^v;>@fQk+R1wFjlx2R<a|vVR%CE2R7^!!5QvhSdWvDY@Fc2b_2tloMKq^FqTHf
zj%Wcn5hLPr1x*GV8#)>A#4xnPx)3c&f?rGazz{Jm0Q9o>Rs7Sz(uyn{-lH?*&dkrx
z&nqe_nwh1O-E0#riwMIsm_>`j$%$B6VKQp8&70FMw?z;H?xwJOND0peD<d&_Powhz
z-oH&{p=sp-Xj+fmt8ZW>y^U>>WC4k^C|JZ0DYh1L>W3IP9|&1r9Lp1#t5!^>Fo}Vf
ztebV)`G|GYc#vsUq;0a(Nt@v%mN9~rb^xPzRi$j^BlY<9d}>BVD`?`5Z$A#A;r4J?
zL}<fB17-<1d<ha?f`m(uA(se|^q%JOxlu}}xT;DyRqm*P>q*HMyJxnf?QeXmC(b<Q
zt78Mj3EJF!WBJ)B=Q^T0`dU+3TP^WDdo%WwBraa+7~FZ*OAn+}W!GFRB-VX6^?b#t
z^mTuZzHw-MeD=`^yDAoByqwlFaoc5G#rpSt-SFf@^4ip~UE9c`|DNmFx%Lx!%?r!6
zYd4{OsiJs!+^(!&V*2ZDoOrE9&h6VDli9OjaNohgDYGtQeRg3Fy<|)Btd(6m=tooE
zpFq`|?avxG%OzvKt?{NWJODmR46x?Teyb<c57lpaD>BP=S|OVXJXqIR%6*ZO;G0r0
z;lRzv&Up<Niuglce!FwWk*@6RmtN`Ku3kBB&Bnf!lhPs*J1Y&n%ezZ@w{|>HuzuqA
zsjHq}c-VF1%%YbDOJXN)dSkEQ$CIaz9bDZzc-ZoB*ZGq1e?M1nRr8a#`+Dle(^kz5
zX4ac|dAm<s(*E;9*Ne}-t=34tqk7MQrL}(*(mBU7zMkCn%?Bk{H~&_4rEVX3y5JAJ
zZOgk&pC3!=G_)+}su^?q<flK^oa($Z`=g_0PWPq@kH1>DBq}dyH>KIMrf>dtrIXZ`
zE8oG6jha@l_yjnXxNA736Di>x>lP)(Es6e^oOHLG+@{J@qQFJQJOzb-z&(5pNFyE#
zXVly*Ash*Qo73)QGq^&usf-$GAs8JFpnJ&z`a5I+JtSZp;5;#%8U=tXI6;st5+q5H
zkC;xWVXmgOs=BO_R~aLWs&*<GX698_nX36nm>}9IeEG#y^<{jNQH5m{$W<mj+DKq2
z5~S*~T3)S)l8=gk3C@EA2J)B4Yp7U|842oz0T)VMOMy)AF~*4YwSHv_97TB;J~q>*
zn8(FR<l{gZ4Y~rdv!VorM`V{Hew0z=lN2M2V3%D89+V%goDtUuK3*A$$Z6xp_>?gQ
z;}fVkfb_q;$v@&hk`>t)*$F=Zj8jH@Vx_O$5l;azi6(xm&yk>bwHCQWPT<G+9C+R5
zqcqB0AcKSFlN|Am7|?VCme=|~MWI~9%`2;~0Tk}xDVI!?hNE1dewNQIzE`#dGtWQH
zi^U<=-(|1EkBz&@7=b^-|3tynk;Drj4QKoN_(K{A9$b7<LVy$C5MjWF%B>&lR^AQH
z$csY2;(@ziz#gc2jkFw`vg$)DmqScFOv|_O|43H`XR!2;GrSIg9`+1Dsd6M@FhSEo
z5XV6>34<7%2S)-1(=0gzupRz5!T<(XZZfLXS%E#tk=et$ajfAs^xdkE7s@iuPS9YV
Yva~zSFXPxX8KZGFggqJp>-%;811cG#$N&HU

literal 0
HcmV?d00001

diff --git a/test/jit/fixtures_srcs/fixtures_src.py b/test/jit/fixtures_srcs/fixtures_src.py
index 545152b6a3a..dff23702311 100644
--- a/test/jit/fixtures_srcs/fixtures_src.py
+++ b/test/jit/fixtures_srcs/fixtures_src.py
@@ -42,3 +42,18 @@ def __init__(self):
 
     def forward(self, a: Union[int, float, complex], b: Union[int, float, complex], out: torch.Tensor):
         return torch.logspace(a, b, out=out)
+
+class TestVersionedGeluV9(torch.nn.Module):
+    def __init__(self):
+        super().__init__()
+
+    def forward(self, x):
+        return torch._C._nn.gelu(x)
+
+class TestVersionedGeluOutV9(torch.nn.Module):
+    def __init__(self):
+        super().__init__()
+
+    def forward(self, x):
+        out = torch.zeros_like(x)
+        return torch._C._nn.gelu(x, out=out)
diff --git a/test/jit/fixtures_srcs/generate_models.py b/test/jit/fixtures_srcs/generate_models.py
index 36b6b5ffe68..980e7dd0324 100644
--- a/test/jit/fixtures_srcs/generate_models.py
+++ b/test/jit/fixtures_srcs/generate_models.py
@@ -94,6 +94,8 @@ def div_Tensor_0_3(self: Tensor, other: Tensor) -> Tensor:
     TestVersionedLinspaceOutV7(): "aten::linspace.out",
     TestVersionedLogspaceV8(): "aten::logspace",
     TestVersionedLogspaceOutV8(): "aten::logspace.out",
+    TestVersionedGeluV9(): "aten::gelu",
+    TestVersionedGeluOutV9(): "aten::gelu.out",
 }
 
 """