|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823 |
- /*******************************************************************************
- Copyright (c) 2015, The OpenBLAS Project
- All rights reserved.
- Redistribution and use in source and binary forms, with or without
- modification, are permitted provided that the following conditions are
- met:
- 1. Redistributions of source code must retain the above copyright
- notice, this list of conditions and the following disclaimer.
- 2. Redistributions in binary form must reproduce the above copyright
- notice, this list of conditions and the following disclaimer in
- the documentation and/or other materials provided with the
- distribution.
- 3. Neither the name of the OpenBLAS project nor the names of
- its contributors may be used to endorse or promote products
- derived from this software without specific prior written permission.
- THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
- AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- ARE DISCLAIMED. IN NO EVENT SHALL THE OPENBLAS PROJECT OR CONTRIBUTORS BE
- LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
- SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
- CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
- OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
- USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- *******************************************************************************/
- #define ASSEMBLER
- #include "common.h"
-
- /* X0 X1 X2 s0 X3 x4 x5 x6 x7 */
- /*int CNAME(BLASLONG bm,BLASLONG bn,BLASLONG bk,FLOAT alpha,FLOAT* ba,FLOAT* bb,FLOAT* C,BLASLONG ldc, BLASLONG offset) */
-
- #define origM x0
- #define origN x1
- #define origK x2
- #define origPA x3
- #define origPB x4
- #define pC x5
- #define LDC x6
- #define offset x7
- #define counterL x8
- #define counterI x9
- #define counterJ x10
- #define pB x11
- #define pCRow0 x12
- #define pCRow1 x13
- #define pCRow2 x14
- #define pA x15
- #define temp x16
- #define tempOffset x17
- #define tempK x18
-
- #define alpha0 s10
- #define alphaV0 v10.s[0]
- #define alpha1 s11
- #define alphaV1 v11.s[0]
- #define alpha2 s14
- #define alphaV2 v14.s[0]
- #define alpha3 s15
- #define alphaV3 v15.s[0]
-
- // 00 origM
- // 01 origN
- // 02 origK
- // 03 origPA
- // 04 origPB
- // 05 pC
- // 06 origLDC -> LDC
- // 07 offset
- // 08 counterL
- // 09 counterI
- // 10 counterJ
- // 11 pB
- // 12 pCRow0
- // 13 pCRow1
- // 14 pCRow2
- // 15 pA
- // 16 temp
- // 17 tempOffset
- // 18 must save tempK
- // 19 must save
- // 20 must save pA0_2, pA0_3
- // 21 must save pA0_6, pA0_7
- // 22 must save pA1_2, pA1_3
- // 23 must save pA1_6, pA1_7
- // 24 must save pB0_2, pB0_3
- // 25 must save pB0_6, pB0_7
- // 26 must save pB1_2, pB1_3
- // 27 must save pB1_6, pB1_7
- // 28 must save
- // 29 frame
- // 30 link
- // 31 sp
-
- //v00 ALPHA -> pA0_0, pA0_1, pA0_2, pA0_3
- //v01 pA0_4, pA0_5, pA0_6, pA0_7
- //v02 pA1_0, pA1_1, pA1_2, pA1_3
- //v03 pA1_4, pA1_5, pA1_6, pA1_7
- //v04 pB0_0, pB0_1, pB0_2, pB0_3
- //v05 pB0_4, pB0_5, pB0_6, pB0_7
- //v06 pB1_0, pB1_1, pB1_2, pB1_3
- //v07 pB1_4, pB1_5, pB1_6, pB1_7
- //v08 must save
- //v09 must save
- //v10 must save ALPHA0
- //v11 must save ALPHA1
- //v12 must save
- //v13 must save
- //v14 must save ALPHA2
- //v15 must save ALPHA3
- //v16 must save C00, C01, C02, C03
- //v17 must save C04, C05, C06, C07
- //v18 C08, C09, C10, C11
- //v19 C12, C13, C14, C15
- //v20 C16, C17, C18, C19
- //v21 C20, C21, C22, C23
- //v22 C24, C25, C26, C27
- //v23 C28, C29, C30, C31
- //v24 C32, C33, C34, C35
- //v25 C36, C37, C38, C39
- //v26 C40, C41, C42, C43
- //v27 C44, C45, C46, C47
- //v28 C48, C49, C50, C51
- //v29 C52, C53, C54, C55
- //v30 C56, C57, C58, C59
- //v31 C60, C61, C62, C63
-
- /*******************************************************************************
- * Macro definitions
- *******************************************************************************/
-
- .macro INIT8x8
- fmov s16, wzr
- fmov s17, wzr
- fmov s18, s16
- fmov s19, s17
- fmov s20, wzr
- fmov s21, s16
- fmov s22, s17
- fmov s23, s18
- fmov s24, wzr
- fmov s25, s16
- fmov s26, s17
- fmov s27, s18
- fmov s28, wzr
- fmov s29, s16
- fmov s30, s17
- fmov s31, s18
- .endm
-
- .macro KERNEL8x8_I
- ld1 {v0.4s, v1.4s}, [pA], #32
- ld1 {v4.4s, v5.4s}, [pB], #32
- ldr d2, [pA], #8
- ldr d6, [pB], #8
- ldr d3, [pA, #8]
- ldr d7, [pB, #8]
-
- ldr x22, [pA], #16
- fmul v16.4s, v0.4s, v4.s[0]
- ldr x26, [pB], #16
- fmul v17.4s, v1.4s, v4.s[0]
- ldr x23, [pA], #8
- fmul v18.4s, v0.4s, v4.s[1]
- ldr x27, [pB], #8
- fmul v19.4s, v1.4s, v4.s[1]
- fmul v20.4s, v0.4s, v4.s[2]
- fmul v21.4s, v1.4s, v4.s[2]
- fmul v22.4s, v0.4s, v4.s[3]
- fmul v23.4s, v1.4s, v4.s[3]
- fmul v24.4s, v0.4s, v5.s[0]
- fmul v25.4s, v1.4s, v5.s[0]
- fmul v26.4s, v0.4s, v5.s[1]
- fmul v27.4s, v1.4s, v5.s[1]
- fmul v28.4s, v0.4s, v5.s[2]
- fmul v29.4s, v1.4s, v5.s[2]
- fmul v30.4s, v0.4s, v5.s[3]
- fmul v31.4s, v1.4s, v5.s[3]
- .endm
-
- .macro KERNEL8x8_M1
- ldr d2, [pA], #8
- fmov v0.d[1], x20
- ldr d6, [pB], #8
- fmov v4.d[1], x24
- ldr d3, [pA, #8]
- fmov v1.d[1], x21
- ldr d7, [pB, #8]
- fmov v5.d[1], x25
- fmla v16.4s, v0.4s, v4.s[0]
- ldr x22, [pA], #16
- fmla v17.4s, v1.4s, v4.s[0]
- ldr x26, [pB], #16
- fmla v18.4s, v0.4s, v4.s[1]
- ldr x23, [pA], #8
- fmla v19.4s, v1.4s, v4.s[1]
- ldr x27, [pB], #8
- fmla v20.4s, v0.4s, v4.s[2]
- fmla v21.4s, v1.4s, v4.s[2]
- fmla v22.4s, v0.4s, v4.s[3]
- fmla v23.4s, v1.4s, v4.s[3]
- fmla v24.4s, v0.4s, v5.s[0]
- fmla v25.4s, v1.4s, v5.s[0]
- fmla v26.4s, v0.4s, v5.s[1]
- fmla v27.4s, v1.4s, v5.s[1]
- fmla v28.4s, v0.4s, v5.s[2]
- fmla v29.4s, v1.4s, v5.s[2]
- fmla v30.4s, v0.4s, v5.s[3]
- fmla v31.4s, v1.4s, v5.s[3]
- .endm
-
- .macro KERNEL8x8_M2
- ldr d0, [pA], #8
- fmov v2.d[1], x22
- ldr d4, [pB], #8
- fmov v6.d[1], x26
- ldr d1, [pA, #8]
- fmov v3.d[1], x23
- ldr d5, [pB, #8]
- fmov v7.d[1], x27
- fmla v16.4s, v2.4s, v6.s[0]
- ldr x20, [pA], #16
- fmla v17.4s, v3.4s, v6.s[0]
- ldr x24, [pB], #16
- fmla v18.4s, v2.4s, v6.s[1]
- ldr x21, [pA], #8
- fmla v19.4s, v3.4s, v6.s[1]
- ldr x25, [pB], #8
- fmla v20.4s, v2.4s, v6.s[2]
- fmla v21.4s, v3.4s, v6.s[2]
- fmla v22.4s, v2.4s, v6.s[3]
- fmla v23.4s, v3.4s, v6.s[3]
- fmla v24.4s, v2.4s, v7.s[0]
- fmla v25.4s, v3.4s, v7.s[0]
- fmla v26.4s, v2.4s, v7.s[1]
- fmla v27.4s, v3.4s, v7.s[1]
- fmla v28.4s, v2.4s, v7.s[2]
- fmla v29.4s, v3.4s, v7.s[2]
- fmla v30.4s, v2.4s, v7.s[3]
- fmla v31.4s, v3.4s, v7.s[3]
- .endm
-
- .macro KERNEL8x8_E
- fmov v2.d[1], x22
- fmov v6.d[1], x26
- fmov v3.d[1], x23
- fmov v7.d[1], x27
- fmla v16.4s, v2.4s, v6.s[0]
- fmla v17.4s, v3.4s, v6.s[0]
- fmla v18.4s, v2.4s, v6.s[1]
- fmla v19.4s, v3.4s, v6.s[1]
- fmla v20.4s, v2.4s, v6.s[2]
- fmla v21.4s, v3.4s, v6.s[2]
- fmla v22.4s, v2.4s, v6.s[3]
- fmla v23.4s, v3.4s, v6.s[3]
- fmla v24.4s, v2.4s, v7.s[0]
- fmla v25.4s, v3.4s, v7.s[0]
- fmla v26.4s, v2.4s, v7.s[1]
- fmla v27.4s, v3.4s, v7.s[1]
- fmla v28.4s, v2.4s, v7.s[2]
- fmla v29.4s, v3.4s, v7.s[2]
- fmla v30.4s, v2.4s, v7.s[3]
- fmla v31.4s, v3.4s, v7.s[3]
- .endm
-
- .macro KERNEL8x8_SUB
- ld1 {v4.4s}, [pB]
- add pB, pB, #16
- ld1 {v5.4s}, [pB]
- add pB, pB, #16
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
-
- fmla v16.4s, v0.4s, v4.s[0]
- fmla v17.4s, v1.4s, v4.s[0]
- fmla v18.4s, v0.4s, v4.s[1]
- fmla v19.4s, v1.4s, v4.s[1]
- fmla v20.4s, v0.4s, v4.s[2]
- fmla v21.4s, v1.4s, v4.s[2]
- fmla v22.4s, v0.4s, v4.s[3]
- fmla v23.4s, v1.4s, v4.s[3]
- fmla v24.4s, v0.4s, v5.s[0]
- fmla v25.4s, v1.4s, v5.s[0]
- fmla v26.4s, v0.4s, v5.s[1]
- fmla v27.4s, v1.4s, v5.s[1]
- fmla v28.4s, v0.4s, v5.s[2]
- fmla v29.4s, v1.4s, v5.s[2]
- fmla v30.4s, v0.4s, v5.s[3]
- fmla v31.4s, v1.4s, v5.s[3]
- .endm
-
- .macro SAVE8x8
- add pCRow1, pCRow0, LDC
-
- fmul v0.4s, v16.4s, alphaV0
- fmul v1.4s, v17.4s, alphaV1
- st1 {v0.4s, v1.4s}, [pCRow0]
-
- add pCRow2, pCRow1, LDC
-
- fmul v2.4s, v18.4s, alphaV2
- fmul v3.4s, v19.4s, alphaV3
- st1 {v2.4s, v3.4s}, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
- fmul v4.4s, v20.4s, alphaV0
- fmul v5.4s, v21.4s, alphaV1
- st1 {v4.4s, v5.4s}, [pCRow2]
-
- add pCRow2, pCRow1, LDC
-
- fmul v6.4s, v22.4s, alphaV2
- fmul v7.4s, v23.4s, alphaV3
- st1 {v6.4s, v7.4s}, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
- fmul v0.4s, v24.4s, alphaV0
- fmul v1.4s, v25.4s, alphaV1
- st1 {v0.4s, v1.4s}, [pCRow2]
-
- add pCRow2, pCRow1, LDC
-
- fmul v2.4s, v26.4s, alphaV2
- fmul v3.4s, v27.4s, alphaV3
- st1 {v2.4s, v3.4s}, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
- fmul v4.4s, v28.4s, alphaV0
- fmul v5.4s, v29.4s, alphaV1
- st1 {v4.4s, v5.4s}, [pCRow2]
-
- fmul v6.4s, v30.4s, alphaV2
- fmul v7.4s, v31.4s, alphaV3
- st1 {v6.4s, v7.4s}, [pCRow1]
-
- add pCRow0, pCRow0, #32
- .endm
-
- /******************************************************************************/
-
-
- .macro INIT4x8
- fmov s16, wzr
- fmov s18, wzr
- fmov s20, wzr
- fmov s22, s16
- fmov s24, wzr
- fmov s26, s16
- fmov s28, s18
- fmov s30, s20
- .endm
-
- .macro KERNEL4x8_I
- ld1 {v0.4s}, [pA], #16
- ld1 {v4.4s, v5.4s}, [pB], #32
-
- ldr d2, [pA], #8
- ldr d6, [pB], #8
- ldr d7, [pB, #8]
- ldr x21, [pA], #8
- fmul v16.4s, v0.4s, v4.s[0]
- ldr x26, [pB], #16
- fmul v18.4s, v0.4s, v4.s[1]
- ldr x27, [pB], #8
- fmul v20.4s, v0.4s, v4.s[2]
- fmul v22.4s, v0.4s, v4.s[3]
- fmul v24.4s, v0.4s, v5.s[0]
- fmul v26.4s, v0.4s, v5.s[1]
- fmul v28.4s, v0.4s, v5.s[2]
- fmul v30.4s, v0.4s, v5.s[3]
- .endm
-
- .macro KERNEL4x8_M1
- ldr d2, [pA], #8
- fmov v0.d[1], x20
- ldr d6, [pB], #8
- fmov v4.d[1], x24
- ldr d7, [pB, #8]
- fmov v5.d[1], x25
- fmla v16.4s, v0.4s, v4.s[0]
- ldr x21, [pA], #8
- fmla v18.4s, v0.4s, v4.s[1]
- ldr x26, [pB], #16
- fmla v20.4s, v0.4s, v4.s[2]
- ldr x27, [pB], #8
- fmla v22.4s, v0.4s, v4.s[3]
- fmla v24.4s, v0.4s, v5.s[0]
- fmla v26.4s, v0.4s, v5.s[1]
- fmla v28.4s, v0.4s, v5.s[2]
- fmla v30.4s, v0.4s, v5.s[3]
- .endm
-
- .macro KERNEL4x8_M2
- ldr d0, [pA], #8
- fmov v2.d[1], x21
- ldr d4, [pB], #8
- fmov v6.d[1], x26
- ldr d5, [pB, #8]
- fmov v7.d[1], x27
- fmla v16.4s, v2.4s, v6.s[0]
- ldr x20, [pA], #8
- fmla v18.4s, v2.4s, v6.s[1]
- ldr x24, [pB], #16
- fmla v20.4s, v2.4s, v6.s[2]
- ldr x25, [pB], #8
- fmla v22.4s, v2.4s, v6.s[3]
- fmla v24.4s, v2.4s, v7.s[0]
- fmla v26.4s, v2.4s, v7.s[1]
- fmla v28.4s, v2.4s, v7.s[2]
- fmla v30.4s, v2.4s, v7.s[3]
- .endm
-
- .macro KERNEL4x8_E
- fmov v2.d[1], x21
- fmov v6.d[1], x26
- fmov v7.d[1], x27
- fmla v16.4s, v2.4s, v6.s[0]
- fmla v18.4s, v2.4s, v6.s[1]
- fmla v20.4s, v2.4s, v6.s[2]
- fmla v22.4s, v2.4s, v6.s[3]
- fmla v24.4s, v2.4s, v7.s[0]
- fmla v26.4s, v2.4s, v7.s[1]
- fmla v28.4s, v2.4s, v7.s[2]
- fmla v30.4s, v2.4s, v7.s[3]
- .endm
-
- .macro KERNEL4x8_SUB
- ld1 {v4.4s}, [pB]
- add pB, pB, #16
- ld1 {v5.4s}, [pB]
- add pB, pB, #16
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
-
- fmla v16.4s, v0.4s, v4.s[0]
- fmla v18.4s, v0.4s, v4.s[1]
- fmla v20.4s, v0.4s, v4.s[2]
- fmla v22.4s, v0.4s, v4.s[3]
- fmla v24.4s, v0.4s, v5.s[0]
- fmla v26.4s, v0.4s, v5.s[1]
- fmla v28.4s, v0.4s, v5.s[2]
- fmla v30.4s, v0.4s, v5.s[3]
- .endm
-
- .macro SAVE4x8
- add pCRow1, pCRow0, LDC
-
-
- fmul v0.4s, v16.4s, alphaV0
- st1 {v0.4s}, [pCRow0]
-
- add pCRow2, pCRow1, LDC
-
-
- fmul v2.4s, v18.4s, alphaV2
- st1 {v2.4s}, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
-
- fmul v4.4s, v20.4s, alphaV0
- st1 {v4.4s}, [pCRow2]
-
- add pCRow2, pCRow1, LDC
-
-
- fmul v6.4s, v22.4s, alphaV2
- st1 {v6.4s}, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
-
- fmul v0.4s, v24.4s, alphaV0
- st1 {v0.4s}, [pCRow2]
-
- add pCRow2, pCRow1, LDC
-
-
- fmul v2.4s, v26.4s, alphaV2
- st1 {v2.4s}, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
-
- fmul v4.4s, v28.4s, alphaV0
- st1 {v4.4s}, [pCRow2]
-
-
- fmul v6.4s, v30.4s, alphaV2
- st1 {v6.4s}, [pCRow1]
-
- add pCRow0, pCRow0, #16
- .endm
-
- /******************************************************************************/
-
- .macro INIT2x8
- fmov s16, wzr
- fmov s18, wzr
- fmov s20, wzr
- fmov s22, s16
- fmov s24, wzr
- fmov s26, s16
- fmov s28, s18
- fmov s30, s20
- .endm
-
- .macro KERNEL2x8_SUB
- ld1 {v4.4s}, [pB]
- add pB, pB, #16
- ld1 {v5.4s}, [pB]
- add pB, pB, #16
- ld1 {v0.2s}, [pA]
- add pA, pA, #8
-
- fmla v16.2s, v0.2s, v4.s[0]
- fmla v18.2s, v0.2s, v4.s[1]
- fmla v20.2s, v0.2s, v4.s[2]
- fmla v22.2s, v0.2s, v4.s[3]
- fmla v24.2s, v0.2s, v5.s[0]
- fmla v26.2s, v0.2s, v5.s[1]
- fmla v28.2s, v0.2s, v5.s[2]
- fmla v30.2s, v0.2s, v5.s[3]
- .endm
-
- .macro SAVE2x8
- add pCRow1, pCRow0, LDC
-
-
- fmul v0.2s, v16.2s, alphaV0
- st1 {v0.2s}, [pCRow0]
-
- add pCRow2, pCRow1, LDC
-
-
- fmul v2.2s, v18.2s, alphaV2
- st1 {v2.2s}, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
-
- fmul v4.2s, v20.2s, alphaV0
- st1 {v4.2s}, [pCRow2]
-
- add pCRow2, pCRow1, LDC
-
-
- fmul v6.2s, v22.2s, alphaV2
- st1 {v6.2s}, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
-
- fmul v0.2s, v24.2s, alphaV0
- st1 {v0.2s}, [pCRow2]
-
- add pCRow2, pCRow1, LDC
-
-
- fmul v2.2s, v26.2s, alphaV2
- st1 {v2.2s}, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
-
- fmul v4.2s, v28.2s, alphaV0
- st1 {v4.2s}, [pCRow2]
-
-
- fmul v6.2s, v30.2s, alphaV2
- st1 {v6.2s}, [pCRow1]
-
- add pCRow0, pCRow0, #8
- .endm
-
- /******************************************************************************/
-
- .macro INIT1x8
- fmov s16, wzr
- fmov s18, wzr
- fmov s20, wzr
- fmov s22, s16
- fmov s24, wzr
- fmov s26, s16
- fmov s28, s18
- fmov s30, s20
- .endm
-
- .macro KERNEL1x8_SUB
- ld1 {v4.4s}, [pB]
- add pB, pB, #16
- ld1 {v5.4s}, [pB]
- add pB, pB, #16
- ldr s0, [pA]
- add pA, pA, #4
-
- fmla s16, s0, v4.s[0]
- fmla s18, s0, v4.s[1]
- fmla s20, s0, v4.s[2]
- fmla s22, s0, v4.s[3]
- fmla s24, s0, v5.s[0]
- fmla s26, s0, v5.s[1]
- fmla s28, s0, v5.s[2]
- fmla s30, s0, v5.s[3]
- .endm
-
- .macro SAVE1x8
- add pCRow1, pCRow0, LDC
-
-
- fmul s0, s16, alphaV0
- str s0, [pCRow0]
-
- add pCRow2, pCRow1, LDC
-
-
- fmul s2, s18, alphaV2
- str s2, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
-
- fmul s4, s20, alphaV0
- str s4, [pCRow2]
-
- add pCRow2, pCRow1, LDC
-
-
- fmul s6, s22, alphaV2
- str s6, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
-
- fmul s0, s24, alphaV0
- str s0, [pCRow2]
-
- add pCRow2, pCRow1, LDC
-
-
- fmul s2, s26, alphaV2
- str s2, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
-
- fmul s4, s28, alphaV0
- str s4, [pCRow2]
-
-
- fmul s6, s30, alphaV2
- str s6, [pCRow1]
-
- add pCRow0, pCRow0, #4
- .endm
-
- /******************************************************************************/
-
- .macro INIT8x4
- fmov s16, wzr
- fmov s17, wzr
- fmov s20, wzr
- fmov s21, s16
- fmov s24, wzr
- fmov s25, s16
- fmov s28, wzr
- fmov s29, s16
- .endm
-
- .macro KERNEL8x4_I
- ld1 {v8.4s}, [pB], #16
- ld1 {v0.4s, v1.4s}, [pA], #32
-
- ldr d9, [pB], #8
- ldr d2, [pA], #8
- ldr d3, [pA, #8]
- fmul v16.4s, v0.4s, v8.s[0]
- ldr x25, [pB], #8
- fmul v17.4s, v1.4s, v8.s[0]
- ldr x22, [pA], #16
- fmul v20.4s, v0.4s, v8.s[1]
- ldr x23, [pA], #8
- fmul v21.4s, v1.4s, v8.s[1]
- fmul v24.4s, v0.4s, v8.s[2]
- fmul v25.4s, v1.4s, v8.s[2]
- fmul v28.4s, v0.4s, v8.s[3]
- fmul v29.4s, v1.4s, v8.s[3]
- .endm
-
- .macro KERNEL8x4_M1
- ldr d9, [pB], #8
- fmov v8.d[1], x24
- ldr d2, [pA], #8
- fmov v0.d[1], x20
- ldr d3, [pA, #8]
- fmov v1.d[1], x21
- fmla v16.4s, v0.4s, v8.s[0]
- ldr x25, [pB], #8
- fmla v17.4s, v1.4s, v8.s[0]
- ldr x22, [pA], #16
- fmla v20.4s, v0.4s, v8.s[1]
- ldr x23, [pA], #8
- fmla v21.4s, v1.4s, v8.s[1]
- fmla v24.4s, v0.4s, v8.s[2]
- fmla v25.4s, v1.4s, v8.s[2]
- fmla v28.4s, v0.4s, v8.s[3]
- fmla v29.4s, v1.4s, v8.s[3]
- .endm
-
- .macro KERNEL8x4_M2
- ldr d8, [pB], #8
- fmov v9.d[1], x25
- ldr d0, [pA], #8
- fmov v2.d[1], x22
- ldr d1, [pA, #8]
- fmov v3.d[1], x23
- fmla v16.4s, v2.4s, v9.s[0]
- ldr x24, [pB], #8
- fmla v17.4s, v3.4s, v9.s[0]
- ldr x20, [pA], #16
- fmla v20.4s, v2.4s, v9.s[1]
- ldr x21, [pA], #8
- fmla v21.4s, v3.4s, v9.s[1]
- fmla v24.4s, v2.4s, v9.s[2]
- fmla v25.4s, v3.4s, v9.s[2]
- fmla v28.4s, v2.4s, v9.s[3]
- fmla v29.4s, v3.4s, v9.s[3]
- .endm
-
- .macro KERNEL8x4_E
- fmov v9.d[1], x25
- fmov v2.d[1], x22
- fmov v3.d[1], x23
- fmla v16.4s, v2.4s, v9.s[0]
- fmla v17.4s, v3.4s, v9.s[0]
- fmla v20.4s, v2.4s, v9.s[1]
- fmla v21.4s, v3.4s, v9.s[1]
- fmla v24.4s, v2.4s, v9.s[2]
- fmla v25.4s, v3.4s, v9.s[2]
- fmla v28.4s, v2.4s, v9.s[3]
- fmla v29.4s, v3.4s, v9.s[3]
- .endm
-
- .macro KERNEL8x4_SUB
- ld1 {v8.4s}, [pB], #16
- ld1 {v0.4s, v1.4s}, [pA], #32
- fmla v16.4s, v0.4s, v8.s[0]
- fmla v17.4s, v1.4s, v8.s[0]
- fmla v20.4s, v0.4s, v8.s[1]
- fmla v21.4s, v1.4s, v8.s[1]
- fmla v24.4s, v0.4s, v8.s[2]
- fmla v25.4s, v1.4s, v8.s[2]
- fmla v28.4s, v0.4s, v8.s[3]
- fmla v29.4s, v1.4s, v8.s[3]
- .endm
-
- .macro SAVE8x4
- add pCRow1, pCRow0, LDC
-
-
- fmul v0.4s, v16.4s, alphaV0
- fmul v1.4s, v17.4s, alphaV1
- st1 {v0.4s, v1.4s}, [pCRow0]
-
- add pCRow2, pCRow1, LDC
-
-
- fmul v4.4s, v20.4s, alphaV0
- fmul v5.4s, v21.4s, alphaV1
- st1 {v4.4s, v5.4s}, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
-
- fmul v0.4s, v24.4s, alphaV0
- fmul v1.4s, v25.4s, alphaV1
- st1 {v0.4s, v1.4s}, [pCRow2]
-
-
- fmul v4.4s, v28.4s, alphaV0
- fmul v5.4s, v29.4s, alphaV1
- st1 {v4.4s, v5.4s}, [pCRow1]
-
- add pCRow0, pCRow0, #32
- .endm
-
- /******************************************************************************/
-
-
- .macro INIT4x4
- fmov s16, wzr
- fmov s17, s16
- fmov s20, s17
- fmov s21, s16
- fmov s24, s17
- fmov s25, s16
- fmov s28, s17
- fmov s29, s16
- .endm
-
- .macro KERNEL4x4_I
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
- ld1 {v0.2s, v1.2s}, [pA]
- add pA, pA, #16
-
- fmul v16.2s, v0.2s, v8.s[0]
- fmul v29.2s, v1.2s, v9.s[1]
-
- fmul v20.2s, v0.2s, v8.s[1]
- fmul v25.2s, v1.2s, v9.s[0]
-
- fmul v24.2s, v0.2s, v9.s[0]
- fmul v21.2s, v1.2s, v8.s[1]
-
- fmul v28.2s, v0.2s, v9.s[1]
- fmul v17.2s, v1.2s, v8.s[0]
-
- ld1 {v12.2s, v13.2s}, [pB]
- add pB, pB, #16
- ld1 {v4.2s, v5.2s}, [pA]
- add pA, pA, #16
- .endm
-
- .macro KERNEL4x4_M1
- fmla v16.2s, v0.2s, v8.s[0]
- fmla v29.2s, v1.2s, v9.s[1]
-
- ld1 {v12.2s, v13.2s}, [pB] // For next round
- add pB, pB, #16
-
- fmla v20.2s, v0.2s, v8.s[1]
- fmla v25.2s, v1.2s, v9.s[0]
-
- ld1 {v4.2s, v5.2s}, [pA] // For next round
- add pA, pA, #16
-
- fmla v24.2s, v0.2s, v9.s[0]
- fmla v21.2s, v1.2s, v8.s[1]
-
- prfm PLDL1KEEP, [pB, #512]
-
- fmla v28.2s, v0.2s, v9.s[1]
- fmla v17.2s, v1.2s, v8.s[0]
- .endm
-
- .macro KERNEL4x4_M2
- fmla v16.2s, v4.2s, v12.s[0]
- fmla v29.2s, v5.2s, v13.s[1]
-
- ld1 {v8.2s, v9.2s}, [pB] // For next round
- add pB, pB, #16
-
- fmla v20.2s, v4.2s, v12.s[1]
- fmla v25.2s, v5.2s, v13.s[0]
-
- ld1 {v0.2s, v1.2s}, [pA] // For next round
- add pA, pA, #16
-
- fmla v24.2s, v4.2s, v13.s[0]
- fmla v21.2s, v5.2s, v12.s[1]
-
- prfm PLDL1KEEP, [pA, #512]
-
- fmla v28.2s, v4.2s, v13.s[1]
- fmla v17.2s, v5.2s, v12.s[0]
- .endm
-
- .macro KERNEL4x4_E
- fmla v16.2s, v4.2s, v12.s[0]
- fmla v29.2s, v5.2s, v13.s[1]
-
- fmla v20.2s, v4.2s, v12.s[1]
- fmla v25.2s, v5.2s, v13.s[0]
-
- fmla v24.2s, v4.2s, v13.s[0]
- fmla v21.2s, v5.2s, v12.s[1]
-
- fmla v28.2s, v4.2s, v13.s[1]
- fmla v17.2s, v5.2s, v12.s[0]
- .endm
-
- .macro KERNEL4x4_SUB
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
- ld1 {v0.2s, v1.2s}, [pA]
- add pA, pA, #16
-
- fmla v16.2s, v0.2s, v8.s[0]
- fmla v29.2s, v1.2s, v9.s[1]
-
- fmla v20.2s, v0.2s, v8.s[1]
- fmla v25.2s, v1.2s, v9.s[0]
-
- fmla v24.2s, v0.2s, v9.s[0]
- fmla v21.2s, v1.2s, v8.s[1]
-
- fmla v28.2s, v0.2s, v9.s[1]
- fmla v17.2s, v1.2s, v8.s[0]
- .endm
-
- .macro SAVE4x4
-
- fmul v8.2s, v16.2s, alphaV0
- fmul v9.2s, v17.2s, alphaV1
- st1 {v8.2s, v9.2s}, [pCRow0]
-
- add pCRow1, pCRow0, LDC
-
- fmul v12.2s, v20.2s, alphaV2
- fmul v13.2s, v21.2s, alphaV3
- st1 {v12.2s, v13.2s}, [pCRow1]
-
- add pCRow2, pCRow1, LDC
-
- fmul v8.2s, v24.2s, alphaV0
- fmul v9.2s, v25.2s, alphaV1
- st1 {v8.2s, v9.2s}, [pCRow2]
-
- add pCRow1, pCRow2, LDC
-
- fmul v12.2s, v28.2s, alphaV2
- fmul v13.2s, v29.2s, alphaV3
- st1 {v12.2s, v13.2s}, [pCRow1]
-
- add pCRow0, pCRow0, #16
- .endm
-
- /******************************************************************************/
-
- .macro INIT2x4
- fmov s16, wzr
- fmov s20, s16
- fmov s24, s20
- fmov s28, s16
- .endm
-
- .macro KERNEL2x4_SUB
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
- ld1 {v0.2s}, [pA]
- add pA, pA, #8
-
- fmla v16.2s, v0.2s, v8.s[0]
- fmla v20.2s, v0.2s, v8.s[1]
- fmla v24.2s, v0.2s, v9.s[0]
- fmla v28.2s, v0.2s, v9.s[1]
- .endm
-
- .macro SAVE2x4
-
- fmul v8.2s, v16.2s, alphaV0
- st1 {v8.2s}, [pCRow0]
-
- add pCRow1, pCRow0, LDC
-
- fmul v12.2s, v20.2s, alphaV1
- st1 {v12.2s}, [pCRow1]
-
- add pCRow2, pCRow1, LDC
-
- fmul v8.2s, v24.2s, alphaV2
- st1 {v8.2s}, [pCRow2]
-
- add pCRow1, pCRow2, LDC
-
- fmul v12.2s, v28.2s, alphaV3
- st1 {v12.2s}, [pCRow1]
-
- add pCRow0, pCRow0, #8
- .endm
-
- /******************************************************************************/
-
- .macro INIT1x4
- fmov s16, wzr
- fmov s20, s16
- .endm
-
- .macro KERNEL1x4_SUB
- ldr s0, [pA]
- add pA, pA, #4
-
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
-
- fmla v16.2s, v8.2s, v0.s[0]
- fmla v20.2s, v9.2s, v0.s[0]
- .endm
-
- .macro SAVE1x4
- add pCRow1, pCRow0, LDC
-
-
- fmul v8.2s, v16.2s, alphaV0
- st1 {v8.s}[0], [pCRow0]
- st1 {v8.s}[1], [pCRow1]
-
- add pCRow2, pCRow1, LDC
- add pCRow1, pCRow2, LDC
-
-
- fmul v12.2s, v20.2s, alphaV1
- st1 {v12.s}[0], [pCRow2]
- st1 {v12.s}[1], [pCRow1]
-
- add pCRow0, pCRow0, #4
- .endm
-
- /******************************************************************************/
-
- .macro INIT8x2
- fmov s16, wzr
- fmov s17, s16
- fmov s20, s17
- fmov s21, s16
- .endm
-
- .macro KERNEL8x2_SUB
- ld1 {v8.2s}, [pB]
- add pB, pB, #8
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
-
- fmla v16.4s, v0.4s, v8.s[0]
- fmla v17.4s, v1.4s, v8.s[0]
-
- fmla v20.4s, v0.4s, v8.s[1]
- fmla v21.4s, v1.4s, v8.s[1]
- .endm
-
- .macro SAVE8x2
- add pCRow1, pCRow0, LDC
-
-
- fmul v0.4s, v16.4s, alphaV0
- fmul v1.4s, v17.4s, alphaV1
- st1 {v0.4s, v1.4s}, [pCRow0]
-
- add pCRow2, pCRow1, LDC
-
-
- fmul v4.4s, v20.4s, alphaV0
- fmul v5.4s, v21.4s, alphaV1
- st1 {v4.4s, v5.4s}, [pCRow1]
-
- add pCRow0, pCRow0, #32
- .endm
-
- /******************************************************************************/
-
- .macro INIT4x2
- fmov s16, wzr
- fmov s17, s16
- fmov s20, s17
- fmov s21, s16
- .endm
-
- .macro KERNEL4x2_SUB
- ld1 {v8.2s}, [pB]
- add pB, pB, #8
- ld1 {v0.2s, v1.2s}, [pA]
- add pA, pA, #16
-
- fmla v16.2s, v0.2s, v8.s[0]
- fmla v17.2s, v1.2s, v8.s[0]
- fmla v20.2s, v0.2s, v8.s[1]
- fmla v21.2s, v1.2s, v8.s[1]
- .endm
-
- .macro SAVE4x2
-
- fmul v8.2s, v16.2s, alphaV0
- fmul v9.2s, v17.2s, alphaV1
- st1 {v8.2s, v9.2s}, [pCRow0]
-
- add pCRow1, pCRow0, LDC
-
- fmul v12.2s, v20.2s, alphaV2
- fmul v13.2s, v21.2s, alphaV3
- st1 {v12.2s, v13.2s}, [pCRow1]
-
- add pCRow0, pCRow0, #16
- .endm
-
- /******************************************************************************/
-
- .macro INIT2x2
- fmov s16, wzr
- fmov s20, s16
- .endm
-
- .macro KERNEL2x2_SUB
- ld1 {v8.2s}, [pB]
- add pB, pB, #8
-
- ld1 {v0.2s}, [pA]
- add pA, pA, #8
-
- fmla v16.2s, v0.2s, v8.s[0]
- fmla v20.2s, v0.2s, v8.s[1]
- .endm
-
- .macro SAVE2x2
-
- fmul v8.2s, v16.2s, alphaV0
- st1 {v8.2s}, [pCRow0]
-
- add pCRow1 , pCRow0, LDC
-
- fmul v12.2s, v20.2s, alphaV1
- st1 {v12.2s}, [pCRow1]
-
- add pCRow0, pCRow0, #8
- .endm
-
- /******************************************************************************/
-
- .macro INIT1x2
- fmov s16, wzr
- .endm
-
- .macro KERNEL1x2_SUB
- ld1 {v8.2s} , [pB]
- add pB , pB, #8
-
- ldr s0 , [pA]
- add pA, pA, #4
-
- fmla v16.2s, v8.2s, v0.s[0]
- .endm
-
- .macro SAVE1x2
- add pCRow1 , pCRow0, LDC
-
-
- fmul v8.2s, v16.2s, alphaV0
- st1 {v8.s}[0], [pCRow0]
- st1 {v8.s}[1], [pCRow1]
-
- add pCRow0, pCRow0, #4
- .endm
-
- /******************************************************************************/
-
- .macro INIT8x1
- fmov s16, wzr
- fmov s17, wzr
- .endm
-
- .macro KERNEL8x1_SUB
- ldr s8, [pB]
- add pB , pB, #4
-
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
-
- fmla v16.4s, v0.4s, v8.s[0]
- fmla v17.4s, v1.4s, v8.s[0]
- .endm
-
- .macro SAVE8x1
-
- fmul v0.4s, v16.4s, alphaV0
- fmul v1.4s, v17.4s, alphaV1
- st1 {v0.4s, v1.4s}, [pCRow0]
-
- add pCRow0, pCRow0, #32
- .endm
-
- /******************************************************************************/
-
- .macro INIT4x1
- fmov s16, wzr
- fmov s17, s16
- .endm
-
- .macro KERNEL4x1_SUB
- ldr s8, [pB]
- add pB , pB, #4
-
- ld1 {v0.2s, v1.2s}, [pA]
- add pA , pA, #16
-
- fmla v16.2s, v0.2s, v8.s[0]
- fmla v17.2s, v1.2s, v8.s[0]
- .endm
-
- .macro SAVE4x1
-
- fmul v8.2s, v16.2s, alphaV0
- fmul v9.2s, v17.2s, alphaV1
- st1 {v8.2s, v9.2s}, [pCRow0]
-
- add pCRow0, pCRow0, #16
- .endm
-
- /******************************************************************************/
-
- .macro INIT2x1
- fmov s16, wzr
- .endm
-
- .macro KERNEL2x1_SUB
- ldr s8, [pB]
- add pB , pB, #4
-
- ld1 {v0.2s}, [pA]
- add pA , pA, #8
-
- fmla v16.2s, v0.2s, v8.s[0]
- .endm
-
- .macro SAVE2x1
-
- fmul v8.2s, v16.2s, alphaV0
- st1 {v8.2s}, [pCRow0]
-
- add pCRow0, pCRow0, #8
- .endm
-
- /******************************************************************************/
-
- .macro INIT1x1
- fmov s16, wzr
- .endm
-
- .macro KERNEL1x1_SUB
- ldr s8, [pB]
- add pB , pB, #4
-
- ldr s0, [pA]
- add pA , pA, #4
-
- fmadd s16, s0, s8, s16
- .endm
-
- .macro SAVE1x1
-
- fmul s8, s16, alpha0
- str s8, [pCRow0]
-
- add pCRow0, pCRow0, #4
- .endm
-
- /*******************************************************************************
- * End of macro definitions
- *******************************************************************************/
-
- PROLOGUE
-
- .Lstrmm_kernel_begin:
-
- .align 5
- add sp, sp, #-(11 * 16)
- stp d8, d9, [sp, #(0 * 16)]
- stp d10, d11, [sp, #(1 * 16)]
- stp d12, d13, [sp, #(2 * 16)]
- stp d14, d15, [sp, #(3 * 16)]
- stp d16, d17, [sp, #(4 * 16)]
- stp x18, x19, [sp, #(5 * 16)]
- stp x20, x21, [sp, #(6 * 16)]
- stp x22, x23, [sp, #(7 * 16)]
- stp x24, x25, [sp, #(8 * 16)]
- stp x26, x27, [sp, #(9 * 16)]
- str x28, [sp, #(10 * 16)]
-
- fmov alpha0, s0
- fmov alpha1, s0
- fmov alpha2, s0
- fmov alpha3, s0
-
- lsl LDC, LDC, #2 // ldc = ldc * 4
-
- #if !defined(LEFT)
- neg tempOffset, offset
- #endif
- mov pB, origPB
-
- mov counterJ, origN
- asr counterJ, counterJ, #3 // J = J / 8
- cmp counterJ, #0
- ble .Lstrmm_kernel_L4_BEGIN
-
- /******************************************************************************/
- /******************************************************************************/
-
- .Lstrmm_kernel_L8_BEGIN:
- mov pCRow0, pC // pCRow0 = C
- add pC, pC, LDC, lsl #3
-
- #if defined(LEFT)
- mov tempOffset, offset
- #endif
-
- mov pA, origPA // pA = start of A array
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L8_M8_BEGIN:
-
- mov counterI, origM
- asr counterI, counterI, #3 // counterI = counterI / 8
- cmp counterI, #0
- ble .Lstrmm_kernel_L8_M4_BEGIN
-
- .Lstrmm_kernel_L8_M8_20:
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #5
- add pA, pA, temp
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #8
- #else
- add tempK, tempOffset, #8
- #endif
-
- asr counterL , tempK, #3 // L = K / 8
- cmp counterL , #2 // is there at least 16 to do?
- blt .Lstrmm_kernel_L8_M8_32
-
- KERNEL8x8_I // do one in the K
- KERNEL8x8_M2 // do another in the K
- KERNEL8x8_M1
- KERNEL8x8_M2
- KERNEL8x8_M1
- KERNEL8x8_M2
- KERNEL8x8_M1
- KERNEL8x8_M2
-
- subs counterL, counterL, #2
- ble .Lstrmm_kernel_L8_M8_22a
- .align 5
-
- .Lstrmm_kernel_L8_M8_22:
-
- KERNEL8x8_M1
- KERNEL8x8_M2
- KERNEL8x8_M1
- KERNEL8x8_M2
- KERNEL8x8_M1
- KERNEL8x8_M2
- KERNEL8x8_M1
- KERNEL8x8_M2
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L8_M8_22
-
- .Lstrmm_kernel_L8_M8_22a:
-
- KERNEL8x8_M1
- KERNEL8x8_M2
- KERNEL8x8_M1
- KERNEL8x8_M2
- KERNEL8x8_M1
- KERNEL8x8_M2
- KERNEL8x8_M1
- KERNEL8x8_E
-
- b .Lstrmm_kernel_L8_M8_44
-
- .Lstrmm_kernel_L8_M8_32:
-
- tst counterL, #1
- ble .Lstrmm_kernel_L8_M8_40
-
- KERNEL8x8_I
- KERNEL8x8_M2
- KERNEL8x8_M1
- KERNEL8x8_M2
- KERNEL8x8_M1
- KERNEL8x8_M2
- KERNEL8x8_M1
- KERNEL8x8_E
-
- b .Lstrmm_kernel_L8_M8_44
-
- .Lstrmm_kernel_L8_M8_40:
-
- INIT8x8
-
- .Lstrmm_kernel_L8_M8_44:
-
- ands counterL , tempK, #7
- ble .Lstrmm_kernel_L8_M8_100
-
- .Lstrmm_kernel_L8_M8_46:
-
- KERNEL8x8_SUB
-
- subs counterL, counterL, 1
- bgt .Lstrmm_kernel_L8_M8_46
-
- .Lstrmm_kernel_L8_M8_100:
-
- SAVE8x8
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #8
- #else
- sub tempK, tempK, #8
- #endif
- lsl temp, tempK, #5
- add pA, pA, temp
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #8
- #endif
-
- .Lstrmm_kernel_L8_M8_END:
- subs counterI, counterI, #1
- bne .Lstrmm_kernel_L8_M8_20
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L8_M4_BEGIN:
-
- mov counterI, origM
- tst counterI , #7
- ble .Lstrmm_kernel_L8_END
-
- tst counterI, #4
- ble .Lstrmm_kernel_L8_M2_BEGIN
-
- .Lstrmm_kernel_L8_M4_20:
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #4
- add pA, pA, temp
- lsl temp, tempOffset, #5
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #4
- #else
- add tempK, tempOffset, #8
- #endif
-
- asr counterL , tempK, #1 // L = K / 2
- cmp counterL , #2 // is there at least 4 to do?
- blt .Lstrmm_kernel_L8_M4_32
-
- KERNEL4x8_I // do one in the K
- KERNEL4x8_M2 // do another in the K
-
- subs counterL, counterL, #2
- ble .Lstrmm_kernel_L8_M4_22a
- .align 5
-
- .Lstrmm_kernel_L8_M4_22:
-
- KERNEL4x8_M1
- KERNEL4x8_M2
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L8_M4_22
-
- .Lstrmm_kernel_L8_M4_22a:
-
- KERNEL4x8_M1
- KERNEL4x8_E
-
- b .Lstrmm_kernel_L8_M4_44
-
- .Lstrmm_kernel_L8_M4_32:
-
- tst counterL, #1
- ble .Lstrmm_kernel_L8_M4_40
-
- KERNEL4x8_I
- KERNEL4x8_E
-
- b .Lstrmm_kernel_L8_M4_44
-
- .Lstrmm_kernel_L8_M4_40:
-
- INIT4x8
-
- .Lstrmm_kernel_L8_M4_44:
-
- ands counterL , tempK, #1
- ble .Lstrmm_kernel_L8_M4_100
-
- .Lstrmm_kernel_L8_M4_46:
-
- KERNEL4x8_SUB
-
- .Lstrmm_kernel_L8_M4_100:
-
- SAVE4x8
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #4
- #else
- sub tempK, tempK, #8
- #endif
- lsl temp, tempK, #4
- add pA, pA, temp
- lsl temp, tempK, #5
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #4
- #endif
-
- .Lstrmm_kernel_L8_M4_END:
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L8_M2_BEGIN:
-
- mov counterI, origM
- tst counterI , #3
- ble .Lstrmm_kernel_L8_END
-
- tst counterI, #2 // counterI = counterI / 2
- ble .Lstrmm_kernel_L8_M1_BEGIN
-
- .Lstrmm_kernel_L8_M2_20:
-
- INIT2x8
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #3
- add pA, pA, temp
- lsl temp, tempOffset, #5
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #2
- #else
- add tempK, tempOffset, #8
- #endif
-
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble .Lstrmm_kernel_L8_M2_40
-
- .Lstrmm_kernel_L8_M2_22:
-
- KERNEL2x8_SUB
- KERNEL2x8_SUB
- KERNEL2x8_SUB
- KERNEL2x8_SUB
-
- KERNEL2x8_SUB
- KERNEL2x8_SUB
- KERNEL2x8_SUB
- KERNEL2x8_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L8_M2_22
-
-
- .Lstrmm_kernel_L8_M2_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble .Lstrmm_kernel_L8_M2_100
-
- .Lstrmm_kernel_L8_M2_42:
-
- KERNEL2x8_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L8_M2_42
-
- .Lstrmm_kernel_L8_M2_100:
-
- SAVE2x8
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #2
- #else
- sub tempK, tempK, #8
- #endif
- lsl temp, tempK, #3
- add pA, pA, temp
- lsl temp, tempK, #5
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #2
- #endif
-
- .Lstrmm_kernel_L8_M2_END:
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L8_M1_BEGIN:
-
- tst counterI, #1 // counterI = counterI % 2
- ble .Lstrmm_kernel_L8_END
-
- .Lstrmm_kernel_L8_M1_20:
-
- INIT1x8
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #2
- add pA, pA, temp
- lsl temp, tempOffset, #5
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #1
- #else
- add tempK, tempOffset, #8
- #endif
-
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble .Lstrmm_kernel_L8_M1_40
-
- .Lstrmm_kernel_L8_M1_22:
- KERNEL1x8_SUB
- KERNEL1x8_SUB
- KERNEL1x8_SUB
- KERNEL1x8_SUB
-
- KERNEL1x8_SUB
- KERNEL1x8_SUB
- KERNEL1x8_SUB
- KERNEL1x8_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L8_M1_22
-
-
- .Lstrmm_kernel_L8_M1_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble .Lstrmm_kernel_L8_M1_100
-
- .Lstrmm_kernel_L8_M1_42:
-
- KERNEL1x8_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L8_M1_42
-
- .Lstrmm_kernel_L8_M1_100:
-
- SAVE1x8
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #1
- #else
- sub tempK, tempK, #8
- #endif
- lsl temp, tempK, #2
- add pA, pA, temp
- lsl temp, tempK, #5
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #1
- #endif
-
- .Lstrmm_kernel_L8_END:
- lsl temp, origK, #5 // B = B + K * 4 * 8
- add origPB, origPB, temp
-
- #if !defined(LEFT)
- add tempOffset, tempOffset, #8
- #endif
-
- subs counterJ, counterJ , #1 // j--
- bgt .Lstrmm_kernel_L8_BEGIN
-
- /******************************************************************************/
- /******************************************************************************/
-
- .Lstrmm_kernel_L4_BEGIN:
-
- mov counterJ , origN
- tst counterJ , #7
- ble .Lstrmm_kernel_L999
-
- tst counterJ , #4
- ble .Lstrmm_kernel_L2_BEGIN
-
- mov pCRow0, pC // pCRow0 = pC
-
- add pC,pC,LDC, lsl #2
-
- #if defined(LEFT)
- mov tempOffset, offset
- #endif
-
- mov pA, origPA // pA = A
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L4_M8_BEGIN:
-
- mov counterI, origM
- asr counterI, counterI, #3 // counterI = counterI / 8
- cmp counterI, #0
- ble .Lstrmm_kernel_L4_M4_BEGIN
-
- .Lstrmm_kernel_L4_M8_20:
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #5
- add pA, pA, temp
- lsl temp, tempOffset, #4
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #8
- #else
- add tempK, tempOffset, #4
- #endif
-
- asr counterL , tempK, #1 // L = K / 2
- cmp counterL , #2 // is there at least 4 to do?
- blt .Lstrmm_kernel_L4_M8_32
-
- KERNEL8x4_I // do one in the K
- KERNEL8x4_M2 // do another in the K
-
- subs counterL, counterL, #2
- ble .Lstrmm_kernel_L4_M8_22a
- .align 5
-
- .Lstrmm_kernel_L4_M8_22:
-
- KERNEL8x4_M1
- KERNEL8x4_M2
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L4_M8_22
-
- .Lstrmm_kernel_L4_M8_22a:
-
- KERNEL8x4_M1
- KERNEL8x4_E
-
- b .Lstrmm_kernel_L4_M8_44
-
- .Lstrmm_kernel_L4_M8_32:
-
- tst counterL, #1
- ble .Lstrmm_kernel_L4_M8_40
-
- KERNEL8x4_I
- KERNEL8x4_E
-
- b .Lstrmm_kernel_L4_M8_44
-
- .Lstrmm_kernel_L4_M8_40:
-
- INIT8x4
-
- .Lstrmm_kernel_L4_M8_44:
-
- ands counterL , tempK, #1
- ble .Lstrmm_kernel_L4_M8_100
-
- .Lstrmm_kernel_L4_M8_46:
-
- KERNEL8x4_SUB
-
- .Lstrmm_kernel_L4_M8_100:
-
- SAVE8x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #8
- #else
- sub tempK, tempK, #4
- #endif
- lsl temp, tempK, #5
- add pA, pA, temp
- lsl temp, tempK, #4
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #8
- #endif
- .Lstrmm_kernel_L4_M8_END:
- subs counterI, counterI, #1
- bne .Lstrmm_kernel_L4_M8_20
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L4_M4_BEGIN:
-
- mov counterI, origM
- tst counterI , #7
- ble .Lstrmm_kernel_L4_END
-
- tst counterI, #4
- ble .Lstrmm_kernel_L4_M2_BEGIN
-
- .Lstrmm_kernel_L4_M4_20:
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #4
- add pB, pB, temp
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #4
- #else
- add tempK, tempOffset, #4
- #endif
- asr counterL , tempK, #1 // L = K / 2
- cmp counterL , #2 // is there at least 4 to do?
- blt .Lstrmm_kernel_L4_M4_32
-
- KERNEL4x4_I // do one in the K
- KERNEL4x4_M2 // do another in the K
-
- subs counterL, counterL, #2
- ble .Lstrmm_kernel_L4_M4_22a
- .align 5
-
- .Lstrmm_kernel_L4_M4_22:
-
- KERNEL4x4_M1
- KERNEL4x4_M2
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L4_M4_22
-
- .Lstrmm_kernel_L4_M4_22a:
-
- KERNEL4x4_M1
- KERNEL4x4_E
-
- b .Lstrmm_kernel_L4_M4_44
-
- .Lstrmm_kernel_L4_M4_32:
-
- tst counterL, #1
- ble .Lstrmm_kernel_L4_M4_40
-
- KERNEL4x4_I
- KERNEL4x4_E
-
- b .Lstrmm_kernel_L4_M4_44
-
- .Lstrmm_kernel_L4_M4_40:
-
- INIT4x4
-
- .Lstrmm_kernel_L4_M4_44:
-
- ands counterL , tempK, #1
- ble .Lstrmm_kernel_L4_M4_100
-
- .Lstrmm_kernel_L4_M4_46:
-
- KERNEL4x4_SUB
-
- .Lstrmm_kernel_L4_M4_100:
-
- SAVE4x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #4
- #else
- sub tempK, tempK, #4
- #endif
- lsl temp, tempK, #4
- add pA, pA, temp
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #4
- #endif
- .Lstrmm_kernel_L4_M4_END:
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L4_M2_BEGIN:
-
- mov counterI, origM
- tst counterI , #3
- ble .Lstrmm_kernel_L4_END
-
- tst counterI, #2 // counterI = counterI / 2
- ble .Lstrmm_kernel_L4_M1_BEGIN
-
- .Lstrmm_kernel_L4_M2_20:
-
- INIT2x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #3
- add pA, pA, temp
- lsl temp, tempOffset, #4
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #2
- #else
- add tempK, tempOffset, #4
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble .Lstrmm_kernel_L4_M2_40
-
- .Lstrmm_kernel_L4_M2_22:
-
- KERNEL2x4_SUB
- KERNEL2x4_SUB
- KERNEL2x4_SUB
- KERNEL2x4_SUB
-
- KERNEL2x4_SUB
- KERNEL2x4_SUB
- KERNEL2x4_SUB
- KERNEL2x4_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L4_M2_22
-
-
- .Lstrmm_kernel_L4_M2_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble .Lstrmm_kernel_L4_M2_100
-
- .Lstrmm_kernel_L4_M2_42:
-
- KERNEL2x4_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L4_M2_42
-
- .Lstrmm_kernel_L4_M2_100:
-
- SAVE2x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #2
- #else
- sub tempK, tempK, #4
- #endif
- lsl temp, tempK, #3
- add pA, pA, temp
- lsl temp, tempK, #4
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #2
- #endif
- .Lstrmm_kernel_L4_M2_END:
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L4_M1_BEGIN:
-
- tst counterI, #1 // counterI = counterI % 2
- ble .Lstrmm_kernel_L4_END
-
- .Lstrmm_kernel_L4_M1_20:
-
- INIT1x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #4
- add pB, pB, temp
- lsl temp, tempOffset, #2
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #1
- #else
- add tempK, tempOffset, #4
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble .Lstrmm_kernel_L4_M1_40
-
- .Lstrmm_kernel_L4_M1_22:
- KERNEL1x4_SUB
- KERNEL1x4_SUB
- KERNEL1x4_SUB
- KERNEL1x4_SUB
-
- KERNEL1x4_SUB
- KERNEL1x4_SUB
- KERNEL1x4_SUB
- KERNEL1x4_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L4_M1_22
-
-
- .Lstrmm_kernel_L4_M1_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble .Lstrmm_kernel_L4_M1_100
-
- .Lstrmm_kernel_L4_M1_42:
-
- KERNEL1x4_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L4_M1_42
-
- .Lstrmm_kernel_L4_M1_100:
-
- SAVE1x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #1
- #else
- sub tempK, tempK, #4
- #endif
- lsl temp, tempK, #2
- add pA, pA, temp
- lsl temp, tempK, #4
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #1
- #endif
- .Lstrmm_kernel_L4_END:
- add origPB, origPB, origK, lsl #4 // B = B + K * 4 * 4
- #if !defined(LEFT)
- add tempOffset, tempOffset, #4
- #endif
-
- /******************************************************************************/
- /******************************************************************************/
-
- .Lstrmm_kernel_L2_BEGIN: // less than 2 left in N direction
-
- mov counterJ , origN
- tst counterJ , #3
- ble .Lstrmm_kernel_L999
-
- tst counterJ , #2
- ble .Lstrmm_kernel_L1_BEGIN
-
- mov pCRow0, pC // pCRow0 = pC
-
- add pC,pC,LDC, lsl #1
-
- #if defined(LEFT)
- mov tempOffset, offset
- #endif
- mov pA, origPA // pA = A
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L2_M8_BEGIN:
-
- mov counterI, origM
- asr counterI, counterI, #3 // counterI = counterI / 8
- cmp counterI,#0
- ble .Lstrmm_kernel_L2_M4_BEGIN
-
- .Lstrmm_kernel_L2_M8_20:
-
- INIT8x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #5
- add pA, pA, temp
- lsl temp, tempOffset, #3
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #8
- #else
- add tempK, tempOffset, #2
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL,#0
- ble .Lstrmm_kernel_L2_M8_40
- .align 5
-
- .Lstrmm_kernel_L2_M8_22:
- KERNEL8x2_SUB
- KERNEL8x2_SUB
- KERNEL8x2_SUB
- KERNEL8x2_SUB
-
- KERNEL8x2_SUB
- KERNEL8x2_SUB
- KERNEL8x2_SUB
- KERNEL8x2_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L2_M8_22
-
-
- .Lstrmm_kernel_L2_M8_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble .Lstrmm_kernel_L2_M8_100
-
- .Lstrmm_kernel_L2_M8_42:
-
- KERNEL8x2_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L2_M8_42
-
- .Lstrmm_kernel_L2_M8_100:
-
- SAVE8x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #8
- #else
- sub tempK, tempK, #2
- #endif
- lsl temp, tempK, #5
- add pA, pA, temp
- lsl temp, tempK, #3
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #8
- #endif
- .Lstrmm_kernel_L2_M8_END:
-
- subs counterI, counterI, #1
- bgt .Lstrmm_kernel_L2_M8_20
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L2_M4_BEGIN:
-
- mov counterI, origM
- tst counterI , #7
- ble .Lstrmm_kernel_L2_END
-
- tst counterI, #4
- ble .Lstrmm_kernel_L2_M2_BEGIN
-
- .Lstrmm_kernel_L2_M4_20:
-
- INIT4x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #3
- add pB, pB, temp
- lsl temp, tempOffset, #4
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #4
- #else
- add tempK, tempOffset, #2
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL,#0
- ble .Lstrmm_kernel_L2_M4_40
- .align 5
-
- .Lstrmm_kernel_L2_M4_22:
- KERNEL4x2_SUB
- KERNEL4x2_SUB
- KERNEL4x2_SUB
- KERNEL4x2_SUB
-
- KERNEL4x2_SUB
- KERNEL4x2_SUB
- KERNEL4x2_SUB
- KERNEL4x2_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L2_M4_22
-
-
- .Lstrmm_kernel_L2_M4_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble .Lstrmm_kernel_L2_M4_100
-
- .Lstrmm_kernel_L2_M4_42:
-
- KERNEL4x2_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L2_M4_42
-
- .Lstrmm_kernel_L2_M4_100:
-
- SAVE4x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #4
- #else
- sub tempK, tempK, #2
- #endif
- lsl temp, tempK, #4
- add pA, pA, temp
- lsl temp, tempK, #3
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #4
- #endif
- .Lstrmm_kernel_L2_M4_END:
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L2_M2_BEGIN:
-
- mov counterI, origM
- tst counterI , #3
- ble .Lstrmm_kernel_L2_END
-
- tst counterI, #2 // counterI = counterI / 2
- ble .Lstrmm_kernel_L2_M1_BEGIN
-
- .Lstrmm_kernel_L2_M2_20:
-
- INIT2x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #3
- add pB, pB, temp
- lsl temp, tempOffset, #3
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #2
- #else
- add tempK, tempOffset, #2
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL,#0
- ble .Lstrmm_kernel_L2_M2_40
-
- .Lstrmm_kernel_L2_M2_22:
-
- KERNEL2x2_SUB
- KERNEL2x2_SUB
- KERNEL2x2_SUB
- KERNEL2x2_SUB
-
- KERNEL2x2_SUB
- KERNEL2x2_SUB
- KERNEL2x2_SUB
- KERNEL2x2_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L2_M2_22
-
-
- .Lstrmm_kernel_L2_M2_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble .Lstrmm_kernel_L2_M2_100
-
- .Lstrmm_kernel_L2_M2_42:
-
- KERNEL2x2_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L2_M2_42
-
- .Lstrmm_kernel_L2_M2_100:
-
- SAVE2x2
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #2
- #else
- sub tempK, tempK, #2
- #endif
- lsl temp, tempK, #3
- add pA, pA, temp
- lsl temp, tempK, #3
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #2
- #endif
-
- .Lstrmm_kernel_L2_M2_END:
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L2_M1_BEGIN:
-
- tst counterI, #1 // counterI = counterI % 2
- ble .Lstrmm_kernel_L2_END
-
- .Lstrmm_kernel_L2_M1_20:
-
- INIT1x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #3
- add pB, pB, temp
- lsl temp, tempOffset, #2
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #1
- #else
- add tempK, tempOffset, #2
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL, #0
- ble .Lstrmm_kernel_L2_M1_40
-
- .Lstrmm_kernel_L2_M1_22:
- KERNEL1x2_SUB
- KERNEL1x2_SUB
- KERNEL1x2_SUB
- KERNEL1x2_SUB
-
- KERNEL1x2_SUB
- KERNEL1x2_SUB
- KERNEL1x2_SUB
- KERNEL1x2_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L2_M1_22
-
-
- .Lstrmm_kernel_L2_M1_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble .Lstrmm_kernel_L2_M1_100
-
- .Lstrmm_kernel_L2_M1_42:
-
- KERNEL1x2_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L2_M1_42
-
- .Lstrmm_kernel_L2_M1_100:
-
- SAVE1x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #1
- #else
- sub tempK, tempK, #2
- #endif
- lsl temp, tempK, #2
- add pA, pA, temp
- lsl temp, tempK, #3
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #1
- #endif
- .Lstrmm_kernel_L2_END:
- #if !defined(LEFT)
- add tempOffset, tempOffset, #2
- #endif
- add origPB, origPB, origK, lsl #3 // B = B + K * 2 * 4
-
- /******************************************************************************/
- /******************************************************************************/
-
- .Lstrmm_kernel_L1_BEGIN:
-
- mov counterJ , origN
- tst counterJ , #1
- ble .Lstrmm_kernel_L999 // done
-
-
- mov pCRow0, pC // pCRow0 = C
- add pC , pC , LDC // Update pC to point to next
-
- #if defined(LEFT)
- mov tempOffset, offset
- #endif
- mov pA, origPA // pA = A
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L1_M8_BEGIN:
-
- mov counterI, origM
- asr counterI, counterI, #3
- cmp counterI, #0
- ble .Lstrmm_kernel_L1_M4_BEGIN
-
- .Lstrmm_kernel_L1_M8_20:
-
- INIT8x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #5
- add pA, pA, temp
- lsl temp, tempOffset, #2
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #8
- #else
- add tempK, tempOffset, #1
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble .Lstrmm_kernel_L1_M8_40
- .align 5
-
- .Lstrmm_kernel_L1_M8_22:
- KERNEL8x1_SUB
- KERNEL8x1_SUB
- KERNEL8x1_SUB
- KERNEL8x1_SUB
-
- KERNEL8x1_SUB
- KERNEL8x1_SUB
- KERNEL8x1_SUB
- KERNEL8x1_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L1_M8_22
-
-
- .Lstrmm_kernel_L1_M8_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble .Lstrmm_kernel_L1_M8_100
-
- .Lstrmm_kernel_L1_M8_42:
-
- KERNEL8x1_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L1_M8_42
-
- .Lstrmm_kernel_L1_M8_100:
-
- SAVE8x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #8
- #else
- sub tempK, tempK, #1
- #endif
- lsl temp, tempK, #5
- add pA, pA, temp
- lsl temp, tempK, #2
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #8
- #endif
- .Lstrmm_kernel_L1_M8_END:
-
- subs counterI, counterI, #1
- bgt .Lstrmm_kernel_L1_M8_20
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L1_M4_BEGIN:
-
- mov counterI, origM
- tst counterI , #7
- ble .Lstrmm_kernel_L1_END
-
- tst counterI, #4
- ble .Lstrmm_kernel_L1_M2_BEGIN
-
- .Lstrmm_kernel_L1_M4_20:
-
- INIT4x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #2
- add pB, pB, temp
- lsl temp, tempOffset, #4
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #4
- #else
- add tempK, tempOffset, #1
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble .Lstrmm_kernel_L1_M4_40
- .align 5
-
- .Lstrmm_kernel_L1_M4_22:
- KERNEL4x1_SUB
- KERNEL4x1_SUB
- KERNEL4x1_SUB
- KERNEL4x1_SUB
-
- KERNEL4x1_SUB
- KERNEL4x1_SUB
- KERNEL4x1_SUB
- KERNEL4x1_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L1_M4_22
-
-
- .Lstrmm_kernel_L1_M4_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble .Lstrmm_kernel_L1_M4_100
-
- .Lstrmm_kernel_L1_M4_42:
-
- KERNEL4x1_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L1_M4_42
-
- .Lstrmm_kernel_L1_M4_100:
-
- SAVE4x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #4
- #else
- sub tempK, tempK, #1
- #endif
- lsl temp, tempK, #4
- add pA, pA, temp
- lsl temp, tempK, #2
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #4
- #endif
- .Lstrmm_kernel_L1_M4_END:
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L1_M2_BEGIN:
-
- mov counterI, origM
- tst counterI , #3
- ble .Lstrmm_kernel_L1_END
-
- tst counterI, #2 // counterI = counterI / 2
- ble .Lstrmm_kernel_L1_M1_BEGIN
-
- .Lstrmm_kernel_L1_M2_20:
-
- INIT2x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #2
- add pB, pB, temp
- lsl temp, tempOffset, #3
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #2
- #else
- add tempK, tempOffset, #1
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble .Lstrmm_kernel_L1_M2_40
-
- .Lstrmm_kernel_L1_M2_22:
-
- KERNEL2x1_SUB
- KERNEL2x1_SUB
- KERNEL2x1_SUB
- KERNEL2x1_SUB
-
- KERNEL2x1_SUB
- KERNEL2x1_SUB
- KERNEL2x1_SUB
- KERNEL2x1_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L1_M2_22
-
-
- .Lstrmm_kernel_L1_M2_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble .Lstrmm_kernel_L1_M2_100
-
- .Lstrmm_kernel_L1_M2_42:
-
- KERNEL2x1_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L1_M2_42
-
- .Lstrmm_kernel_L1_M2_100:
-
- SAVE2x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #2
- #else
- sub tempK, tempK, #1
- #endif
- lsl temp, tempK, #3
- add pA, pA, temp
- lsl temp, tempK, #2
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #2
- #endif
- .Lstrmm_kernel_L1_M2_END:
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L1_M1_BEGIN:
-
- tst counterI, #1 // counterI = counterI % 2
- ble .Lstrmm_kernel_L1_END
-
- .Lstrmm_kernel_L1_M1_20:
-
- INIT1x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #2
- add pB, pB, temp
- lsl temp, tempOffset, #2
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #1
- #else
- add tempK, tempOffset, #1
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble .Lstrmm_kernel_L1_M1_40
-
- .Lstrmm_kernel_L1_M1_22:
- KERNEL1x1_SUB
- KERNEL1x1_SUB
- KERNEL1x1_SUB
- KERNEL1x1_SUB
-
- KERNEL1x1_SUB
- KERNEL1x1_SUB
- KERNEL1x1_SUB
- KERNEL1x1_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L1_M1_22
-
-
- .Lstrmm_kernel_L1_M1_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble .Lstrmm_kernel_L1_M1_100
-
- .Lstrmm_kernel_L1_M1_42:
-
- KERNEL1x1_SUB
-
- subs counterL, counterL, #1
- bgt .Lstrmm_kernel_L1_M1_42
-
- .Lstrmm_kernel_L1_M1_100:
-
- SAVE1x1
-
- .Lstrmm_kernel_L1_END:
-
- /******************************************************************************/
-
- .Lstrmm_kernel_L999:
- mov x0, #0 // set return value
- ldp d8, d9, [sp, #(0 * 16)]
- ldp d10, d11, [sp, #(1 * 16)]
- ldp d12, d13, [sp, #(2 * 16)]
- ldp d14, d15, [sp, #(3 * 16)]
- ldp d16, d17, [sp, #(4 * 16)]
- ldp x18, x19, [sp, #(5 * 16)]
- ldp x20, x21, [sp, #(6 * 16)]
- ldp x22, x23, [sp, #(7 * 16)]
- ldp x24, x25, [sp, #(8 * 16)]
- ldp x26, x27, [sp, #(9 * 16)]
- ldr x28, [sp, #(10 * 16)]
- add sp, sp, #(11*16)
- ret
-
- EPILOGUE
-
|