test_cdist/0000755000175400017540000000000014732212054013117 5ustar jenkinsjenkinstest_cdist/mint_cdist_binary_case1.py0000755000175400017540000000161014732212054020247 0ustar jenkinsjenkinsimport torch from common.utils import auto_generate_data @auto_generate_data() def mint_cdist_benchmark(): """ benchmark: torch.cdist inputs: x1: ((10, 10, 10), float32) x2: ((10, 10, 10), float32) outputs: output: ((10, 10, 10), float32) x1_grad: ((10, 10, 10), float32) x2_grad: ((10, 10, 10), float32) """ x1 = torch.randn((10, 10, 10), dtype=torch.float32) x1.requires_grad = True x2 = torch.randn((10, 10, 10), dtype=torch.float32) x2.requires_grad = True output = torch.cdist(x1, x2, 2.0) output.backward(torch.ones_like(output)) x1_grad = x1.grad x2_grad = x2.grad return [x1.detach().numpy(), x2.detach().numpy()], [ output.detach().numpy(), x1_grad.detach().numpy(), x2_grad.detach().numpy(), ] if __name__ == "__main__": mint_cdist_benchmark() test_cdist/mint_cdist_binary_case1_input0.npy0000755000175400017540000001004014770223724021731 0ustar jenkinsjenkinsNUMPYv{'descr': 'vG>P ԶY?<>?ms;j?#WV?]?4j(ԿO?T[?@?Ƥ?%o>T?hX?G/e=aM=x9@H?8:,Ha ?N @?3NM>N+"оM{o&^+e?>?<(c@e?SN>Jz;?\QgZ>?]'(¾n)5ih(?E\b¿LYC;;;)&5?ƴ = ?fk\>r&?G)>>_dL/Cn>~ݰɿ欣?\?&Pv>[FU?aD8Cg?b7=>51?p?")?6!?S>iJ?G<>2c?ҿ%.7ڂ?p/>fz?"+נxQw?"?Ͽs>K?|=>\>ϖZz3?[,x>t>" =KPV'砿HE&m`?B?"1g>,ѿ>?P}>% ?vο+=4p؀{?yX>u?#?<>,?gx?'e??iF?Q,?9ſ4z5 ??,~?nם=&?*mdD?X)v<>>JD?e@>>)e>mp?[?]ſd?]]7>}s>>XDx?@?2~ >>?>'RI?o?W ? -?@&=m眿?=qsI?_= G_潮y>RH= &>m =e9?ea?f>ۏF@?ӓ3&@>+=e>?74?G <7C[< 8:VǙA%?>uF>Z kYKK?T+Ƚ胊>> D|v=[溿 E@n? ߽4+k?M=$?m>e?T?nl^ug>`{?/WD?m6(K> @;t9ɿZ}K?؜.@@Ϯ>Ѕk!?_6?3 @œ>>Ԇ?|=G>˪?i~$Կٔ>2:?c ھmMb >!*Wп뿲j?5@?zLFV@?y%>D?l^>e>kԿ hrv+>hC?͕x?[V>31ʲؿ?s" f?ȫֿ ?\?ru>Bj?7޽ ^ʔ?6b?DPH?xk.YB6?媾ba濙??8 >˵!?t{? i?*\?um> JUX%@].kN,>0y=)P=/IP*?;g]𿜃>D>ZTf4hww}?,>RUU7򽩳5Gھʂ>]bi՛?9\'79@@.(W佑 ?W<>>`>z?e;L>עs#־& 0?lp?)l;E¿2">~)?2k=Y?B>3c>P۾ۦ쿶0??y%ÿ$*y5WS>@@|?H}?]K3\>P?4{ԭ:R=<ʕ>hA=Yp??yO_>V=m.M:?>|JA?$ZQ>?fv}?W?co{*7H(uN>uUv>2?yy:s<2>>s[@T>H> Q?Ŝ=aԆZnL!y^?R?|] J?k> ?=8 =Ý>QSgv??j h;>fDʿd5>+>$X#@=@TBJ|? j>09>dL?nDk?Կ>J< ?XT?YMy*k.>!p?+>SyW>J?>W?Ok2M@避?)?u?޿<7>9Go)?N@:.5>\? ?Sj?{e ɾP~¿wԿ ǿv>0?5?=Ú>fx#3? B?J?\P[^=-X!?J@孖>_sښiSXWRs,'fVG&Eٿ]:žD ?) fF4>78?Y>?}<"?\J?ƥ> >q>T0ea?^Uv? &u!?z?P>aX0>mԾx? ?rSӿn f=N0?lm[ѷx>4Wž3apJ"Q?@C%F;?>:!EG>?zC>r7%?툻˄2>eW¾?=?>?zW=l@ߵGԂn? ?h?U笙> !TeQ;-?P?߃3@1>G> ON >y >>8?峾i???@)fcL?qn?6@pZ>0[ܒ#Cf:΋`>[s;қ<n?_4?>+(bA !=weeW> ?<>%N0?[2I>N7X?.>`=yɄu?lEſ7?E?O?/?ϿRd׿1>b=>.b??K,?3۳C?*ګq?Q?OQ???4#6|>K[hQjL?)!R@?u>x>|ϾAN6?4?g>?ӻ >%=@}?=j ؾD?s_N >RFS?>ŌL{[?򇿭Sk? ?<3>.\?U?quPS>~Yg?7_=fh>$'?}S>$E?iӄ<u>$wnb7޿sᆕS?GQ>G? $?? of@>S#5G ]?;azͿwؿH-?`br?L?E=ؿj1?$2'z?,??* Ai?F>+޿1r|03K>1?i?^??= ?@#Uh>56?~?W>r=jbν.<>^?@U?fxGz gh*,CU;{?B?K6[+.=? w>o0?>Ɲ|?̽ҽm'?test_cdist/mint_cdist_binary_case1_output0.npy0000755000175400017540000001004014770223724022132 0ustar jenkinsjenkinsNUMPYv{'descr': '@y@4>@ @-@\@٤@g@2t@Ol@*g@T@@+@ak@I@@< @c@eΖ@!@F@Y@lh@@V@z@@@6@t@&l@/@wHJ@L+@@ A@FQ@@gTq@C@[@{@d@x@?N@|@D@,@σ_@@+@>B@U@A_@œ@f@F@Ut@q@x@h:@y@@(@U@}@Ї@wBH@dM@vė@tu@@c@t@`O@@s@ޤD@.@÷@y@|o@=O@)@O]@.@v@w@*@h@@Ux@j@z>@&@Sc@TC2@@@yd@ m@n0@Zh@\@#@]@@!@^N@&7@i@ @zf@|s@@ _@.@[Q@~k@Q|@g@@# i@N@h@x@d@~+@hl@SN@x@J˨@@Ή@ǰ@m@Z@Og@}o@@V@@9F@x@@@ғ@W}@IJ@Ђ@d@a@â@À@_>@Dv@ͮ@ y@ eL@4@Z@Cl@R@@1~@a@x@)5@v@@ (l@r@W+@y@*&f@:@@@v@@Ri@:@v@E @kO@9@+@+օ@RJ@@9@l @J@$G@`L@@qj@/@,J@$@w@@eg@N`@?`@S@ȡ@w@K@ǐ@@ @Z@_)o@ؖ@r@M@ޒ@@v@y"@Ə@R@i@@Ncc@@E\@M^@r@߉@ك@K@H~@l@y@Z@Q|@E@A!@ۀ@@xo@q}@Y@8@Aʯ@8@@@ej@-ʌ@(o?*, @ $@@@F2@ل@I@"@ l@@1@č@B|@M@7Z@ˌ@'@:r@P@f6@@Ս@@e@C@ @@^@Y@#@@/@4@s@y@r@9@@@6}@@l@X@uD@Fz@qW@M@=֤@3@f(@?̕@?@Ab@!@߰@C_@W@V@n@@@@'EY@Vt@qm@@@D@@e@,@e@<7@Ѩ@n@@F@@@J@E@jĈ@@cP@6@А@@|F@qg@@7C@l߰@x@~@1/@U@@i@\@dK@x@}@O$@喽@ϕ@֗@m@X@:T.@'{~@͟@=@-@b@j@r@kB@ѿ@g@,@˪P@@@"|@J@π@@@@ Ѥ@w{@@ <@O@@U@붆@@rY:@@-@o@ʧ@Q;@@Ї@Q@@ @;@Z@ݱn@*@@J@d@z@ E@{X@@c@ R@J@@>7@@2~@h@C@ @@sk@m*i@@@~ׇ@|U@sg@ @@oF@8>/@ |@-x@t@P@i @@6@圊@â@#c@6@=@~@6@aD@k@\@{]@@ @@ͻ@\@|@P@G@W@O@uB@@G0K@g@o:R@/څ@&ʈ@H8@U@_6d@߅@@]h@x@Α@&@F?H@QJ@sSE@@lZQ@ P@(@/Ҹ@@@(?@咍@@?@<@jyR@H@5U@$@X@9@:@?@N@,@'@nC@@*ȥ@@u@bC@6@T4@_@(8@m@t5@ s@mL@d1W@L5@2@J2@T@2^@@<[@h@Z@Fq@@N@E@@ @>@7,@D@@/@5+f@q@u@4@tX@.@V@|@Ɏ@ @?p@`@+7@ e@ױ@z@^B@@@j2@ @1@1@@C@,۲@ht@Y@XȘ@q{@z@혽@@,@N@F@@Z@@\@/k@9@6o@.@9k@mo@t@ k@Q@l3a@Sh@[@@T@޽@5@@Bk@@R@@a@l@w@a@@u@@ހ@hr@s@ @іp@@޳@`\@D@@o9@Ҁ@{.@'+N@!@$y@@y@)@Wx@"[@g+P@g\@^a@@]@(@@Y@+@@}NI@,@I@4#@&@V0@@P@@@r@u@?w@U@5H@P@ry@@3Z@༃@@@¬@}@S @@х@@AQ@@ @@/`@!@T@Z_@,@t@U@|@G@M*@A@n@=w@sF@K@@@̺@i@p@@@H@Ld@@@Ev@|t@g@|@.ڬ@&@@Ag@8@@r@n~@ ]@z@=@y@9@ll@h@p@@@3I@a@ p@M@K@-@q@A@-h@]@@#;@tD@=~@@wj@l@v@CZ@@%Mv@G1U@Z@tt@c@ nh@F@m@= @fu@eJ@0&@ΐY@4@| @@3~@U |@Rz@@d@@fG@nw,@-`@,/@A@σ@¶?1@test_cdist/mint_cdist_binary_case1_output2.npy0000755000175400017540000001004014770223724022134 0ustar jenkinsjenkinsNUMPYv{'descr': 'fm??E?!XH7⁑@@?R>]?cNT@Xz:IB;@s@"'~gB7>ii@Mƿhl?ME@yɿ?k?~4@3 6@ @ }jd￙C-@gǔ?e}@<?e |,?<8J@h?v`?\C?Re[@蠾҂q 툽1?D?E?Q3W+&ݷDq@?Zq+LL$?{0N.@.{ +>Q-@=!@ʧ?U/T@ !4Q/,?E"`>/bN;>a"?;w?`Q?=8?ƞ_^ H=r@v>fe,[)G"?2~^;W"=F,{apJ? j@po>Z7u: Ɗ?A>¿Y> > @+?d9?B>@TpVc@1Q,/,?N@Q6@X@p\%?=@R?fg @,Ѝ @W+ @}@k#?@Ǣ?`/?K@.?Ƀ@~x;@\9@2Խ<>? =^?\@Y!@i{4@TU?Oa>W>tfnfx?Z+?i@n ?wzL]>s+@ć]Lq)B? 0\@C?7@^ ?֕?y`>7z5@)@$>Iņ6??1@ 7@Q?_8m *<6J?^R@@>>3j򮲿t"@&=^]ܿj@ia@r@HB'A@:5@ ?C@@"?ѩ[@~]Y@FX@[?v??H7?)?줴2kä?忀2> 8C*@?.!o+??q=/:@@k? !@<9l>@`"˚Zg.N>s@?-=.QΝ@f9VzB>]+(@[>s?7@&c@Ky?? @->+?p񗖿i@-0@&k|0@h!?DdS@NZ,ȿFI?d@&t?)V@?v@z@dJ?IC_q Xr0@wz@_  ,]?W@@ 7@%#gE;Wj.V:ο̿e'??}Q=?#E?.B=oCd?x=cuC?cK@utA@gt迪P@yD@+74>MKBl?@C>X˾ɑ L?B+<@3?,\/q@j3?16=a9,L^@H@\?0(?8O{7?J! ]@@<ܿ:d@B?JNF;9@j_$@Gn?:?؎<9ڮ@.@P{6G@?rR@(.@\Š?$/>?JDG'@f?KQ@\@/?pӅ?Zں:@D0@1V\@'N?WeHySz @Ea>̈́>h #@??7?E5#Ɉ@O(пd=\e@zpp@ڀ2>3'9@)p!U ?UQ)?`]@?%@@F > zi?j~?&%eb ΙEtJ*w@ãtT?U k@jOj?v?L@ ?j6d?e@p@о*/@*> ;@@=@?{8>ك,y?K,Y?Ԃsn)?vg 0??R^ǻzGkV@uؿ]6e@>Xa=R΂3@v_i*?0%F9@E \kTC!?l߫?uT? !iV@ y6@y/@c1{1\]ڥUNI?HV@H@?@>=.4u+9ۥ)t=-%@yt@N>e3ӡ"?Jݣ6s f&>r>nlν:H_0]$cW=U[؈@7L/V@լ5@13VgDziU?V|Zg}@@о T@X&?b Yh;? $D@م?1@3?e>{w,}@_?9@n!(xDn?50?M@G@TRÿ @#@$ƿ$h^$?8uz=FkB?l>s=@?]$?V@Ŷ???b?|?6ET@ 6??U^j@Ȉ?y?Bt{?D@?Y@ @x\/f)?KIf{:?` ?@ ?W%2@zlL@`<տ@?-:@'?`?(Qxz R@y,?ҧLHן? ?Ʀ H@?$@]>N~h?>U@H0 >s?j)@[@n4r@QLAg@@;ÿdf`?e䀿@ @Unf5&ؿ}j]/H}}?e]"]?5?87@D_^??@޳?j`KY?9?d)?|b@} J@`}ً?㹿=C@zD.??jje뿳8?;17?M@:=/ XU.@L̿@>S?^۾M|? `@">,f@1{?-Q2?|@L$??@u̩6?Aߣ@=mx3gcm@L#>fI3!?8\@" 7, b9@Q}p?/?3*迡ES@tӏ? +0>R@e?5CAu2G"?=@`6B!@i @@{ƽ2L@Au?pWMh??2Mm?zVsi?J4n%ڗ>Õ?)@<1K?,LJ?G ?`@#_*?=>/??@?[=`F@ףf@I>Bt@^k92›`Z=@test_cdist/mint_cdist_binary_case1_input1.npy0000755000175400017540000001004014770223724021732 0ustar jenkinsjenkinsNUMPYv{'descr': 'Ѽڝ!+>N{>RS?_yQ@Qc?=? I)?A@>c:ÿ#i?1=S2:?]p}>Lӈ?U3H?`]?s>>Es==u?+ >N?=2?M7:οR?:ޥ?O?e>*?$<=Q?j>> ?r?: ľӍ!'>8!?R~>yQː籿>= Z)/ XT>Ls $_?Y}_>uBL[??,w?R"?bt?ᖿ#M=ȿ@T>O>\Έ=˜{a?ʽ>K޾xU?wˌ>ֿ5ھoE?ӿ>1lHq]$?{1eg nrCj->?_ֶw?58߼. Ɋr>k ۽8B>JK"mr*,Gv? dN=ĩ?-? Tݿã>H?1Y?u?ԝיL? ?ҾS?j:%?@@"G+htI?@la?Fђ?P5U?|=[>6m@|Zj?v??=`w?0ͧ.ZE&2Ǽ,dw?bYxh}??kh鄉= =,2V>r?Z?J>7!6蒿øZF?IKܿ; ?pqd?s>7?5=$%?ɾRȾ?1)@3:I9>?v?ja=dqXM.>7?P?ORϿ֮Y5Gƾ]s?尿S r$@X A?Lq,g8=?Ss'< ~?0?~H?T?G?Ԉ?9?)?8?@dwO? >P(8>?Q;+HBe8!*?>ۿ(&?}#? -("zG?@ z ?se?v KM!=?E3?=yƾm?*=QV?ڸ'@LCIN?r4 ?!=+?%y*,WRk?H?M)>??[yr?)ѿe)􀿒?$m???kK\=O?U?^?^it??:1о{Jz?H<˺>@ڑ?/?#Ѕۿpx{Iî~9ɿ>K>&ſ n>M>-?Z՛?f=yj>>{?}-b=$Ur?P<}k>e?Wz?"r`e?Ბ?Czm%>Xӿ2= ?k5?<gI|4>% ?V>,z0m?7{=,9YWD\p?߾민4?9 =ɾv=޾Ѵݸ?CC?[T?GiZ>ex(?j\@B> ?m6Zw?Ji0L<ӄ?Kw7?z?^O?b>wvF=/i?l[i @>iҿ߳?a?Z? >gL ?r:?"Pӿ0?w'->&˞8?6?<8LPN))@d>->ܾTG?fMs?=Ԃ%? 0?K?/?O}?Q4?Ҝ?`?>?ƿ?>=lM?F=+?C>u^?@3?e%<ݚ/&rf>y٘=|LJS>pc?u;(Z7ž׿OP?k*))?]`is>Hh?f^˿roC>D^$2pmh>2> F`Hr9恽ͪ ? ڌrAqB=|8CTa6>;^}h>IXs>[@}> [?/> ?ze޾$߸9M&l>0'*?:E?S-??u]?8B?{,1=<ྟ&>aۿeE??0?2f=&N,>GG@4@#u;gDο{w?zC>DR/?(f>Jn8wl?w@?)G2?ֱ_>Ku?4?t!/ER?1=?a\G@(?c匿Eݝ?~Z>?Z+\?>꣼Կd?$*>M T?md?;&>k?x?gM!Q?P_?El8l>ܿ8ļ?.=^s\]W??>n,@IM?c$>y?ؿ?(??ڄ ?ݿ=4?n>?ey>z?9=l?ꝿg?t@Hb>'?f> ?{3#ž?3Iξ>j?>z.?#_qm-iw>Y>P8[>-C"icԼ ?_<`?U"?/@T?o?|;V??Ψ?=?b2?^?B?ğֿLG?켿-3Y%?^i?px7]ݽ9PcBI߿?x B?~)>#z?? ;??e?t?Y4? 0>X$??1]?}>M> )@?Ͽ}ͯp?"P ,\=̿ ?@\NVT>,Ǿݿ1?,Yk?`h>ѯr'qC;=;?#*x>W"?o>>;Kɾ>Ԑ4Z?HI>m?^_PK? f @ E>E\i 6H?e=`Z?b>?¾IG)e%>?*|[?8?S漾\?>n|0ϭw?ӳa?X==̷Q%fi׿h?Hy?h?];d?|t)?0>/WT ? ^>MQ>F'=$/!IE< ?\( tE>x??v@ޯ9/ ?F}B>$1)H'~?̱?SG>)G|#?s?(> T?Qoq&z?> @!N>1O?X >p(fY @>/?߽ޡЖ ?3>Ƞ7>?/ ?!߿p??Ec?8kr^F?test_cdist/mint_cdist_binary_case1_output1.npy0000755000175400017540000001004014770223724022133 0ustar jenkinsjenkinsNUMPYv{'descr': '&@ @ ¿01@@0飿?;@1wFRy@^{>ާ>DX]->@&.?\ZJLg0?>@>l@ s?gۿD9p￁ȿ{?@6t,@3>@('@('@ G?#%@W oy@+ʫ w<B?L& .Y@4򕸿#;l}@G?hvg? @3??c",e?@:Mb3*@F>@vQE?㙿sǿ99b!@pC>` u.8?{,@_@V?4c?ضܬ> ?[ÿo?cM~J5-(z\3ۿ]~@FW@ݕ&F쾘d~z͍.b@ԟ35I@YHpuDp#+>H;@>6@5W=b;@!? @>6?9@<@?T-wޯu]@3@_ (!<ɖ>8/-@KFXA9@l?ɒs>$휿M@O3?6"a^f?뿽s@*%}@b%?- ? tP?!$zߩ>6@uC,p?#y"PЃ|۾! @ak!@Ū&?Z>!k?!@f,>D ?ޞUc| @e?̿撿Z>,u?羘<Ȥ>@!sd":?F?޿{p?r5@gn?BE6?dѴ´Љq応KcЦ? >3O?έC(*2?T;r\N??@\+6ܿZ炾; T v@Q(k * @r ?5@@,+@|?!#_0T?z߿?™??k@5ĿA~:?]֤@@c;-{dax@]?aҿ3n@D>_;?}:E@¾nk+m@Gh⢿Ry [? @:՝U= =@>\ɿV@ >x{B@Yp@ c$<>ezW@ S~?wPpLe@bd?>x.>J_+@3?䵿A?%- ^@>ٿUQq@@{}>Q?Zʙ!@ֿ.?)wV?uV_'i>r7F?Ќ@Qb[jta?Q$4GW@ Mq? @?ySW6;W諂( @T*HD?~Lmnނ>UWw@L鱿|Hпi*?®_ƿ} @DM@FG8??|pcO21O@`@ ?q@Ϻ@s>AOz~m?^Y?ET?"` (ũǿs?+@@Ώ+1>x@mɩ?R?p)pٽ\"N ?6(n@:{NsQM?~]K??.ܿۿy@߉P@_@nӿ:?)?>b5G`.seޑ7?@@pȍ?E؝TL?B>QyH@!?y@ƫA3@2(=@"^Y aAT+vp?Q7!$=?D]}>b?L q@8f@D=@?@F~*慽)@#?^vL[?X >8"b@x? >n%?2v˿ܛ.@yG@Ϳ6?"\ӗ?)O̬?KM=@!@4 +?~>E>JT<!>G@@ Ө ?Pvq6?t#@ܵ?:~?*?%@N2l?h?x?Y?@ذ>Z@{)@6_{@?rz@b~<>`܉XA&@@MXhM<@qi@?p3s[JfƄRD¿UOG'7@n6@~>-?Κ?R@7^f@NC?^?f'*@ǣֿI>@3MX3b>9ETx*;&Wտ(RJ6! @t*K}kH1?@sT?%F??u@n?:>{d??@J@[]\?Uu?? {Po>^e&m@,gAtC@#d6?\uC?07 @ ٍb߿lNؙ>?B@^3 ">?2F@HaT M[+?c^@g >R9݃@&v~Ҟ5 @p@!@ !DL\`F[?M?Ð8PO=FM^7k@5ۿP܌@0aM^] ?^&y@K?"/s??ں^?>g 8=X>t>?৾?0@`&@Ϻ>Ud)(@o?_??@V+,|]@~@|@Ch>m V^>dqXL^ =``9Xow*@x?\B?*AcbuDh({],˿44@w?&;%?b|?*@ ?e>uNdB@{I??г>??w=Mم?|?|m?'^O?˿c ??ma=@J?;>jq?2psD@>[ ?k='򠿿k"@P*6?¶Ϳp?W^?=?mYO?@94MW_<>@@?ʖ(59,u@L>?@DÈ1Ϳ<Ǚ>#\q`>KM@=_5> ??d@Bۿ@y ?˾a2mL$@ݿ9$*^:cl\w@s4H@Tn?|nvWah@8>?n@ʿW@+3j(>N22\? ?:@;?#R@.@J8;b0? w@[@X>ѦޓXO>lj"?Nk%@P3n,L޿g%s>@d@m m9?Ϳ}=Jat@VVib@?r;@?@