|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814 |
- /***************************************************************************
- Copyright (c) 2013-2020, 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 MY_ALIGN .align 3
- b CGEMM_L4
- /* MINI SUBROUTINES */
- /* 4x8 MAIN 128x+2 LOOP */
-
-
- CGEMM_L4x8_LMAIN_SUB:
- /*----------------------------------------*/
- mtctr T8
- LOAD4x8_2
- MY_ALIGN
- CGEMM_L4x8_LOOP:
- /*----------------------------------------*/
- dcbt AO, PRE
- dcbt BO, PRE
- KERNEL4x8_L2 128,64,0,0
- CGEMM_L4x8_K128:
- /*----------------------------------------*/
- KERNEL4x8_L2 128,64,1,0
- dcbt AO, T2
- KERNEL4x8_L2 128,64,2,0
- KERNEL4x8_L2 128,64,3,0
- dcbt AO, T3
- dcbt BO, T2
- KERNEL4x8_L2 128,64,4,0
- KERNEL4x8_L2 128,64,5,0
- dcbt AO, T4
- KERNEL4x8_L2 128,64,6,0
- KERNEL4x8_L2 128,64,7,0
- dcbt AO, T5
- dcbt BO, T3
- KERNEL4x8_L2 128,64,8,0
- KERNEL4x8_L2 128,64,9,0
- KERNEL4x8_L2 128,64,10,0
- KERNEL4x8_L2 128,64,11,0
- dcbt BO, T4
- KERNEL4x8_L2 128,64,12,0
- KERNEL4x8_L2 128,64,13,0
- KERNEL4x8_L2 128,64,14,0
- KERNEL4x8_L2 128,64,15,0
- KERNEL4x8_L2 128,64,16,0
- KERNEL4x8_L2 128,64,17,0
- KERNEL4x8_L2 128,64,18,0
- KERNEL4x8_L2 128,64,19,0
- KERNEL4x8_L2 128,64,20,0
- KERNEL4x8_L2 128,64,21,0
- KERNEL4x8_L2 128,64,22,0
- KERNEL4x8_L2 128,64,23,0
- KERNEL4x8_L2 128,64,24,0
- KERNEL4x8_L2 128,64,25,0
- KERNEL4x8_L2 128,64,26,0
- KERNEL4x8_L2 128,64,27,0
- KERNEL4x8_L2 128,64,28,0
- KERNEL4x8_L2 128,64,29,0
- KERNEL4x8_L2 128,64,30,0
- KERNEL4x8_L2 128,64,31,0
- KERNEL4x8_L2 128,64,32,0
- KERNEL4x8_L2 128,64,33,0
- KERNEL4x8_L2 128,64,34,0
- KERNEL4x8_L2 128,64,35,0
- KERNEL4x8_L2 128,64,36,0
- KERNEL4x8_L2 128,64,37,0
- KERNEL4x8_L2 128,64,38,0
- KERNEL4x8_L2 128,64,39,0
- KERNEL4x8_L2 128,64,40,0
- KERNEL4x8_L2 128,64,41,0
- KERNEL4x8_L2 128,64,42,0
- KERNEL4x8_L2 128,64,43,0
- KERNEL4x8_L2 128,64,44,0
- KERNEL4x8_L2 128,64,45,0
- KERNEL4x8_L2 128,64,46,0
- KERNEL4x8_L2 128,64,47,0
- KERNEL4x8_L2 128,64,48,0
- KERNEL4x8_L2 128,64,49,0
- KERNEL4x8_L2 128,64,50,0
- KERNEL4x8_L2 128,64,51,0
- KERNEL4x8_L2 128,64,52,0
- KERNEL4x8_L2 128,64,53,0
- KERNEL4x8_L2 128,64,54,0
- KERNEL4x8_L2 128,64,55,0
- KERNEL4x8_L2 128,64,56,0
- KERNEL4x8_L2 128,64,57,0
- KERNEL4x8_L2 128,64,58,0
- KERNEL4x8_L2 128,64,59,0
- KERNEL4x8_L2 128,64,60,0
- KERNEL4x8_L2 128,64,61,0
- KERNEL4x8_L2 128,64,62,0
- KERNEL4x8_L2 128,64,63,1
- bdnz CGEMM_L4x8_LOOP
- MY_ALIGN
- CGEMM_L4x8_LOOP_END:
- /*----------------------------------------*/
- END4x8_2
- blr
- MY_ALIGN
-
-
- CGEMM_4x8_L64_SUB:
- /*----------------------------------------*/
- LOAD4x8_2
- dcbt AO, PRE
- dcbt BO, PRE
- KERNEL4x8_L2 128,64,0,0
- KERNEL4x8_L2 128,64,1,0
- dcbt AO, T2
- KERNEL4x8_L2 128,64,2,0
- KERNEL4x8_L2 128,64,3,0
- dcbt AO, T3
- dcbt BO, T2
- KERNEL4x8_L2 128,64,4,0
- KERNEL4x8_L2 128,64,5,0
- dcbt AO, T4
- KERNEL4x8_L2 128,64,6,0
- KERNEL4x8_L2 128,64,7,0
- dcbt AO, T5
- dcbt BO, T3
- KERNEL4x8_L2 128,64,8,0
- KERNEL4x8_L2 128,64,9,0
- KERNEL4x8_L2 128,64,10,0
- KERNEL4x8_L2 128,64,11,0
- dcbt BO, T4
- KERNEL4x8_L2 128,64,12,0
- KERNEL4x8_L2 128,64,13,0
- KERNEL4x8_L2 128,64,14,0
- KERNEL4x8_L2 128,64,15,0
- KERNEL4x8_L2 128,64,16,0
- KERNEL4x8_L2 128,64,17,0
- KERNEL4x8_L2 128,64,18,0
- KERNEL4x8_L2 128,64,19,0
- KERNEL4x8_L2 128,64,20,0
- KERNEL4x8_L2 128,64,21,0
- KERNEL4x8_L2 128,64,22,0
- KERNEL4x8_L2 128,64,23,0
- KERNEL4x8_L2 128,64,24,0
- KERNEL4x8_L2 128,64,25,0
- KERNEL4x8_L2 128,64,26,0
- KERNEL4x8_L2 128,64,27,0
- KERNEL4x8_L2 128,64,28,0
- KERNEL4x8_L2 128,64,29,0
- KERNEL4x8_L2 128,64,30,0
- KERNEL4x8_E2 128,64,31,1
- blr
- MY_ALIGN
-
-
- CGEMM_4x8_L32_SUB:
- /*----------------------------------------*/
- LOAD4x8_2
- dcbt AO, PRE
- dcbt BO, PRE
- KERNEL4x8_L2 128,64,0,0
- KERNEL4x8_L2 128,64,1,0
- dcbt AO, T2
- KERNEL4x8_L2 128,64,2,0
- KERNEL4x8_L2 128,64,3,0
- dcbt AO, T3
- dcbt BO, T2
- KERNEL4x8_L2 128,64,4,0
- KERNEL4x8_L2 128,64,5,0
- dcbt AO, T4
- KERNEL4x8_L2 128,64,6,0
- KERNEL4x8_L2 128,64,7,0
- dcbt AO, T5
- dcbt BO, T3
- KERNEL4x8_L2 128,64,8,0
- KERNEL4x8_L2 128,64,9,0
- KERNEL4x8_L2 128,64,10,0
- KERNEL4x8_L2 128,64,11,0
- dcbt BO, T4
- KERNEL4x8_L2 128,64,12,0
- KERNEL4x8_L2 128,64,13,0
- KERNEL4x8_L2 128,64,14,0
- KERNEL4x8_E2 128,64,15,1
- blr
- MY_ALIGN
-
-
- CGEMM_4x8_L16_SUB:
- /*----------------------------------------*/
- LOAD4x8_2
- dcbt AO, PRE
- dcbt BO, PRE
- KERNEL4x8_L2 128,64,0,0
- KERNEL4x8_L2 128,64,1,0
- dcbt AO, T2
- KERNEL4x8_L2 128,64,2,0
- KERNEL4x8_L2 128,64,3,0
- dcbt AO, T3
- dcbt BO, T2
- KERNEL4x8_L2 128,64,4,0
- KERNEL4x8_L2 128,64,5,0
- dcbt AO, T4
- KERNEL4x8_L2 128,64,6,0
- KERNEL4x8_E2 128,64,7,1
- blr
- MY_ALIGN
-
-
- CGEMM_4x4_LMAIN_SUB:
- /*----------------------------------------*/
- mtctr T8
- LOAD4x4_2
- MY_ALIGN
- CGEMM_L4x4_LOOP:
- /*----------------------------------------*/
- KERNEL4x4_L2 64,64,0,0
- CGEMM_L4x4_K32:
- /*----------------------------------------*/
- KERNEL4x4_L2 64,64,1,0
- KERNEL4x4_L2 64,64,2,0
- KERNEL4x4_L2 64,64,3,0
- KERNEL4x4_L2 64,64,4,0
- KERNEL4x4_L2 64,64,5,0
- KERNEL4x4_L2 64,64,6,0
- KERNEL4x4_L2 64,64,7,0
- KERNEL4x4_L2 64,64,8,0
- KERNEL4x4_L2 64,64,9,0
- KERNEL4x4_L2 64,64,10,0
- KERNEL4x4_L2 64,64,11,0
- KERNEL4x4_L2 64,64,12,0
- KERNEL4x4_L2 64,64,13,0
- KERNEL4x4_L2 64,64,14,0
- KERNEL4x4_L2 64,64,15,1
- bdnz CGEMM_L4x4_LOOP
- MY_ALIGN
- CGEMM_L4x4_LOOP_END:
- /*----------------------------------------*/
- END4x4_2
- blr
- MY_ALIGN
-
-
- CGEMM_4x4_L16_SUB:
- /*----------------------------------------*/
- LOAD4x4_2
- KERNEL4x4_L2 64,64,0,0
- KERNEL4x4_L2 64,64,1,0
- KERNEL4x4_L2 64,64,2,0
- KERNEL4x4_L2 64,64,3,0
- KERNEL4x4_L2 64,64,4,0
- KERNEL4x4_L2 64,64,5,0
- KERNEL4x4_L2 64,64,6,0
- KERNEL4x4_E2 64,64,7,1
- blr
- MY_ALIGN
-
-
- CGEMM_4x4_L8_SUB:
- /*----------------------------------------*/
- LOAD4x4_2
- KERNEL4x4_L2 64,64,0,0
- KERNEL4x4_L2 64,64,1,0
- KERNEL4x4_L2 64,64,2,0
- KERNEL4x4_E2 64,64,3,1
- blr
-
-
- CGEMM_4x2_LMAIN_SUB:
- /*----------------------------------------*/
- mtctr T8
- LOAD4x2_2
- MY_ALIGN
- CGEMM_L4x2_LOOP:
- /*----------------------------------------*/
- KERNEL4x2_L2 32,64,0,0
- CGEMM_L4x2_K32:
- /*----------------------------------------*/
- KERNEL4x2_L2 32,64,1,0
- KERNEL4x2_L2 32,64,2,0
- KERNEL4x2_L2 32,64,3,0
- KERNEL4x2_L2 32,64,4,0
- KERNEL4x2_L2 32,64,5,0
- KERNEL4x2_L2 32,64,6,0
- KERNEL4x2_L2 32,64,7,0
- KERNEL4x2_L2 32,64,8,0
- KERNEL4x2_L2 32,64,9,0
- KERNEL4x2_L2 32,64,10,0
- KERNEL4x2_L2 32,64,11,0
- KERNEL4x2_L2 32,64,12,0
- KERNEL4x2_L2 32,64,13,0
- KERNEL4x2_L2 32,64,14,0
- KERNEL4x2_L2 32,64,15,1
- bdnz CGEMM_L4x2_LOOP
- MY_ALIGN
-
-
- CGEMM_L4x2_LOOP_END:
- /*----------------------------------------*/
- END4x2_2
- blr
- MY_ALIGN
- CGEMM_4x2_L16_SUB:
- /*----------------------------------------*/
- LOAD4x2_2
- KERNEL4x2_L2 32,64,0,0
- KERNEL4x2_L2 32,64,1,0
- KERNEL4x2_L2 32,64,2,0
- KERNEL4x2_L2 32,64,3,0
- KERNEL4x2_L2 32,64,4,0
- KERNEL4x2_L2 32,64,5,0
- KERNEL4x2_L2 32,64,6,0
- KERNEL4x2_E2 32,64,7,1
- blr
- MY_ALIGN
- CGEMM_4x2_L8_SUB:
- /*----------------------------------------*/
- LOAD4x2_2
- KERNEL4x2_L2 32,64,0,0
- KERNEL4x2_L2 32,64,1,0
- KERNEL4x2_L2 32,64,2,0
- KERNEL4x2_E2 32,64,3,1
- blr
-
-
- CGEMM_4x1_LMAIN_SUB:
- /*----------------------------------------*/
- mtctr T8
- LOAD4x1_2
- MY_ALIGN
- CGEMM_L4x1_LOOP:
- /*----------------------------------------*/
- KERNEL4x1_L2 16,64,0,0
- CGEMM_L4x1_K32:
- /*----------------------------------------*/
- KERNEL4x1_L2 16,64,1,0
- KERNEL4x1_L2 16,64,2,0
- KERNEL4x1_L2 16,64,3,0
- KERNEL4x1_L2 16,64,4,0
- KERNEL4x1_L2 16,64,5,0
- KERNEL4x1_L2 16,64,6,0
- KERNEL4x1_L2 16,64,7,0
- KERNEL4x1_L2 16,64,8,0
- KERNEL4x1_L2 16,64,9,0
- KERNEL4x1_L2 16,64,10,0
- KERNEL4x1_L2 16,64,11,0
- KERNEL4x1_L2 16,64,12,0
- KERNEL4x1_L2 16,64,13,0
- KERNEL4x1_L2 16,64,14,0
- KERNEL4x1_L2 16,64,15,1
- bdnz CGEMM_L4x1_LOOP
- MY_ALIGN
- CGEMM_L4x1_LOOP_END:
- /*----------------------------------------*/
- END4x1_2
- blr
-
- MY_ALIGN
- CGEMM_4x1_L16_SUB:
- /*----------------------------------------*/
- LOAD4x1_2
- KERNEL4x1_L2 16,64,0,0
- KERNEL4x1_L2 16,64,1,0
- KERNEL4x1_L2 16,64,2,0
- KERNEL4x1_L2 16,64,3,0
- KERNEL4x1_L2 16,64,4,0
- KERNEL4x1_L2 16,64,5,0
- KERNEL4x1_L2 16,64,6,0
- KERNEL4x1_E2 16,64,7,1
- blr
- MY_ALIGN
-
-
- CGEMM_4x1_L8_SUB:
- /*----------------------------------------*/
- LOAD4x1_2
- KERNEL4x1_L2 16,64,0,0
- KERNEL4x1_L2 16,64,1,0
- KERNEL4x1_L2 16,64,2,0
- KERNEL4x1_E2 16,64,3,1
- blr
-
-
-
- /* MAIN LOOP BEGINS */
- MY_ALIGN
-
-
- CGEMM_L4:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL) && !defined(LEFT)
- neg TEMP_REG, OFFSET
- #endif
- /* Pre set value in vs57 as 0xffff0000ffff0000 for masking */
- vspltisb v24, -1
- vspltisb v25, 0
- xxsldwi vs57, vs56, vs57, 1
- xxpermdi vs57, vs57, vs57, 3
- srawi. J, N, 2
- ble CGEMM_L4_END
-
-
- CGEMM_L4_BEGIN:
- /*----------------------------------------*/
- mr CO, C
- slwi T1, LDC , 2
- add T2,C,LDC
- mr AO, A
- add C, C, T1
- #if defined(TRMMKERNEL) && defined(LEFT)
- mr TEMP_REG, OFFSET /*off = offset;*/
- #endif
- srawi. I, M, 3
- ble CGEMM_L4x8_END
- dcbt CO,r0 /*just prefetch*/
- dcbt T2,r0
-
-
- CGEMM_L4x8_BEGIN:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- REFRESH_POINTERS AO,BO,TEMP_REG,B,8,4
- #else
- mr BO, B
- dcbt B, r0
- #endif
- dcbt AO, r0
- #if defined(TRMMKERNEL)
- REFRESH_TEMP_BK T6,K,TEMP_REG,8,4
- mr T1, T6
- /* TEMPS FOR PREFETCH */
- li T2, 1024
- li T3, 1024+512
- addi T1,T1, -2
- /* TEMPS FOR PREFETCH */
- li T4, 2048
- li T5, 2048+512
- srawi. T8, T1, 7 /**(T1-2) % 128x */
- #else
- mr T1, K
- /* TEMPS FOR PREFETCH */
- li T2, 1024
- li T3, 1024+512
- addi T1,T1, -2
- /* TEMPS FOR PREFETCH */
- li T4, 2048
- li T5, 2048+512
- srawi. T8, T1, 7 /**(K-2) % 128x */
- #endif
- ZERO4x8
- ble CGEMM_L4x8_SUB0
- bl CGEMM_L4x8_LMAIN_SUB
- andi. L, T1, 127
- ble CGEMM_L4x8_SAVE
- b CGEMM_L4x8_SUB2
-
-
- CGEMM_L4x8_SUB0:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- andi. L, T6, 255
- cmpwi T6,129
- #else
- andi. L, K, 255
- cmpwi K,129
- #endif
- li T8,1
- bne CMP4x8_128K
- addi BO,BO,-32
- addi AO,AO,-64
- LOAD4x8O 64,32
- END4x8_WITHOUT_ADD
- LOAD4x8_2O 128, 64
- mtctr T8
- bl CGEMM_L4x8_K128
- b CGEMM_L4x8_SAVE
- CMP4x8_128K:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- cmpwi T6,128
- #else
- cmpwi K,128
- #endif
- bne CGEMM_L4x8_SUB2
- MY_ALIGN
- mtctr T8
- addi BO,BO,-64
- addi AO,AO,-128
- LOAD4x8_2O 128,64
- bl CGEMM_L4x8_K128
- b CGEMM_L4x8_SAVE
- MY_ALIGN
-
-
- CGEMM_L4x8_SUB2:
- /*----------------------------------------*/
- andi. T1,L, 64
- ble CGEMM_L4x8_SUB2_32
- bl CGEMM_4x8_L64_SUB
- MY_ALIGN
-
-
- CGEMM_L4x8_SUB2_32:
- /*----------------------------------------*/
- andi. T1,L, 32
- ble CGEMM_L4x8_SUB2_16
- bl CGEMM_4x8_L32_SUB
- MY_ALIGN
-
-
- CGEMM_L4x8_SUB2_16:
- /*----------------------------------------*/
- andi. T1,L, 16
- ble CGEMM_L4x8_SUB2_8
- bl CGEMM_4x8_L16_SUB
- MY_ALIGN
-
-
- CGEMM_L4x8_SUB2_8:
- /*----------------------------------------*/
- andi. T1,L, 8
- ble CGEMM_L4x8_SUB2_4
- LOAD4x8_2
- KERNEL4x8_L2 128,64, 0,0
- KERNEL4x8_L2 128,64, 1,0
- KERNEL4x8_L2 128,64, 2,0
- KERNEL4x8_E2 128,64, 3,1
- MY_ALIGN
-
-
- CGEMM_L4x8_SUB2_4:
- /*----------------------------------------*/
- andi. T1,L, 4
- ble CGEMM_L4x8_SUB2_2
- LOAD4x8_2
- KERNEL4x8_L2 128,64, 0,0
- KERNEL4x8_E2 128,64, 1,1
- MY_ALIGN
-
-
- CGEMM_L4x8_SUB2_2:
- /*----------------------------------------*/
- andi. T1,L, 2
- ble CGEMM_L4x8_SUB2_1
- LOAD4x8_2
- KERNEL4x8_E2 128,64, 0,1
- MY_ALIGN
-
-
- CGEMM_L4x8_SUB2_1:
- /*----------------------------------------*/
- andi. T1,L, 1
- ble CGEMM_L4x8_SAVE
- KERNEL4x8
-
- MY_ALIGN
- CGEMM_L4x8_SAVE:
- /*----------------------------------------*/
- addic. I, I, -1
- MY_ALIGN
- SAVE4x8
- #if defined(TRMMKERNEL)
- REFRESH_AFTER_SAVE T6,K,TEMP_REG,BO,AO,8,4
- #endif
- bgt CGEMM_L4x8_BEGIN
- andi. T2, M, 7
- ble CGEMM_L4x1_END
- andi. T1, M, 4
- ble CGEMM_L4x4_END
- b CGEMM_L4x4_BEGIN
- MY_ALIGN
-
-
- CGEMM_L4x8_END:
- /*----------------------------------------*/
-
-
- CGEMM_L4x4_BEGIN:
- /*----------------------------------------*/
- andi. T2, M, 7
- ble CGEMM_L4x1_END
- andi. T1, M, 4
- ble CGEMM_L4x4_END
- #if defined(TRMMKERNEL)
- REFRESH_POINTERS AO,BO,TEMP_REG,B,4,4
- #else
- mr BO, B
- #endif
- #if defined(TRMMKERNEL)
- REFRESH_TEMP_BK T6,K,TEMP_REG,4,4
- mr T1, T6
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(T1-2) % 32x */
- #else
- mr T1, K
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(K-2) % 32x */
- #endif
- ZERO4x4
- ble CGEMM_L4x4_SUB0
- bl CGEMM_4x4_LMAIN_SUB
- andi. L, T1, 31
- ble CGEMM_L4x4_SAVE
- b CGEMM_L4x4_SUB2
-
-
- CGEMM_L4x4_SUB0:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- andi. L, T6, 63
- cmpwi T6,33
- #else
- andi. L, K, 63
- cmpwi K,33
- #endif
- li T8,1
- bne CMP4x4_32K
- addi BO,BO,-32
- addi AO,AO,-32
- LOAD4x4O 32,32
- END4x4_WITHOUT_ADD
- LOAD4x4_2O 64, 64
- mtctr T8
- bl CGEMM_L4x4_K32
- b CGEMM_L4x4_SAVE
- CMP4x4_32K:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- cmpwi T6,32
- #else
- cmpwi K,32
- #endif
- bne CGEMM_L4x4_SUB2
- MY_ALIGN
- mtctr T8
- addi BO,BO,-64
- addi AO,AO,-64
- LOAD4x4_2O 64,64
- bl CGEMM_L4x4_K32
- b CGEMM_L4x4_SAVE
- MY_ALIGN
- MY_ALIGN
-
-
- CGEMM_L4x4_SUB2:
- /*----------------------------------------*/
- andi. T1,L, 16
- ble CGEMM_L4x4_SUB2_8
- bl CGEMM_4x4_L16_SUB
- MY_ALIGN
-
-
- CGEMM_L4x4_SUB2_8:
- /*----------------------------------------*/
- andi. T1,L, 8
- ble CGEMM_L4x4_SUB2_4
- bl CGEMM_4x4_L8_SUB
- MY_ALIGN
-
-
- CGEMM_L4x4_SUB2_4:
- /*----------------------------------------*/
- andi. T1,L, 4
- ble CGEMM_L4x4_SUB2_2
- LOAD4x4_2
- KERNEL4x4_L2 64,64, 0,0
- KERNEL4x4_E2 64,64, 1,1
- MY_ALIGN
-
-
- CGEMM_L4x4_SUB2_2:
- /*----------------------------------------*/
- andi. T1,L, 2
- ble CGEMM_L4x4_SUB2_1
- LOAD4x4_2
- KERNEL4x4_E2 64,64, 0,1
- MY_ALIGN
-
-
- CGEMM_L4x4_SUB2_1:
- /*----------------------------------------*/
- andi. T1,L, 1
- ble CGEMM_L4x4_SAVE
- KERNEL4x4
-
-
- CGEMM_L4x4_SAVE:
- /*----------------------------------------*/
- SAVE4x4
- #if defined(TRMMKERNEL)
- REFRESH_AFTER_SAVE T6,K,TEMP_REG,BO,AO,4,4
- #endif
-
-
- CGEMM_L4x4_END:
- /*----------------------------------------*/
-
-
- CGEMM_L4x2_BEGIN:
- /*----------------------------------------*/
- andi. T1, M, 2
- ble CGEMM_L4x2_END
- #if defined(TRMMKERNEL)
- REFRESH_POINTERS AO,BO,TEMP_REG,B,2,4
- #else
- mr BO, B
- #endif
- #if defined(TRMMKERNEL)
- REFRESH_TEMP_BK T6,K,TEMP_REG,2,4
- mr T1, T6
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(T1-2) % 32x */
- #else
- mr T1, K
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(K-2) % 32x */
- #endif
- ZERO4x2
- ble CGEMM_L4x2_SUB0
- bl CGEMM_4x2_LMAIN_SUB
- andi. L, T1, 31
- ble CGEMM_L4x2_SAVE
- b CGEMM_L4x2_SUB2
-
-
- CGEMM_L4x2_SUB0:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- andi. L, T6, 63
- cmpwi T6,33
- #else
- andi. L, K, 63
- cmpwi K,33
- #endif
- li T8,1
- bne CMP4x2_32K
- addi BO,BO,-32
- addi AO,AO,-16
- LOAD4x2O 16,32
- END4x2_WITHOUT_ADD
- LOAD4x2_2O 32, 64
- mtctr T8
- bl CGEMM_L4x2_K32
- b CGEMM_L4x2_SAVE
- CMP4x2_32K:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- cmpwi T6,32
- #else
- cmpwi K,32
- #endif
- bne CGEMM_L4x2_SUB2
- MY_ALIGN
- mtctr T8
- addi BO,BO,-64
- addi AO,AO,-32
- LOAD4x2_2O 32,64
- bl CGEMM_L4x2_K32
- b CGEMM_L4x2_SAVE
- MY_ALIGN
- MY_ALIGN
-
-
- CGEMM_L4x2_SUB2:
- /*----------------------------------------*/
- andi. T1,L, 16
- ble CGEMM_L4x2_SUB2_8
- bl CGEMM_4x2_L16_SUB
- MY_ALIGN
-
-
- CGEMM_L4x2_SUB2_8:
- /*----------------------------------------*/
- andi. T1,L, 8
- ble CGEMM_L4x2_SUB2_4
- bl CGEMM_4x2_L8_SUB
- MY_ALIGN
-
-
- CGEMM_L4x2_SUB2_4:
- /*----------------------------------------*/
- andi. T1,L, 4
- ble CGEMM_L4x2_SUB2_2
- LOAD4x2_2
- KERNEL4x2_L2 32,64, 0,0
- KERNEL4x2_E2 32,64, 1,1
- MY_ALIGN
-
-
- CGEMM_L4x2_SUB2_2:
- /*----------------------------------------*/
- andi. T1,L, 2
- ble CGEMM_L4x2_SUB2_1
- LOAD4x2_2
- KERNEL4x2_E2 32,64, 0,1
- MY_ALIGN
-
-
- CGEMM_L4x2_SUB2_1:
- /*----------------------------------------*/
- andi. T1,L, 1
- ble CGEMM_L4x2_SAVE
- KERNEL4x2
-
- MY_ALIGN
- CGEMM_L4x2_SAVE:
- /*----------------------------------------*/
- SAVE4x2
- #if defined(TRMMKERNEL)
- REFRESH_AFTER_SAVE T6,K,TEMP_REG,BO,AO,2,4
- #endif
-
-
- CGEMM_L4x2_END:
- /*----------------------------------------*/
-
-
- CGEMM_L4x1_BEGIN:
- /*----------------------------------------*/
- andi. T1, M, 1
- ble CGEMM_L4x1_END
- #if defined(TRMMKERNEL)
- REFRESH_POINTERS AO,BO,TEMP_REG,B,1,4
- #else
- mr BO, B
- #endif
- #if defined(TRMMKERNEL)
- REFRESH_TEMP_BK T6,K,TEMP_REG,1,4
- mr T1, T6
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(T1-2) % 32x */
- #else
- mr T1, K
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(K-2) % 32x */
- #endif
- ZERO4x1
- ble CGEMM_L4x1_SUB0
- bl CGEMM_4x1_LMAIN_SUB
- andi. L, T1, 31
- ble CGEMM_L4x1_SAVE
- b CGEMM_L4x1_SUB2
-
-
- CGEMM_L4x1_SUB0:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- andi. L, T6, 63
- cmpwi T6,33
- #else
- andi. L, K, 63
- cmpwi K,33
- #endif
- li T8,1
- bne CMP4x1_32K
- addi BO,BO,-32
- addi AO,AO,-8
- LOAD4x1O 8,32
- END4x1_WITHOUT_ADD
- LOAD4x1_2O 16, 64
- mtctr T8
- bl CGEMM_L4x1_K32
- b CGEMM_L4x1_SAVE
- CMP4x1_32K:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- cmpwi T6,32
- #else
- cmpwi K,32
- #endif
- bne CGEMM_L4x1_SUB2
- MY_ALIGN
- mtctr T8
- addi BO,BO,-64
- addi AO,AO,-16
- LOAD4x1_2O 16,64
- bl CGEMM_L4x1_K32
- b CGEMM_L4x1_SAVE
- MY_ALIGN
- MY_ALIGN
-
-
- CGEMM_L4x1_SUB2:
- /*----------------------------------------*/
- andi. T1,L, 16
- ble CGEMM_L4x1_SUB2_8
- bl CGEMM_4x1_L16_SUB
- MY_ALIGN
-
-
- CGEMM_L4x1_SUB2_8:
- /*----------------------------------------*/
- andi. T1,L, 8
- ble CGEMM_L4x1_SUB2_4
- bl CGEMM_4x1_L8_SUB
- MY_ALIGN
-
-
- CGEMM_L4x1_SUB2_4:
- /*----------------------------------------*/
- andi. T1,L, 4
- ble CGEMM_L4x1_SUB2_2
- LOAD4x1_2
- KERNEL4x1_L2 16,64, 0,0
- KERNEL4x1_E2 16,64, 1,1
- MY_ALIGN
-
-
- CGEMM_L4x1_SUB2_2:
- /*----------------------------------------*/
- andi. T1,L, 2
- ble CGEMM_L4x1_SUB2_1
- LOAD4x1_2
- KERNEL4x1_E2 16,64, 0,1
- MY_ALIGN
-
-
- CGEMM_L4x1_SUB2_1:
- /*----------------------------------------*/
- andi. T1,L, 1
- ble CGEMM_L4x1_SAVE
- KERNEL4x1
-
- MY_ALIGN
- CGEMM_L4x1_SAVE:
- /*----------------------------------------*/
-
- SAVE4x1
- #if defined(TRMMKERNEL)
- REFRESH_AFTER_SAVE T6,K,TEMP_REG,BO,AO,1,4
- #endif
-
-
- CGEMM_L4x1_END:
- /*----------------------------------------*/
- slwi T1, K, 5
- addic. J, J, -1
- add B, B, T1
- #if defined(TRMMKERNEL) && !defined(LEFT)
- addi TEMP_REG, TEMP_REG, 4
- #endif
- bgt CGEMM_L4_BEGIN
-
-
- CGEMM_L4_END:
-
- b CGEMM_L2
- /* MINI SUBROUTINES */
- /* 2x8 MAIN 128x+2 LOOP */
-
-
- CGEMM_L2x8_LMAIN_SUB:
- /*----------------------------------------*/
- mtctr T8
- LOAD2x8_2
- MY_ALIGN
- CGEMM_L2x8_LOOP:
- /*----------------------------------------*/
- dcbt AO, PRE
- dcbt BO, PRE
- KERNEL2x8_L2 128,32,0,0
- CGEMM_L2x8_K128:
- /*----------------------------------------*/
- KERNEL2x8_L2 128,32,1,0
- dcbt AO, T2
- KERNEL2x8_L2 128,32,2,0
- KERNEL2x8_L2 128,32,3,0
- dcbt AO, T3
- dcbt BO, T2
- KERNEL2x8_L2 128,32,4,0
- KERNEL2x8_L2 128,32,5,0
- dcbt AO, T4
- KERNEL2x8_L2 128,32,6,0
- KERNEL2x8_L2 128,32,7,0
- dcbt AO, T5
- dcbt BO, T3
- KERNEL2x8_L2 128,32,8,0
- KERNEL2x8_L2 128,32,9,0
- KERNEL2x8_L2 128,32,10,0
- KERNEL2x8_L2 128,32,11,0
- dcbt BO, T4
- KERNEL2x8_L2 128,32,12,0
- KERNEL2x8_L2 128,32,13,0
- KERNEL2x8_L2 128,32,14,0
- KERNEL2x8_L2 128,32,15,0
- KERNEL2x8_L2 128,32,16,0
- KERNEL2x8_L2 128,32,17,0
- KERNEL2x8_L2 128,32,18,0
- KERNEL2x8_L2 128,32,19,0
- KERNEL2x8_L2 128,32,20,0
- KERNEL2x8_L2 128,32,21,0
- KERNEL2x8_L2 128,32,22,0
- KERNEL2x8_L2 128,32,23,0
- KERNEL2x8_L2 128,32,24,0
- KERNEL2x8_L2 128,32,25,0
- KERNEL2x8_L2 128,32,26,0
- KERNEL2x8_L2 128,32,27,0
- KERNEL2x8_L2 128,32,28,0
- KERNEL2x8_L2 128,32,29,0
- KERNEL2x8_L2 128,32,30,0
- KERNEL2x8_L2 128,32,31,0
- KERNEL2x8_L2 128,32,32,0
- KERNEL2x8_L2 128,32,33,0
- KERNEL2x8_L2 128,32,34,0
- KERNEL2x8_L2 128,32,35,0
- KERNEL2x8_L2 128,32,36,0
- KERNEL2x8_L2 128,32,37,0
- KERNEL2x8_L2 128,32,38,0
- KERNEL2x8_L2 128,32,39,0
- KERNEL2x8_L2 128,32,40,0
- KERNEL2x8_L2 128,32,41,0
- KERNEL2x8_L2 128,32,42,0
- KERNEL2x8_L2 128,32,43,0
- KERNEL2x8_L2 128,32,44,0
- KERNEL2x8_L2 128,32,45,0
- KERNEL2x8_L2 128,32,46,0
- KERNEL2x8_L2 128,32,47,0
- KERNEL2x8_L2 128,32,48,0
- KERNEL2x8_L2 128,32,49,0
- KERNEL2x8_L2 128,32,50,0
- KERNEL2x8_L2 128,32,51,0
- KERNEL2x8_L2 128,32,52,0
- KERNEL2x8_L2 128,32,53,0
- KERNEL2x8_L2 128,32,54,0
- KERNEL2x8_L2 128,32,55,0
- KERNEL2x8_L2 128,32,56,0
- KERNEL2x8_L2 128,32,57,0
- KERNEL2x8_L2 128,32,58,0
- KERNEL2x8_L2 128,32,59,0
- KERNEL2x8_L2 128,32,60,0
- KERNEL2x8_L2 128,32,61,0
- KERNEL2x8_L2 128,32,62,0
- KERNEL2x8_L2 128,32,63,1
- bdnz CGEMM_L2x8_LOOP
- MY_ALIGN
- CGEMM_L2x8_LOOP_END:
- /*----------------------------------------*/
- END2x8_2
- blr
- MY_ALIGN
-
-
- CGEMM_2x8_L64_SUB:
- /*----------------------------------------*/
- LOAD2x8_2
- dcbt AO, PRE
- dcbt BO, PRE
- KERNEL2x8_L2 128,32,0,0
- KERNEL2x8_L2 128,32,1,0
- dcbt AO, T2
- KERNEL2x8_L2 128,32,2,0
- KERNEL2x8_L2 128,32,3,0
- dcbt AO, T3
- dcbt BO, T2
- KERNEL2x8_L2 128,32,4,0
- KERNEL2x8_L2 128,32,5,0
- dcbt AO, T4
- KERNEL2x8_L2 128,32,6,0
- KERNEL2x8_L2 128,32,7,0
- dcbt AO, T5
- dcbt BO, T3
- KERNEL2x8_L2 128,32,8,0
- KERNEL2x8_L2 128,32,9,0
- KERNEL2x8_L2 128,32,10,0
- KERNEL2x8_L2 128,32,11,0
- dcbt BO, T4
- KERNEL2x8_L2 128,32,12,0
- KERNEL2x8_L2 128,32,13,0
- KERNEL2x8_L2 128,32,14,0
- KERNEL2x8_L2 128,32,15,0
- KERNEL2x8_L2 128,32,16,0
- KERNEL2x8_L2 128,32,17,0
- KERNEL2x8_L2 128,32,18,0
- KERNEL2x8_L2 128,32,19,0
- KERNEL2x8_L2 128,32,20,0
- KERNEL2x8_L2 128,32,21,0
- KERNEL2x8_L2 128,32,22,0
- KERNEL2x8_L2 128,32,23,0
- KERNEL2x8_L2 128,32,24,0
- KERNEL2x8_L2 128,32,25,0
- KERNEL2x8_L2 128,32,26,0
- KERNEL2x8_L2 128,32,27,0
- KERNEL2x8_L2 128,32,28,0
- KERNEL2x8_L2 128,32,29,0
- KERNEL2x8_L2 128,32,30,0
- KERNEL2x8_E2 128,32,31,1
- blr
- MY_ALIGN
-
-
- CGEMM_2x8_L32_SUB:
- /*----------------------------------------*/
- LOAD2x8_2
- dcbt AO, PRE
- dcbt BO, PRE
- KERNEL2x8_L2 128,32,0,0
- KERNEL2x8_L2 128,32,1,0
- dcbt AO, T2
- KERNEL2x8_L2 128,32,2,0
- KERNEL2x8_L2 128,32,3,0
- dcbt AO, T3
- dcbt BO, T2
- KERNEL2x8_L2 128,32,4,0
- KERNEL2x8_L2 128,32,5,0
- dcbt AO, T4
- KERNEL2x8_L2 128,32,6,0
- KERNEL2x8_L2 128,32,7,0
- dcbt AO, T5
- dcbt BO, T3
- KERNEL2x8_L2 128,32,8,0
- KERNEL2x8_L2 128,32,9,0
- KERNEL2x8_L2 128,32,10,0
- KERNEL2x8_L2 128,32,11,0
- dcbt BO, T4
- KERNEL2x8_L2 128,32,12,0
- KERNEL2x8_L2 128,32,13,0
- KERNEL2x8_L2 128,32,14,0
- KERNEL2x8_E2 128,32,15,1
- blr
- MY_ALIGN
-
-
- CGEMM_2x8_L16_SUB:
- /*----------------------------------------*/
- LOAD2x8_2
- dcbt AO, PRE
- dcbt BO, PRE
- KERNEL2x8_L2 128,32,0,0
- KERNEL2x8_L2 128,32,1,0
- dcbt AO, T2
- KERNEL2x8_L2 128,32,2,0
- KERNEL2x8_L2 128,32,3,0
- dcbt AO, T3
- dcbt BO, T2
- KERNEL2x8_L2 128,32,4,0
- KERNEL2x8_L2 128,32,5,0
- dcbt AO, T4
- KERNEL2x8_L2 128,32,6,0
- KERNEL2x8_E2 128,32,7,1
- blr
- MY_ALIGN
-
-
- CGEMM_2x4_LMAIN_SUB:
- /*----------------------------------------*/
- mtctr T8
- LOAD2x4_2
- MY_ALIGN
- CGEMM_L2x4_LOOP:
- /*----------------------------------------*/
- KERNEL2x4_L2 64,32,0,0
- CGEMM_L2x4_K32:
- /*----------------------------------------*/
- KERNEL2x4_L2 64,32,1,0
- KERNEL2x4_L2 64,32,2,0
- KERNEL2x4_L2 64,32,3,0
- KERNEL2x4_L2 64,32,4,0
- KERNEL2x4_L2 64,32,5,0
- KERNEL2x4_L2 64,32,6,0
- KERNEL2x4_L2 64,32,7,0
- KERNEL2x4_L2 64,32,8,0
- KERNEL2x4_L2 64,32,9,0
- KERNEL2x4_L2 64,32,10,0
- KERNEL2x4_L2 64,32,11,0
- KERNEL2x4_L2 64,32,12,0
- KERNEL2x4_L2 64,32,13,0
- KERNEL2x4_L2 64,32,14,0
- KERNEL2x4_L2 64,32,15,1
- bdnz CGEMM_L2x4_LOOP
- MY_ALIGN
- CGEMM_L2x4_LOOP_END:
- /*----------------------------------------*/
- END2x4_2
- blr
- MY_ALIGN
-
-
- CGEMM_2x4_L16_SUB:
- /*----------------------------------------*/
- LOAD2x4_2
- KERNEL2x4_L2 64,32,0,0
- KERNEL2x4_L2 64,32,1,0
- KERNEL2x4_L2 64,32,2,0
- KERNEL2x4_L2 64,32,3,0
- KERNEL2x4_L2 64,32,4,0
- KERNEL2x4_L2 64,32,5,0
- KERNEL2x4_L2 64,32,6,0
- KERNEL2x4_E2 64,32,7,1
- blr
- MY_ALIGN
-
-
- CGEMM_2x4_L8_SUB:
- /*----------------------------------------*/
- LOAD2x4_2
- KERNEL2x4_L2 64,32,0,0
- KERNEL2x4_L2 64,32,1,0
- KERNEL2x4_L2 64,32,2,0
- KERNEL2x4_E2 64,32,3,1
- blr
-
-
- CGEMM_2x2_LMAIN_SUB:
- /*----------------------------------------*/
- mtctr T8
- LOAD2x2_2
- MY_ALIGN
- CGEMM_L2x2_LOOP:
- /*----------------------------------------*/
- KERNEL2x2_L2 32,32,0,0
- CGEMM_L2x2_K32:
- /*----------------------------------------*/
- KERNEL2x2_L2 32,32,1,0
- KERNEL2x2_L2 32,32,2,0
- KERNEL2x2_L2 32,32,3,0
- KERNEL2x2_L2 32,32,4,0
- KERNEL2x2_L2 32,32,5,0
- KERNEL2x2_L2 32,32,6,0
- KERNEL2x2_L2 32,32,7,0
- KERNEL2x2_L2 32,32,8,0
- KERNEL2x2_L2 32,32,9,0
- KERNEL2x2_L2 32,32,10,0
- KERNEL2x2_L2 32,32,11,0
- KERNEL2x2_L2 32,32,12,0
- KERNEL2x2_L2 32,32,13,0
- KERNEL2x2_L2 32,32,14,0
- KERNEL2x2_L2 32,32,15,1
- bdnz CGEMM_L2x2_LOOP
- MY_ALIGN
-
-
- CGEMM_L2x2_LOOP_END:
- /*----------------------------------------*/
- END2x2_2
- blr
- MY_ALIGN
- CGEMM_2x2_L16_SUB:
- /*----------------------------------------*/
- LOAD2x2_2
- KERNEL2x2_L2 32,32,0,0
- KERNEL2x2_L2 32,32,1,0
- KERNEL2x2_L2 32,32,2,0
- KERNEL2x2_L2 32,32,3,0
- KERNEL2x2_L2 32,32,4,0
- KERNEL2x2_L2 32,32,5,0
- KERNEL2x2_L2 32,32,6,0
- KERNEL2x2_E2 32,32,7,1
- blr
- MY_ALIGN
- CGEMM_2x2_L8_SUB:
- /*----------------------------------------*/
- LOAD2x2_2
- KERNEL2x2_L2 32,32,0,0
- KERNEL2x2_L2 32,32,1,0
- KERNEL2x2_L2 32,32,2,0
- KERNEL2x2_E2 32,32,3,1
- blr
-
-
- CGEMM_2x1_LMAIN_SUB:
- /*----------------------------------------*/
- mtctr T8
- LOAD2x1_2
- MY_ALIGN
- CGEMM_L2x1_LOOP:
- /*----------------------------------------*/
- KERNEL2x1_L2 16,32,0,0
- CGEMM_L2x1_K32:
- /*----------------------------------------*/
- KERNEL2x1_L2 16,32,1,0
- KERNEL2x1_L2 16,32,2,0
- KERNEL2x1_L2 16,32,3,0
- KERNEL2x1_L2 16,32,4,0
- KERNEL2x1_L2 16,32,5,0
- KERNEL2x1_L2 16,32,6,0
- KERNEL2x1_L2 16,32,7,0
- KERNEL2x1_L2 16,32,8,0
- KERNEL2x1_L2 16,32,9,0
- KERNEL2x1_L2 16,32,10,0
- KERNEL2x1_L2 16,32,11,0
- KERNEL2x1_L2 16,32,12,0
- KERNEL2x1_L2 16,32,13,0
- KERNEL2x1_L2 16,32,14,0
- KERNEL2x1_L2 16,32,15,1
- bdnz CGEMM_L2x1_LOOP
- MY_ALIGN
- CGEMM_L2x1_LOOP_END:
- /*----------------------------------------*/
- END2x1_2
- blr
-
- MY_ALIGN
- CGEMM_2x1_L16_SUB:
- /*----------------------------------------*/
- LOAD2x1_2
- KERNEL2x1_L2 16,32,0,0
- KERNEL2x1_L2 16,32,1,0
- KERNEL2x1_L2 16,32,2,0
- KERNEL2x1_L2 16,32,3,0
- KERNEL2x1_L2 16,32,4,0
- KERNEL2x1_L2 16,32,5,0
- KERNEL2x1_L2 16,32,6,0
- KERNEL2x1_E2 16,32,7,1
- blr
- MY_ALIGN
-
-
- CGEMM_2x1_L8_SUB:
- /*----------------------------------------*/
- LOAD2x1_2
- KERNEL2x1_L2 16,32,0,0
- KERNEL2x1_L2 16,32,1,0
- KERNEL2x1_L2 16,32,2,0
- KERNEL2x1_E2 16,32,3,1
- blr
-
-
-
- /* MAIN LOOP BEGINS */
- MY_ALIGN
-
-
- CGEMM_L2:
- /*----------------------------------------*/
-
- andi. J, N, 2
- ble CGEMM_L2_END
-
-
- CGEMM_L2_BEGIN:
- /*----------------------------------------*/
- mr CO, C
- slwi T1, LDC , 1
- add T2,C,LDC
- mr AO, A
- add C, C, T1
- #if defined(TRMMKERNEL) && defined(LEFT)
- mr TEMP_REG, OFFSET /*off = offset;*/
- #endif
- srawi. I, M, 3
- ble CGEMM_L2x8_END
- dcbt CO,r0 /*just prefetch*/
- dcbt T2,r0
-
-
- CGEMM_L2x8_BEGIN:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- REFRESH_POINTERS AO,BO,TEMP_REG,B,8,2
- #else
- mr BO, B
- dcbt B, r0
- #endif
- dcbt AO, r0
- #if defined(TRMMKERNEL)
- REFRESH_TEMP_BK T6,K,TEMP_REG,8,2
- mr T1, T6
- /* TEMPS FOR PREFETCH */
- li T2, 1024
- li T3, 1024+512
- addi T1,T1, -2
- /* TEMPS FOR PREFETCH */
- li T4, 2048
- li T5, 2048+512
- srawi. T8, T1, 7 /**(T1-2) % 128x */
- #else
- mr T1, K
- /* TEMPS FOR PREFETCH */
- li T2, 1024
- li T3, 1024+512
- addi T1,T1, -2
- /* TEMPS FOR PREFETCH */
- li T4, 2048
- li T5, 2048+512
- srawi. T8, T1, 7 /**(K-2) % 128x */
- #endif
- ZERO2x8
- ble CGEMM_L2x8_SUB0
- bl CGEMM_L2x8_LMAIN_SUB
- andi. L, T1, 127
- ble CGEMM_L2x8_SAVE
- b CGEMM_L2x8_SUB2
-
-
- CGEMM_L2x8_SUB0:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- andi. L, T6, 255
- cmpwi T6,129
- #else
- andi. L, K, 255
- cmpwi K,129
- #endif
- li T8,1
- bne CMP2x8_128K
- addi BO,BO,-16
- addi AO,AO,-64
- LOAD2x8O 64,16
- END2x8_WITHOUT_ADD
- LOAD2x8_2O 128, 32
- mtctr T8
- bl CGEMM_L2x8_K128
- b CGEMM_L2x8_SAVE
- CMP2x8_128K:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- cmpwi T6,128
- #else
- cmpwi K,128
- #endif
- bne CGEMM_L2x8_SUB2
- MY_ALIGN
- mtctr T8
- addi BO,BO,-32
- addi AO,AO,-128
- LOAD2x8_2O 128,32
- bl CGEMM_L2x8_K128
- b CGEMM_L2x8_SAVE
- MY_ALIGN
-
-
- CGEMM_L2x8_SUB2:
- /*----------------------------------------*/
- andi. T1,L, 64
- ble CGEMM_L2x8_SUB2_32
- bl CGEMM_2x8_L64_SUB
- MY_ALIGN
-
-
- CGEMM_L2x8_SUB2_32:
- /*----------------------------------------*/
- andi. T1,L, 32
- ble CGEMM_L2x8_SUB2_16
- bl CGEMM_2x8_L32_SUB
- MY_ALIGN
-
-
- CGEMM_L2x8_SUB2_16:
- /*----------------------------------------*/
- andi. T1,L, 16
- ble CGEMM_L2x8_SUB2_8
- bl CGEMM_2x8_L16_SUB
- MY_ALIGN
-
-
- CGEMM_L2x8_SUB2_8:
- /*----------------------------------------*/
- andi. T1,L, 8
- ble CGEMM_L2x8_SUB2_4
- LOAD2x8_2
- KERNEL2x8_L2 128,32, 0,0
- KERNEL2x8_L2 128,32, 1,0
- KERNEL2x8_L2 128,32, 2,0
- KERNEL2x8_E2 128,32, 3,1
- MY_ALIGN
-
-
- CGEMM_L2x8_SUB2_4:
- /*----------------------------------------*/
- andi. T1,L, 4
- ble CGEMM_L2x8_SUB2_2
- LOAD2x8_2
- KERNEL2x8_L2 128,32, 0,0
- KERNEL2x8_E2 128,32, 1,1
- MY_ALIGN
-
-
- CGEMM_L2x8_SUB2_2:
- /*----------------------------------------*/
- andi. T1,L, 2
- ble CGEMM_L2x8_SUB2_1
- LOAD2x8_2
- KERNEL2x8_E2 128,32, 0,1
- MY_ALIGN
-
-
- CGEMM_L2x8_SUB2_1:
- /*----------------------------------------*/
- andi. T1,L, 1
- ble CGEMM_L2x8_SAVE
- KERNEL2x8
-
- MY_ALIGN
- CGEMM_L2x8_SAVE:
- /*----------------------------------------*/
- addic. I, I, -1
- MY_ALIGN
- SAVE2x8
- #if defined(TRMMKERNEL)
- REFRESH_AFTER_SAVE T6,K,TEMP_REG,BO,AO,8,2
- #endif
- bgt CGEMM_L2x8_BEGIN
- andi. T2, M, 7
- ble CGEMM_L2x1_END
- andi. T1, M, 4
- ble CGEMM_L2x4_END
- b CGEMM_L2x4_BEGIN
- MY_ALIGN
-
-
- CGEMM_L2x8_END:
- /*----------------------------------------*/
-
-
- CGEMM_L2x4_BEGIN:
- /*----------------------------------------*/
- andi. T2, M, 7
- ble CGEMM_L2x1_END
- andi. T1, M, 4
- ble CGEMM_L2x4_END
- #if defined(TRMMKERNEL)
- REFRESH_POINTERS AO,BO,TEMP_REG,B,4,2
- #else
- mr BO, B
- #endif
- #if defined(TRMMKERNEL)
- REFRESH_TEMP_BK T6,K,TEMP_REG,4,2
- mr T1, T6
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(T1-2) % 32x */
- #else
- mr T1, K
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(K-2) % 32x */
- #endif
- ZERO2x4
- ble CGEMM_L2x4_SUB0
- bl CGEMM_2x4_LMAIN_SUB
- andi. L, T1, 31
- ble CGEMM_L2x4_SAVE
- b CGEMM_L2x4_SUB2
-
-
- CGEMM_L2x4_SUB0:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- andi. L, T6, 63
- cmpwi T6,33
- #else
- andi. L, K, 63
- cmpwi K,33
- #endif
- li T8,1
- bne CMP2x4_32K
- addi BO,BO,-16
- addi AO,AO,-32
- LOAD2x4O 32,16
- END2x4_WITHOUT_ADD
- LOAD2x4_2O 64, 32
- mtctr T8
- bl CGEMM_L2x4_K32
- b CGEMM_L2x4_SAVE
- CMP2x4_32K:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- cmpwi T6,32
- #else
- cmpwi K,32
- #endif
- bne CGEMM_L2x4_SUB2
- MY_ALIGN
- mtctr T8
- addi BO,BO,-32
- addi AO,AO,-64
- LOAD2x4_2O 64,32
- bl CGEMM_L2x4_K32
- b CGEMM_L2x4_SAVE
- MY_ALIGN
- MY_ALIGN
-
-
- CGEMM_L2x4_SUB2:
- /*----------------------------------------*/
- andi. T1,L, 16
- ble CGEMM_L2x4_SUB2_8
- bl CGEMM_2x4_L16_SUB
- MY_ALIGN
-
-
- CGEMM_L2x4_SUB2_8:
- /*----------------------------------------*/
- andi. T1,L, 8
- ble CGEMM_L2x4_SUB2_4
- bl CGEMM_2x4_L8_SUB
- MY_ALIGN
-
-
- CGEMM_L2x4_SUB2_4:
- /*----------------------------------------*/
- andi. T1,L, 4
- ble CGEMM_L2x4_SUB2_2
- LOAD2x4_2
- KERNEL2x4_L2 64,32, 0,0
- KERNEL2x4_E2 64,32, 1,1
- MY_ALIGN
-
-
- CGEMM_L2x4_SUB2_2:
- /*----------------------------------------*/
- andi. T1,L, 2
- ble CGEMM_L2x4_SUB2_1
- LOAD2x4_2
- KERNEL2x4_E2 64,32, 0,1
- MY_ALIGN
-
-
- CGEMM_L2x4_SUB2_1:
- /*----------------------------------------*/
- andi. T1,L, 1
- ble CGEMM_L2x4_SAVE
- KERNEL2x4
-
-
- CGEMM_L2x4_SAVE:
- /*----------------------------------------*/
- SAVE2x4
- #if defined(TRMMKERNEL)
- REFRESH_AFTER_SAVE T6,K,TEMP_REG,BO,AO,4,2
- #endif
-
-
- CGEMM_L2x4_END:
- /*----------------------------------------*/
-
-
- CGEMM_L2x2_BEGIN:
- /*----------------------------------------*/
- andi. T1, M, 2
- ble CGEMM_L2x2_END
- #if defined(TRMMKERNEL)
- REFRESH_POINTERS AO,BO,TEMP_REG,B,2,2
- #else
- mr BO, B
- #endif
- #if defined(TRMMKERNEL)
- REFRESH_TEMP_BK T6,K,TEMP_REG,2,2
- mr T1, T6
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(T1-2) % 32x */
- #else
- mr T1, K
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(K-2) % 32x */
- #endif
- ZERO2x2
- ble CGEMM_L2x2_SUB0
- bl CGEMM_2x2_LMAIN_SUB
- andi. L, T1, 31
- ble CGEMM_L2x2_SAVE
- b CGEMM_L2x2_SUB2
-
-
- CGEMM_L2x2_SUB0:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- andi. L, T6, 63
- cmpwi T6,33
- #else
- andi. L, K, 63
- cmpwi K,33
- #endif
- li T8,1
- bne CMP2x2_32K
- addi BO,BO,-16
- addi AO,AO,-16
- LOAD2x2O 16,16
- END2x2_WITHOUT_ADD
- LOAD2x2_2O 32, 32
- mtctr T8
- bl CGEMM_L2x2_K32
- b CGEMM_L2x2_SAVE
- CMP2x2_32K:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- cmpwi T6,32
- #else
- cmpwi K,32
- #endif
- bne CGEMM_L2x2_SUB2
- MY_ALIGN
- mtctr T8
- addi BO,BO,-32
- addi AO,AO,-32
- LOAD2x2_2O 32,32
- bl CGEMM_L2x2_K32
- b CGEMM_L2x2_SAVE
- MY_ALIGN
- MY_ALIGN
-
-
- CGEMM_L2x2_SUB2:
- /*----------------------------------------*/
- andi. T1,L, 16
- ble CGEMM_L2x2_SUB2_8
- bl CGEMM_2x2_L16_SUB
- MY_ALIGN
-
-
- CGEMM_L2x2_SUB2_8:
- /*----------------------------------------*/
- andi. T1,L, 8
- ble CGEMM_L2x2_SUB2_4
- bl CGEMM_2x2_L8_SUB
- MY_ALIGN
-
-
- CGEMM_L2x2_SUB2_4:
- /*----------------------------------------*/
- andi. T1,L, 4
- ble CGEMM_L2x2_SUB2_2
- LOAD2x2_2
- KERNEL2x2_L2 32,32, 0,0
- KERNEL2x2_E2 32,32, 1,1
- MY_ALIGN
-
-
- CGEMM_L2x2_SUB2_2:
- /*----------------------------------------*/
- andi. T1,L, 2
- ble CGEMM_L2x2_SUB2_1
- LOAD2x2_2
- KERNEL2x2_E2 32,32, 0,1
- MY_ALIGN
-
-
- CGEMM_L2x2_SUB2_1:
- /*----------------------------------------*/
- andi. T1,L, 1
- ble CGEMM_L2x2_SAVE
- KERNEL2x2
-
- MY_ALIGN
- CGEMM_L2x2_SAVE:
- /*----------------------------------------*/
- SAVE2x2
- #if defined(TRMMKERNEL)
- REFRESH_AFTER_SAVE T6,K,TEMP_REG,BO,AO,2,2
- #endif
-
-
- CGEMM_L2x2_END:
- /*----------------------------------------*/
-
-
- CGEMM_L2x1_BEGIN:
- /*----------------------------------------*/
- andi. T1, M, 1
- ble CGEMM_L2x1_END
- #if defined(TRMMKERNEL)
- REFRESH_POINTERS AO,BO,TEMP_REG,B,1,2
- #else
- mr BO, B
- #endif
- #if defined(TRMMKERNEL)
- REFRESH_TEMP_BK T6,K,TEMP_REG,1,2
- mr T1, T6
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(T1-2) % 32x */
- #else
- mr T1, K
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(K-2) % 32x */
- #endif
- ZERO2x1
- ble CGEMM_L2x1_SUB0
- bl CGEMM_2x1_LMAIN_SUB
- andi. L, T1, 31
- ble CGEMM_L2x1_SAVE
- b CGEMM_L2x1_SUB2
-
-
- CGEMM_L2x1_SUB0:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- andi. L, T6, 63
- cmpwi T6,33
- #else
- andi. L, K, 63
- cmpwi K,33
- #endif
- li T8,1
- bne CMP2x1_32K
- addi BO,BO,-16
- addi AO,AO,-8
- LOAD2x1O 8,16
- END2x1_WITHOUT_ADD
- LOAD2x1_2O 16, 32
- mtctr T8
- bl CGEMM_L2x1_K32
- b CGEMM_L2x1_SAVE
- CMP2x1_32K:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- cmpwi T6,32
- #else
- cmpwi K,32
- #endif
- bne CGEMM_L2x1_SUB2
- MY_ALIGN
- mtctr T8
- addi BO,BO,-32
- addi AO,AO,-16
- LOAD2x1_2O 16,32
- bl CGEMM_L2x1_K32
- b CGEMM_L2x1_SAVE
- MY_ALIGN
- MY_ALIGN
-
-
- CGEMM_L2x1_SUB2:
- /*----------------------------------------*/
- andi. T1,L, 16
- ble CGEMM_L2x1_SUB2_8
- bl CGEMM_2x1_L16_SUB
- MY_ALIGN
-
-
- CGEMM_L2x1_SUB2_8:
- /*----------------------------------------*/
- andi. T1,L, 8
- ble CGEMM_L2x1_SUB2_4
- bl CGEMM_2x1_L8_SUB
- MY_ALIGN
-
-
- CGEMM_L2x1_SUB2_4:
- /*----------------------------------------*/
- andi. T1,L, 4
- ble CGEMM_L2x1_SUB2_2
- LOAD2x1_2
- KERNEL2x1_L2 16,32, 0,0
- KERNEL2x1_E2 16,32, 1,1
- MY_ALIGN
-
-
- CGEMM_L2x1_SUB2_2:
- /*----------------------------------------*/
- andi. T1,L, 2
- ble CGEMM_L2x1_SUB2_1
- LOAD2x1_2
- KERNEL2x1_E2 16,32, 0,1
- MY_ALIGN
-
-
- CGEMM_L2x1_SUB2_1:
- /*----------------------------------------*/
- andi. T1,L, 1
- ble CGEMM_L2x1_SAVE
- KERNEL2x1
-
- MY_ALIGN
- CGEMM_L2x1_SAVE:
- /*----------------------------------------*/
-
- SAVE2x1
- #if defined(TRMMKERNEL)
- REFRESH_AFTER_SAVE T6,K,TEMP_REG,BO,AO,1,2
- #endif
-
-
- CGEMM_L2x1_END:
- /*----------------------------------------*/
- slwi T1, K, 4
-
- add B, B, T1
- #if defined(TRMMKERNEL) && !defined(LEFT)
- addi TEMP_REG, TEMP_REG, 2
- #endif
-
- CGEMM_L2_END:
-
-
- b CGEMM_L1
- /* MINI SUBROUTINES */
- /* 1x8 MAIN 128x+2 LOOP */
-
-
- CGEMM_L1x8_LMAIN_SUB:
- /*----------------------------------------*/
- mtctr T8
- LOAD1x8_2
- MY_ALIGN
- CGEMM_L1x8_LOOP:
- /*----------------------------------------*/
- dcbt AO, PRE
- dcbt BO, PRE
- KERNEL1x8_L2 128,16,0,0
- CGEMM_L1x8_K128:
- /*----------------------------------------*/
- KERNEL1x8_L2 128,16,1,0
- dcbt AO, T2
- KERNEL1x8_L2 128,16,2,0
- KERNEL1x8_L2 128,16,3,0
- dcbt AO, T3
- dcbt BO, T2
- KERNEL1x8_L2 128,16,4,0
- KERNEL1x8_L2 128,16,5,0
- dcbt AO, T4
- KERNEL1x8_L2 128,16,6,0
- KERNEL1x8_L2 128,16,7,0
- dcbt AO, T5
- dcbt BO, T3
- KERNEL1x8_L2 128,16,8,0
- KERNEL1x8_L2 128,16,9,0
- KERNEL1x8_L2 128,16,10,0
- KERNEL1x8_L2 128,16,11,0
- dcbt BO, T4
- KERNEL1x8_L2 128,16,12,0
- KERNEL1x8_L2 128,16,13,0
- KERNEL1x8_L2 128,16,14,0
- KERNEL1x8_L2 128,16,15,0
- KERNEL1x8_L2 128,16,16,0
- KERNEL1x8_L2 128,16,17,0
- KERNEL1x8_L2 128,16,18,0
- KERNEL1x8_L2 128,16,19,0
- KERNEL1x8_L2 128,16,20,0
- KERNEL1x8_L2 128,16,21,0
- KERNEL1x8_L2 128,16,22,0
- KERNEL1x8_L2 128,16,23,0
- KERNEL1x8_L2 128,16,24,0
- KERNEL1x8_L2 128,16,25,0
- KERNEL1x8_L2 128,16,26,0
- KERNEL1x8_L2 128,16,27,0
- KERNEL1x8_L2 128,16,28,0
- KERNEL1x8_L2 128,16,29,0
- KERNEL1x8_L2 128,16,30,0
- KERNEL1x8_L2 128,16,31,0
- KERNEL1x8_L2 128,16,32,0
- KERNEL1x8_L2 128,16,33,0
- KERNEL1x8_L2 128,16,34,0
- KERNEL1x8_L2 128,16,35,0
- KERNEL1x8_L2 128,16,36,0
- KERNEL1x8_L2 128,16,37,0
- KERNEL1x8_L2 128,16,38,0
- KERNEL1x8_L2 128,16,39,0
- KERNEL1x8_L2 128,16,40,0
- KERNEL1x8_L2 128,16,41,0
- KERNEL1x8_L2 128,16,42,0
- KERNEL1x8_L2 128,16,43,0
- KERNEL1x8_L2 128,16,44,0
- KERNEL1x8_L2 128,16,45,0
- KERNEL1x8_L2 128,16,46,0
- KERNEL1x8_L2 128,16,47,0
- KERNEL1x8_L2 128,16,48,0
- KERNEL1x8_L2 128,16,49,0
- KERNEL1x8_L2 128,16,50,0
- KERNEL1x8_L2 128,16,51,0
- KERNEL1x8_L2 128,16,52,0
- KERNEL1x8_L2 128,16,53,0
- KERNEL1x8_L2 128,16,54,0
- KERNEL1x8_L2 128,16,55,0
- KERNEL1x8_L2 128,16,56,0
- KERNEL1x8_L2 128,16,57,0
- KERNEL1x8_L2 128,16,58,0
- KERNEL1x8_L2 128,16,59,0
- KERNEL1x8_L2 128,16,60,0
- KERNEL1x8_L2 128,16,61,0
- KERNEL1x8_L2 128,16,62,0
- KERNEL1x8_L2 128,16,63,1
- bdnz CGEMM_L1x8_LOOP
- MY_ALIGN
- CGEMM_L1x8_LOOP_END:
- /*----------------------------------------*/
- END1x8_2
- blr
- MY_ALIGN
-
-
- CGEMM_1x8_L64_SUB:
- /*----------------------------------------*/
- LOAD1x8_2
- dcbt AO, PRE
- dcbt BO, PRE
- KERNEL1x8_L2 128,16,0,0
- KERNEL1x8_L2 128,16,1,0
- dcbt AO, T2
- KERNEL1x8_L2 128,16,2,0
- KERNEL1x8_L2 128,16,3,0
- dcbt AO, T3
- dcbt BO, T2
- KERNEL1x8_L2 128,16,4,0
- KERNEL1x8_L2 128,16,5,0
- dcbt AO, T4
- KERNEL1x8_L2 128,16,6,0
- KERNEL1x8_L2 128,16,7,0
- dcbt AO, T5
- dcbt BO, T3
- KERNEL1x8_L2 128,16,8,0
- KERNEL1x8_L2 128,16,9,0
- KERNEL1x8_L2 128,16,10,0
- KERNEL1x8_L2 128,16,11,0
- dcbt BO, T4
- KERNEL1x8_L2 128,16,12,0
- KERNEL1x8_L2 128,16,13,0
- KERNEL1x8_L2 128,16,14,0
- KERNEL1x8_L2 128,16,15,0
- KERNEL1x8_L2 128,16,16,0
- KERNEL1x8_L2 128,16,17,0
- KERNEL1x8_L2 128,16,18,0
- KERNEL1x8_L2 128,16,19,0
- KERNEL1x8_L2 128,16,20,0
- KERNEL1x8_L2 128,16,21,0
- KERNEL1x8_L2 128,16,22,0
- KERNEL1x8_L2 128,16,23,0
- KERNEL1x8_L2 128,16,24,0
- KERNEL1x8_L2 128,16,25,0
- KERNEL1x8_L2 128,16,26,0
- KERNEL1x8_L2 128,16,27,0
- KERNEL1x8_L2 128,16,28,0
- KERNEL1x8_L2 128,16,29,0
- KERNEL1x8_L2 128,16,30,0
- KERNEL1x8_E2 128,16,31,1
- blr
- MY_ALIGN
-
-
- CGEMM_1x8_L32_SUB:
- /*----------------------------------------*/
- LOAD1x8_2
- dcbt AO, PRE
- dcbt BO, PRE
- KERNEL1x8_L2 128,16,0,0
- KERNEL1x8_L2 128,16,1,0
- dcbt AO, T2
- KERNEL1x8_L2 128,16,2,0
- KERNEL1x8_L2 128,16,3,0
- dcbt AO, T3
- dcbt BO, T2
- KERNEL1x8_L2 128,16,4,0
- KERNEL1x8_L2 128,16,5,0
- dcbt AO, T4
- KERNEL1x8_L2 128,16,6,0
- KERNEL1x8_L2 128,16,7,0
- dcbt AO, T5
- dcbt BO, T3
- KERNEL1x8_L2 128,16,8,0
- KERNEL1x8_L2 128,16,9,0
- KERNEL1x8_L2 128,16,10,0
- KERNEL1x8_L2 128,16,11,0
- dcbt BO, T4
- KERNEL1x8_L2 128,16,12,0
- KERNEL1x8_L2 128,16,13,0
- KERNEL1x8_L2 128,16,14,0
- KERNEL1x8_E2 128,16,15,1
- blr
- MY_ALIGN
-
-
- CGEMM_1x8_L16_SUB:
- /*----------------------------------------*/
- LOAD1x8_2
- dcbt AO, PRE
- dcbt BO, PRE
- KERNEL1x8_L2 128,16,0,0
- KERNEL1x8_L2 128,16,1,0
- dcbt AO, T2
- KERNEL1x8_L2 128,16,2,0
- KERNEL1x8_L2 128,16,3,0
- dcbt AO, T3
- dcbt BO, T2
- KERNEL1x8_L2 128,16,4,0
- KERNEL1x8_L2 128,16,5,0
- dcbt AO, T4
- KERNEL1x8_L2 128,16,6,0
- KERNEL1x8_E2 128,16,7,1
- blr
- MY_ALIGN
-
-
- CGEMM_1x4_LMAIN_SUB:
- /*----------------------------------------*/
- mtctr T8
- LOAD1x4_2
- MY_ALIGN
- CGEMM_L1x4_LOOP:
- /*----------------------------------------*/
- KERNEL1x4_L2 64,16,0,0
- CGEMM_L1x4_K32:
- /*----------------------------------------*/
- KERNEL1x4_L2 64,16,1,0
- KERNEL1x4_L2 64,16,2,0
- KERNEL1x4_L2 64,16,3,0
- KERNEL1x4_L2 64,16,4,0
- KERNEL1x4_L2 64,16,5,0
- KERNEL1x4_L2 64,16,6,0
- KERNEL1x4_L2 64,16,7,0
- KERNEL1x4_L2 64,16,8,0
- KERNEL1x4_L2 64,16,9,0
- KERNEL1x4_L2 64,16,10,0
- KERNEL1x4_L2 64,16,11,0
- KERNEL1x4_L2 64,16,12,0
- KERNEL1x4_L2 64,16,13,0
- KERNEL1x4_L2 64,16,14,0
- KERNEL1x4_L2 64,16,15,1
- bdnz CGEMM_L1x4_LOOP
- MY_ALIGN
- CGEMM_L1x4_LOOP_END:
- /*----------------------------------------*/
- END1x4_2
- blr
- MY_ALIGN
-
-
- CGEMM_1x4_L16_SUB:
- /*----------------------------------------*/
- LOAD1x4_2
- KERNEL1x4_L2 64,16,0,0
- KERNEL1x4_L2 64,16,1,0
- KERNEL1x4_L2 64,16,2,0
- KERNEL1x4_L2 64,16,3,0
- KERNEL1x4_L2 64,16,4,0
- KERNEL1x4_L2 64,16,5,0
- KERNEL1x4_L2 64,16,6,0
- KERNEL1x4_E2 64,16,7,1
- blr
- MY_ALIGN
-
-
- CGEMM_1x4_L8_SUB:
- /*----------------------------------------*/
- LOAD1x4_2
- KERNEL1x4_L2 64,16,0,0
- KERNEL1x4_L2 64,16,1,0
- KERNEL1x4_L2 64,16,2,0
- KERNEL1x4_E2 64,16,3,1
- blr
-
-
- CGEMM_1x2_LMAIN_SUB:
- /*----------------------------------------*/
- mtctr T8
- LOAD1x2_2
- MY_ALIGN
- CGEMM_L1x2_LOOP:
- /*----------------------------------------*/
- KERNEL1x2_L2 32,16,0,0
- CGEMM_L1x2_K32:
- /*----------------------------------------*/
- KERNEL1x2_L2 32,16,1,0
- KERNEL1x2_L2 32,16,2,0
- KERNEL1x2_L2 32,16,3,0
- KERNEL1x2_L2 32,16,4,0
- KERNEL1x2_L2 32,16,5,0
- KERNEL1x2_L2 32,16,6,0
- KERNEL1x2_L2 32,16,7,0
- KERNEL1x2_L2 32,16,8,0
- KERNEL1x2_L2 32,16,9,0
- KERNEL1x2_L2 32,16,10,0
- KERNEL1x2_L2 32,16,11,0
- KERNEL1x2_L2 32,16,12,0
- KERNEL1x2_L2 32,16,13,0
- KERNEL1x2_L2 32,16,14,0
- KERNEL1x2_L2 32,16,15,1
- bdnz CGEMM_L1x2_LOOP
- MY_ALIGN
-
-
- CGEMM_L1x2_LOOP_END:
- /*----------------------------------------*/
- END1x2_2
- blr
- MY_ALIGN
- CGEMM_1x2_L16_SUB:
- /*----------------------------------------*/
- LOAD1x2_2
- KERNEL1x2_L2 32,16,0,0
- KERNEL1x2_L2 32,16,1,0
- KERNEL1x2_L2 32,16,2,0
- KERNEL1x2_L2 32,16,3,0
- KERNEL1x2_L2 32,16,4,0
- KERNEL1x2_L2 32,16,5,0
- KERNEL1x2_L2 32,16,6,0
- KERNEL1x2_E2 32,16,7,1
- blr
- MY_ALIGN
- CGEMM_1x2_L8_SUB:
- /*----------------------------------------*/
- LOAD1x2_2
- KERNEL1x2_L2 32,16,0,0
- KERNEL1x2_L2 32,16,1,0
- KERNEL1x2_L2 32,16,2,0
- KERNEL1x2_E2 32,16,3,1
- blr
-
-
- CGEMM_1x1_LMAIN_SUB:
- /*----------------------------------------*/
- mtctr T8
- LOAD1x1_2
- MY_ALIGN
- CGEMM_L1x1_LOOP:
- /*----------------------------------------*/
- KERNEL1x1_L2 16,16,0,0
- CGEMM_L1x1_K32:
- /*----------------------------------------*/
- KERNEL1x1_L2 16,16,1,0
- KERNEL1x1_L2 16,16,2,0
- KERNEL1x1_L2 16,16,3,0
- KERNEL1x1_L2 16,16,4,0
- KERNEL1x1_L2 16,16,5,0
- KERNEL1x1_L2 16,16,6,0
- KERNEL1x1_L2 16,16,7,0
- KERNEL1x1_L2 16,16,8,0
- KERNEL1x1_L2 16,16,9,0
- KERNEL1x1_L2 16,16,10,0
- KERNEL1x1_L2 16,16,11,0
- KERNEL1x1_L2 16,16,12,0
- KERNEL1x1_L2 16,16,13,0
- KERNEL1x1_L2 16,16,14,0
- KERNEL1x1_L2 16,16,15,1
- bdnz CGEMM_L1x1_LOOP
- MY_ALIGN
- CGEMM_L1x1_LOOP_END:
- /*----------------------------------------*/
- END1x1_2
- blr
-
- MY_ALIGN
- CGEMM_1x1_L16_SUB:
- /*----------------------------------------*/
- LOAD1x1_2
- KERNEL1x1_L2 16,16,0,0
- KERNEL1x1_L2 16,16,1,0
- KERNEL1x1_L2 16,16,2,0
- KERNEL1x1_L2 16,16,3,0
- KERNEL1x1_L2 16,16,4,0
- KERNEL1x1_L2 16,16,5,0
- KERNEL1x1_L2 16,16,6,0
- KERNEL1x1_E2 16,16,7,1
- blr
- MY_ALIGN
-
-
- CGEMM_1x1_L8_SUB:
- /*----------------------------------------*/
- LOAD1x1_2
- KERNEL1x1_L2 16,16,0,0
- KERNEL1x1_L2 16,16,1,0
- KERNEL1x1_L2 16,16,2,0
- KERNEL1x1_E2 16,16,3,1
- blr
-
-
-
- /* MAIN LOOP BEGINS */
- MY_ALIGN
-
-
- CGEMM_L1:
- /*----------------------------------------*/
-
- andi. J, N, 1
- ble CGEMM_L1_END
-
- CGEMM_L1_BEGIN:
- /*----------------------------------------*/
- mr CO, C
- add T2,C,LDC
- mr AO, A
- add C, C, T1
- #if defined(TRMMKERNEL) && defined(LEFT)
- mr TEMP_REG, OFFSET /*off = offset;*/
- #endif
- srawi. I, M, 3
- ble CGEMM_L1x8_END
- dcbt CO,r0 /*just prefetch*/
- dcbt T2,r0
-
-
- CGEMM_L1x8_BEGIN:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- REFRESH_POINTERS AO,BO,TEMP_REG,B,8,1
- #else
- mr BO, B
- dcbt B, r0
- #endif
- dcbt AO, r0
- #if defined(TRMMKERNEL)
- REFRESH_TEMP_BK T6,K,TEMP_REG,8,1
- mr T1, T6
- /* TEMPS FOR PREFETCH */
- li T2, 1024
- li T3, 1024+512
- addi T1,T1, -2
- /* TEMPS FOR PREFETCH */
- li T4, 2048
- li T5, 2048+512
- srawi. T8, T1, 7 /**(T1-2) % 128x */
- #else
- mr T1, K
- /* TEMPS FOR PREFETCH */
- li T2, 1024
- li T3, 1024+512
- addi T1,T1, -2
- /* TEMPS FOR PREFETCH */
- li T4, 2048
- li T5, 2048+512
- srawi. T8, T1, 7 /**(K-2) % 128x */
- #endif
- ZERO1x8
- ble CGEMM_L1x8_SUB0
- bl CGEMM_L1x8_LMAIN_SUB
- andi. L, T1, 127
- ble CGEMM_L1x8_SAVE
- b CGEMM_L1x8_SUB2
-
-
- CGEMM_L1x8_SUB0:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- andi. L, T6, 255
- cmpwi T6,129
- #else
- andi. L, K, 255
- cmpwi K,129
- #endif
- li T8,1
- bne CMP1x8_128K
- addi BO,BO,-8
- addi AO,AO,-64
- LOAD1x8O 64,8
- END1x8_WITHOUT_ADD
- LOAD1x8_2O 128, 16
- mtctr T8
- bl CGEMM_L1x8_K128
- b CGEMM_L1x8_SAVE
- CMP1x8_128K:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- cmpwi T6,128
- #else
- cmpwi K,128
- #endif
- bne CGEMM_L1x8_SUB2
- MY_ALIGN
- mtctr T8
- addi BO,BO,-16
- addi AO,AO,-128
- LOAD1x8_2O 128,16
- bl CGEMM_L1x8_K128
- b CGEMM_L1x8_SAVE
- MY_ALIGN
-
-
- CGEMM_L1x8_SUB2:
- /*----------------------------------------*/
- andi. T1,L, 64
- ble CGEMM_L1x8_SUB2_32
- bl CGEMM_1x8_L64_SUB
- MY_ALIGN
-
-
- CGEMM_L1x8_SUB2_32:
- /*----------------------------------------*/
- andi. T1,L, 32
- ble CGEMM_L1x8_SUB2_16
- bl CGEMM_1x8_L32_SUB
- MY_ALIGN
-
-
- CGEMM_L1x8_SUB2_16:
- /*----------------------------------------*/
- andi. T1,L, 16
- ble CGEMM_L1x8_SUB2_8
- bl CGEMM_1x8_L16_SUB
- MY_ALIGN
-
-
- CGEMM_L1x8_SUB2_8:
- /*----------------------------------------*/
- andi. T1,L, 8
- ble CGEMM_L1x8_SUB2_4
- LOAD1x8_2
- KERNEL1x8_L2 128,16, 0,0
- KERNEL1x8_L2 128,16, 1,0
- KERNEL1x8_L2 128,16, 2,0
- KERNEL1x8_E2 128,16, 3,1
- MY_ALIGN
-
-
- CGEMM_L1x8_SUB2_4:
- /*----------------------------------------*/
- andi. T1,L, 4
- ble CGEMM_L1x8_SUB2_2
- LOAD1x8_2
- KERNEL1x8_L2 128,16, 0,0
- KERNEL1x8_E2 128,16, 1,1
- MY_ALIGN
-
-
- CGEMM_L1x8_SUB2_2:
- /*----------------------------------------*/
- andi. T1,L, 2
- ble CGEMM_L1x8_SUB2_1
- LOAD1x8_2
- KERNEL1x8_E2 128,16, 0,1
- MY_ALIGN
-
-
- CGEMM_L1x8_SUB2_1:
- /*----------------------------------------*/
- andi. T1,L, 1
- ble CGEMM_L1x8_SAVE
- KERNEL1x8
-
- MY_ALIGN
- CGEMM_L1x8_SAVE:
- /*----------------------------------------*/
- addic. I, I, -1
- MY_ALIGN
- SAVE1x8
- #if defined(TRMMKERNEL)
- REFRESH_AFTER_SAVE T6,K,TEMP_REG,BO,AO,8,1
- #endif
- bgt CGEMM_L1x8_BEGIN
- andi. T2, M, 7
- ble CGEMM_L1x1_END
- andi. T1, M, 4
- ble CGEMM_L1x4_END
- b CGEMM_L1x4_BEGIN
- MY_ALIGN
-
-
- CGEMM_L1x8_END:
- /*----------------------------------------*/
-
-
- CGEMM_L1x4_BEGIN:
- /*----------------------------------------*/
- andi. T2, M, 7
- ble CGEMM_L1x1_END
- andi. T1, M, 4
- ble CGEMM_L1x4_END
- #if defined(TRMMKERNEL)
- REFRESH_POINTERS AO,BO,TEMP_REG,B,4,1
- #else
- mr BO, B
- #endif
- #if defined(TRMMKERNEL)
- REFRESH_TEMP_BK T6,K,TEMP_REG,4,1
- mr T1, T6
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(T1-2) % 31x */
- #else
- mr T1, K
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(K-2) % 31x */
- #endif
- ZERO1x4
- ble CGEMM_L1x4_SUB0
- bl CGEMM_1x4_LMAIN_SUB
- andi. L, T1, 31
- ble CGEMM_L1x4_SAVE
- b CGEMM_L1x4_SUB2
-
-
- CGEMM_L1x4_SUB0:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- andi. L, T6, 63
- cmpwi T6,33
- #else
- andi. L, K, 63
- cmpwi K,33
- #endif
- li T8,1
- bne CMP1x4_32K
- addi BO,BO,-8
- addi AO,AO,-32
- LOAD1x4O 32,8
- END1x4_WITHOUT_ADD
- LOAD1x4_2O 64, 16
- mtctr T8
- bl CGEMM_L1x4_K32
- b CGEMM_L1x4_SAVE
- CMP1x4_32K:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- cmpwi T6,32
- #else
- cmpwi K,32
- #endif
- bne CGEMM_L1x4_SUB2
- MY_ALIGN
- mtctr T8
- addi BO,BO,-16
- addi AO,AO,-64
- LOAD1x4_2O 64,16
- bl CGEMM_L1x4_K32
- b CGEMM_L1x4_SAVE
- MY_ALIGN
- MY_ALIGN
-
-
- CGEMM_L1x4_SUB2:
- /*----------------------------------------*/
- andi. T1,L, 16
- ble CGEMM_L1x4_SUB2_8
- bl CGEMM_1x4_L16_SUB
- MY_ALIGN
-
-
- CGEMM_L1x4_SUB2_8:
- /*----------------------------------------*/
- andi. T1,L, 8
- ble CGEMM_L1x4_SUB2_4
- bl CGEMM_1x4_L8_SUB
- MY_ALIGN
-
-
- CGEMM_L1x4_SUB2_4:
- /*----------------------------------------*/
- andi. T1,L, 4
- ble CGEMM_L1x4_SUB2_2
- LOAD1x4_2
- KERNEL1x4_L2 64,16, 0,0
- KERNEL1x4_E2 64,16, 1,1
- MY_ALIGN
-
-
- CGEMM_L1x4_SUB2_2:
- /*----------------------------------------*/
- andi. T1,L, 2
- ble CGEMM_L1x4_SUB2_1
- LOAD1x4_2
- KERNEL1x4_E2 64,16, 0,1
- MY_ALIGN
-
-
- CGEMM_L1x4_SUB2_1:
- /*----------------------------------------*/
- andi. T1,L, 1
- ble CGEMM_L1x4_SAVE
- KERNEL1x4
-
-
- CGEMM_L1x4_SAVE:
- /*----------------------------------------*/
- SAVE1x4
- #if defined(TRMMKERNEL)
- REFRESH_AFTER_SAVE T6,K,TEMP_REG,BO,AO,4,1
- #endif
-
-
- CGEMM_L1x4_END:
- /*----------------------------------------*/
-
-
- CGEMM_L1x2_BEGIN:
- /*----------------------------------------*/
- andi. T1, M, 2
- ble CGEMM_L1x2_END
- #if defined(TRMMKERNEL)
- REFRESH_POINTERS AO,BO,TEMP_REG,B,2,1
- #else
- mr BO, B
- #endif
- #if defined(TRMMKERNEL)
- REFRESH_TEMP_BK T6,K,TEMP_REG,2,1
- mr T1, T6
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(T1-2) % 31x */
- #else
- mr T1, K
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(K-2) % 31x */
- #endif
- ZERO1x2
- ble CGEMM_L1x2_SUB0
- bl CGEMM_1x2_LMAIN_SUB
- andi. L, T1, 31
- ble CGEMM_L1x2_SAVE
- b CGEMM_L1x2_SUB2
-
-
- CGEMM_L1x2_SUB0:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- andi. L, T6, 63
- cmpwi T6,33
- #else
- andi. L, K, 63
- cmpwi K,33
- #endif
- li T8,1
- bne CMP1x2_32K
- addi BO,BO,-8
- addi AO,AO,-16
- LOAD1x2O 16,8
- END1x2_WITHOUT_ADD
- LOAD1x2_2O 32, 16
- mtctr T8
- bl CGEMM_L1x2_K32
- b CGEMM_L1x2_SAVE
- CMP1x2_32K:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- cmpwi T6,32
- #else
- cmpwi K,32
- #endif
- bne CGEMM_L1x2_SUB2
- MY_ALIGN
- mtctr T8
- addi BO,BO,-16
- addi AO,AO,-32
- LOAD1x2_2O 32,16
- bl CGEMM_L1x2_K32
- b CGEMM_L1x2_SAVE
- MY_ALIGN
- MY_ALIGN
-
-
- CGEMM_L1x2_SUB2:
- /*----------------------------------------*/
- andi. T1,L, 16
- ble CGEMM_L1x2_SUB2_8
- bl CGEMM_1x2_L16_SUB
- MY_ALIGN
-
-
- CGEMM_L1x2_SUB2_8:
- /*----------------------------------------*/
- andi. T1,L, 8
- ble CGEMM_L1x2_SUB2_4
- bl CGEMM_1x2_L8_SUB
- MY_ALIGN
-
-
- CGEMM_L1x2_SUB2_4:
- /*----------------------------------------*/
- andi. T1,L, 4
- ble CGEMM_L1x2_SUB2_2
- LOAD1x2_2
- KERNEL1x2_L2 32,16, 0,0
- KERNEL1x2_E2 32,16, 1,1
- MY_ALIGN
-
-
- CGEMM_L1x2_SUB2_2:
- /*----------------------------------------*/
- andi. T1,L, 2
- ble CGEMM_L1x2_SUB2_1
- LOAD1x2_2
- KERNEL1x2_E2 32,16, 0,1
- MY_ALIGN
-
-
- CGEMM_L1x2_SUB2_1:
- /*----------------------------------------*/
- andi. T1,L, 1
- ble CGEMM_L1x2_SAVE
- KERNEL1x2
-
- MY_ALIGN
- CGEMM_L1x2_SAVE:
- /*----------------------------------------*/
- SAVE1x2
- #if defined(TRMMKERNEL)
- REFRESH_AFTER_SAVE T6,K,TEMP_REG,BO,AO,2,1
- #endif
-
-
- CGEMM_L1x2_END:
- /*----------------------------------------*/
-
-
- CGEMM_L1x1_BEGIN:
- /*----------------------------------------*/
- andi. T1, M, 1
- ble CGEMM_L1x1_END
- #if defined(TRMMKERNEL)
- REFRESH_POINTERS AO,BO,TEMP_REG,B,1,1
- #else
- mr BO, B
- #endif
- #if defined(TRMMKERNEL)
- REFRESH_TEMP_BK T6,K,TEMP_REG,1,1
- mr T1, T6
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(T1-2) % 31x */
- #else
- mr T1, K
- addi T1,T1, -2
- srawi. T8, T1, 5 /**(K-2) % 31x */
- #endif
- ZERO1x1
- ble CGEMM_L1x1_SUB0
- bl CGEMM_1x1_LMAIN_SUB
- andi. L, T1, 31
- ble CGEMM_L1x1_SAVE
- b CGEMM_L1x1_SUB2
-
-
- CGEMM_L1x1_SUB0:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- andi. L, T6, 63
- cmpwi T6,33
- #else
- andi. L, K, 63
- cmpwi K,33
- #endif
- li T8,1
- bne CMP1x1_32K
- addi BO,BO,-8
- addi AO,AO,-8
- LOAD1x1O 8,8
- END1x1_WITHOUT_ADD
- LOAD1x1_2O 16, 16
- mtctr T8
- bl CGEMM_L1x1_K32
- b CGEMM_L1x1_SAVE
- CMP1x1_32K:
- /*----------------------------------------*/
- #if defined(TRMMKERNEL)
- cmpwi T6,32
- #else
- cmpwi K,32
- #endif
- bne CGEMM_L1x1_SUB2
- MY_ALIGN
- mtctr T8
- addi BO,BO,-16
- addi AO,AO,-16
- LOAD1x1_2O 16,16
- bl CGEMM_L1x1_K32
- b CGEMM_L1x1_SAVE
- MY_ALIGN
- MY_ALIGN
-
-
- CGEMM_L1x1_SUB2:
- /*----------------------------------------*/
- andi. T1,L, 16
- ble CGEMM_L1x1_SUB2_8
- bl CGEMM_1x1_L16_SUB
- MY_ALIGN
-
-
- CGEMM_L1x1_SUB2_8:
- /*----------------------------------------*/
- andi. T1,L, 8
- ble CGEMM_L1x1_SUB2_4
- bl CGEMM_1x1_L8_SUB
- MY_ALIGN
-
-
- CGEMM_L1x1_SUB2_4:
- /*----------------------------------------*/
- andi. T1,L, 4
- ble CGEMM_L1x1_SUB2_2
- LOAD1x1_2
- KERNEL1x1_L2 16,16, 0,0
- KERNEL1x1_E2 16,16, 1,1
- MY_ALIGN
-
-
- CGEMM_L1x1_SUB2_2:
- /*----------------------------------------*/
- andi. T1,L, 2
- ble CGEMM_L1x1_SUB2_1
- LOAD1x1_2
- KERNEL1x1_E2 16,16, 0,1
- MY_ALIGN
-
-
- CGEMM_L1x1_SUB2_1:
- /*----------------------------------------*/
- andi. T1,L, 1
- ble CGEMM_L1x1_SAVE
- KERNEL1x1
-
- MY_ALIGN
- CGEMM_L1x1_SAVE:
- /*----------------------------------------*/
-
- SAVE1x1
- #if defined(TRMMKERNEL)
- REFRESH_AFTER_SAVE T6,K,TEMP_REG,BO,AO,1,1
- #endif
-
-
- CGEMM_L1x1_END:
- /*----------------------------------------*/
- slwi T1, K, 3
-
- add B, B, T1
- #if defined(TRMMKERNEL) && !defined(LEFT)
- addi TEMP_REG, TEMP_REG, 1
- #endif
-
- CGEMM_L1_END:
-
-
-
-
|