From c397223352f7510e208fb83b62282e5b954f9638 Mon Sep 17 00:00:00 2001 From: fengyuentau Date: Thu, 15 Aug 2024 15:57:34 +0800 Subject: [PATCH] add model and data --- testdata/dnn/onnx/data/input_matmul_bcast.npy | Bin 0 -> 4224 bytes testdata/dnn/onnx/data/output_matmul_bcast.npy | Bin 0 -> 2176 bytes .../onnx/generate_onnx_models_with_onnxscript.py | 10 ++++++++++ testdata/dnn/onnx/models/matmul_bcast.onnx | Bin 0 -> 704 bytes 4 files changed, 10 insertions(+) create mode 100644 testdata/dnn/onnx/data/input_matmul_bcast.npy create mode 100644 testdata/dnn/onnx/data/output_matmul_bcast.npy create mode 100644 testdata/dnn/onnx/models/matmul_bcast.onnx diff --git a/testdata/dnn/onnx/data/input_matmul_bcast.npy b/testdata/dnn/onnx/data/input_matmul_bcast.npy new file mode 100644 index 0000000000000000000000000000000000000000..1220926e8ccc6006fa188df19da88562248a3363 GIT binary patch literal 4224 zcmbVP`9Ib9|0OPAP)YW!Huqs_J3 zY|b^(=57uP|0_Ho)GsVNFzEmM6#?5Lf`0iCAptvserbIRb0cljU$ihV(mtsD|ADkr z^#FO>q0F*}=R@5X2T6D+8t<3InAKMF(0(H-n%GVvgYt114Fu=QCg88qLB0=cB|B@I znUJr~*t*CR*!TM(EV3QO$2-)CRR4Ke-s+9kc^;7QliI}xHDTEX>5E{TK9p4Ec){^#*+{cSuX6#_rkh{G!* zNm^SxOVHf$o~AokL(7_28vCJvN_ama&-IUzuEM=IV6Gw1_M3tk(b~}a`2yWwTTY(D z$?(Y89c1MuNf2^!V3(|mAnRBbmGC$TF(s5nZhw776z-WqTJlLomP z{($jgN^mOCfqdq>60iK_D9-rO;|I8GSMn=NK50lxyPNC+(zJn@$^w3wH1D#ePrcRJ z2#}m(hCkD#VTYk4+xNnhCLJ@Sy1qhk$XbaPEmuxUMw94{=4@DcG7V%^v%$^O2VSRK zWEE!wVcV~ax(1x+n*}PE-Otb;rk4m>E28J;X@b{f6GUEo1>Q2bN)vnx*<0q@S)oZ5 zyJoC`3CSIz4jgHy`=KZ}=vVZeqqW4;DgvBQ9d2o!qxV#N@X_cS+F@ct ze^ZRXNp26J=3rXSiYpktjn7_LvnE}3;O9v6+F3Sso# za0Pv6e-?~gh3MkuM4C?pKu`BPO1ca2tyvU4T?Ba6|1qm|_!27n%tVhroT!PeB(0uR zOpB{WuyS!4={R+9Nnv4HD& z*t}F6miTZ<`ce}NCK|+USpum&q>IAwYO33o2%J&}Sk0JYzQzI!yS;(^V)>HKSUe>v zv&C^`*Hfx=MH7~K_!Et}-DJ-?KFKVPgn$=Lxa~_Jvruk^K&OVo({`_=(^hVQ(<%E& zYSlrUx3Y#&Xe+_7U^g^ayBog+O5izPMV^E(&Q8l+3tKmwL#LNfP%Q6^if;N)0Z*8( zO+_@P`a3opu}0|;NvvDr05YkqWKKmKwznx_+|W-N)golwIE_#^GMz5U3dg3&$Fymx z8k(fa*zYrr!*>^13=}IMGvXb<_V8BXrpthY7QhUO}Y@w~h$ zN%+r?I54}6|D&W4`aAzX*K_rlbN(j0we$mpS0#ZMxl{0UnNZ;{AJWuBq75$+xC z#Oj7%%(1j3^V}=gBC`^741a(gvKmme&>sWE&Cu4M9K&DSpxGBzllw-+5ayUizR0(* z+ZL$egHMKJGDIKFZ>B7E-GlwYWn%msv8T1!otdL-thhcvAGKd0^7d0y%t{@;$O?#p zQxaJC=F*^*uqpWq~zFzR9$5fR?U)@gYCk4rSP=O`L) z(17Z?7=UM@5R|-u`WrSv;7D%0+|(N;=Y&4aS2+fSJ!6>PAWJ82&S51+uEK1V9z2hph|Kb*=9hx!mp(-RzwBz#LG^n(bf>S+WV6-`vMs(%j>&{Vj z)&4U49DW{*8;hVJbC~9|nvl`Lb4=mHZK#ZLrWxC~FjQ*{J6rBE(QX$Z-BlV(+$Wfw zkNyJfG84AhelDzb_>D1foCzZ%)nucokggcG3D(nNpecDC+4T2G`pRP^EbPjLySi+_K=KI=&$s<3?A@gfBaYg1p`IBOqBXNR-$c;2 zUK4BE)6u|MLSW|>0J}^|(7r*H27K>ls{8%%u(lj@j~bx-Mq7}ocm}(gZ1LwB25q*# zh9fy?_#pHIntJ3yl2#-AcV#kUZqq;A z!GBL|2Xno8R6ZlbY-Sh_S53n{DGpDn^%Y6sl;A?Oe{kuFNYb`l2d))dhV_>nfir3a zU#m1=Jm&!MiCIM}F3-ZwN;hCGw~`4L1ksXp?6V}m_`XP#)Dz;WtZL}^&ju!3t}w$P zV#M+J71UoQ!ZWM&#Jo-p>(INy?#S+Ew4*%+M2pMW=Nv`!Ui}*HAKglpKKTdz-4)?~IUHWxKYHjcB}uy$w=gZ1 zMYvx^3KsAyiPGF45-)ZOtxN^P>*pk!n{Em7F9YNX;~AcB0$v}^28qd=(A$NWHdq7O zm8T%7b{VM2Ujwd*ETl`#gz)5i_|f=;9V%H)Z4C_Y^VPfTzT;ASo!hF=Zt)3+7G9(M zMUimz&K$bN>mDQVeh$%{_{x5ZOrZ})X0!YJpVD>iSs13HgT^1c$)iV$u<>0qR%=V+ zjbraIrOyv9Yj6b*pD$(lQ!l|qn+p1A=kHkSRRg_W38V??33h0Y$X#gHKx!E63v)Nw5!R(3(qtn?Py zciWl!@21h=*|u2OIYfT5C}oYBH)8JNYHX4a(5s2R`ooC<+x0U*T_u=|uDeO=nvb&U zt;^`J!g*Rpi?RMoA?#5;iL+LTgP6&A5~g5)Z|$RTeX9uLpB%&7^8W@dT_S=*qg}tY zK_-rGk0s&d*Jx*4HrC9&NKPo`fos1jSWC>tJ9ATTmXRCW`;>zfO6A0ke}eQER*{sO zI&fh2D5LKy1oLqPQqa4EhOM_J61&QA#!xp&4!R37-W3uvg=|>h_X=9o9?>sJNbmV2 zAlE@e;GX9}jVG;%&wv!v$&R4UJa7E5_bDs0u>sRmb1)&&2~-`AkxI=hJSOr78QH|e z-y&B)-6dVH;rP*}SrxXzw*u0hvldEQ&d}bvKUp)0M!YI_33p3x0OKDObnEkeh8KZ{p>%CYJ36ow-!S4wi0ZN zio>mqSt!&qVnwnC*k|v|A>zR;nEWyyi^b~EmI$zana zhub3VGA%EkK)84w`M63NqE;MZEI(ZW*?bPx5BGrAzq>&tDvaUSpMkN|2s%*y1Y&f5 zO$0Ds`nc44?IT=;}r$lC5}=n<1lhBbQADO=Ylld29LZi;KDc|UGTAt zmij*?3C?PE;+I6Iq1iO_uFD6F!D!%}(FA?G{@Av-~K?e7}uO1;zCDs93Z+@)|x{W#H+m3ihvkYoKeikVphe2o6861^!)W zUir5;wATGdQZ@33MvDb{aL>^@)E=T`pW^&uJlG!k0;g5Gfyql_L35TVicM&NoqiPD zmzg3S<8@Sp83OsRC|vg~fb1;N2Vq|WzP}a%QKz@?Lo#O2SUpc@^XewW4vCFP!V!eSNw=Tyd5*7Q1X&LMMbyt2;XRIC!gXWUCuFuT$4z zt2=vo`+9Rd4tsh#I{nXY<+}Sg{qjC8+~ZEa_Nw*kwAgF3*c&vo*k{=P7n1#U@2KLB zo8-IaJRFkm$TXW6~>yTQIk&vbnymLOuTQ21($7LN&7DHJq($lm$jVxY$eC3VsnF9$p&AqhbjI#K?BsNl1pn*6su zkpybLhi;30M(%((*>TVje~w?|JNi3OyQ*X`O%272nRMb@B8x&@UtnurEHR5+0A&j( z1T0lyGF(GYsiPggY!>El9@?T#%1IC|(}3KBGQLfmM`c;DK51a*K}q2~B0M<{WW%ly z+4s%3B|aa#*M*a!ZOW9r@&&0`k`6K(MUAz8SVH9UWgPK^hNwMq9VXhN8I|rEr0C*0 zZ1R2z2Rc1yh0$_g=a)m!ha`{*%7rm$JJRyuHKV$HihMWfCK@_1pi}=2v{)%*n^!T= z-+XcCc{XgklaA6Kq6za;7w11%M5Vd!sotC?^ajzyv(S8kE&Q~ZHvW6 zRk6b^2swN0`Laqu*ta2^Nng=GjzzSQ&t6<&<|xFGS}M(KXN<6Lp)GyS(}WxGuSmP7 zCNuT822QQFBQbZ&na$0UI6XW|dM4V)+h;udF_DGkhgE64MG4bYw;GcZ^x;5NCamW7 zGC~1`w7|WcZl^p@ceQ{Y84?)K6h}`vmx1Ytr*zQ!JSfiaVM05Jf8_Fc;F#-^vzMYU zu3HkE1&wHb2iN5S32M6M;7fP6F=R;?io zqMw<{%kB_m{|Qd#1~XP)GvHQW8ge)2!OAzkKImPeGEus4c1|{x(v?U5%v$VRQ$TAA zg*g}7o}+SOBzt2mE*%g?l%R(5=f|&g_xc9gM zGBQi?TkCyTqtJ@#yX+vFn+n}03YozVMqtJZ0@tbx+}-jLO^@C&{{56Z=(n8%sq>md z*Ik&O+Tw!Y^^){jB$o`jKc;~WO60F?!8HF5E0|Q(6+9dwC>67tj7aXGN-~oWJ(!Fi z=P6_P$F+D+(*v@sa!^-E5y#s1;mh`D6#hO3j;2V$>vLDZz`_FTdo4&#Q9RB*4#9#L zJ0jvG1;!CE&|NZ3`%R|FwYBjy;7BX&uia(b^K1ZOXWel0Nf|Boy-F6wiwG7B&lkkp zafgc?ujp5k3Fy=khobgB8CK>oNY|Mns$M)C(|?4m2_yUiTT&otSvXS!JMf*CBiiO) zA^HPx5T{lHzPgGyPg)HX{44p5UnJ0>%K$Bp@kpsbELHWqLX{I|$W&83$s3WxywNCZ z*g3^i+J!=M4;yuk1@kS{%0V5~1gx7JylWJTimzg@Cv+J;=(ED*{?&ARmmIl&xBvsq zufg7XJgR*ufsR!eg6G;r9O;vd^n$`1$VifdfC^ELzWX4oh-Aq0bw@~hw~4fG+J>4H z4^Z)=1={fXs7B;xs%-O_KKK_0i0CS~7G*`gzbq&ChdkXDCI)G}z0|@`54nHp;dIGu z*c2Dc+#7Fz<}`J+27bmTe?Dv&{B^Gwj@A8%3afpjqvJRDJb_#(?YzdqeN`$)V zXtF+RmK>@R1D~Ng*cV#InC>xw{A?CzWNVVJ9Ti00;2=6a*X1W#j?-9^8u0te1wt|t zskQSCTxFZcv}6F3e$&RASvRoidKys`5#@N4OhZm-0-S!4ORJ{J>444%b?BNSXc~Kj zB0~oJ=Q(50=^TM;t#YC9yE*VjLokHZKv{zuAprNovegk#K4plu|JjQdcr1tw{tCQ( zqol{s1fAj%(8W9q$|T(AO5YI-7E9-Q|MWxq-JwP)4qHi7-2xzuWl)bibd`rqOr<8~ zMMc7<>LBVU&f~i*F~@Hu)-+(E7*>htgF^2*=yYo&tRX{)mjE#Heui?97jf$KcI*xk zCu?WoX_w_AcrtDa8Cy%i=-oJs4rSs4FFx^~tVA1aH}vTf6O`K@!rq&FVpZP+125)M z)-x7IT{;v46{^TWwGrB~CJk*D9K=~8BfM2wMS@#5Gt1LOsrUUir1Xjtz;r#Hy$}H% z8I4%|<}j`_t>A};wLx;`B|01#4_-ZO#;$c{q}EZ42CIof#J~CIV-tn8wZ_0MQ$!#4 ugZTFuP1^8xDll#{#J0Z#*|#;p^vDmK%kHKHKLxNz$%pQeXvVOotKlEh3mfPF 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 35cf57eb5..7240e044c 100644 --- a/testdata/dnn/onnx/generate_onnx_models_with_onnxscript.py +++ b/testdata/dnn/onnx/generate_onnx_models_with_onnxscript.py @@ -380,3 +380,13 @@ def clip_div_shared_constant(x: ost.FLOAT[1, 8, 12, 10]) -> ost.FLOAT[1, 8, 12, clip = op.Clip(div, Constant_output_0, Constant_1_output_0) return clip make_model_and_data(clip_div_shared_constant, np.random.rand(1, 8, 12, 10).astype(np.float32)) + +''' Subgraph [Input] -> MatMul -> [Outpunt] +''' + +B = np.random.randn(16, 8).astype(np.float32) + +@ost.script() +def matmul_bcast(x: ost.FLOAT[64, 1, 16]) -> ost.FLOAT[64, 1, 8]: + return op.MatMul(x, op.Constant(value=onnx.numpy_helper.from_array(B))) +make_model_and_data(matmul_bcast, np.random.randn(64, 1, 16).astype(np.float32)) diff --git a/testdata/dnn/onnx/models/matmul_bcast.onnx b/testdata/dnn/onnx/models/matmul_bcast.onnx new file mode 100644 index 0000000000000000000000000000000000000000..978d332335fcb9c7c7ea837f54d94f31ddbf1ec2 GIT binary patch literal 704 zcmZvaTS(Jk7>Aur`}b$)+U6n8E|XF_s3khw_lk+u%E*%Nglp`-8?Nwk%bcNZH=g7lx3FpMrke2JGFi(~a zi^eiotsRHgqEyn^DI?wI$B-ZTT^^IfVpxy3 z`9=z#N@7tS6rm1-J*iwbkP5|Va?i6GLGuz4MOCMsGh?y_^qu7&W;Y@tdJRk z^usu`K!o{zS@7kF4kn7ziSvhBL?xKS4GgBhhUU|Ze0nPx4xJ}eYx0z0Ybe%yg8BCG z5_P;rA+X`FvL)(ljY6D}DN(6YT&xC%L$fIcheg{`QC1jM*cm?bKNJ7CpyW<}KRM7W Qt4wW4R;XwuZNujN0o>3akN^Mx literal 0 HcmV?d00001