|
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816 |
- /***************************************************************************
- Copyright (c) 2013-2019, 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.
- *****************************************************************************/
-
- /**************************************************************************************
- * Abdelrauf(quickwritereader@gmail.com)
- * BLASTEST : OK
- * CTEST : OK
- * TEST : OK
- * LAPACK-TEST : OK
- **************************************************************************************/
- #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
- 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:
-
-
-
-
|