|
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431 |
- /*******************************************************************************
- Copyright (c) 2015, The OpenBLAS Project
- All rights reserved.
- Redistribution and use in source and binary forms, with or without
- modification, are permitted provided that the following conditions are
- met:
- 1. Redistributions of source code must retain the above copyright
- notice, this list of conditions and the following disclaimer.
- 2. Redistributions in binary form must reproduce the above copyright
- notice, this list of conditions and the following disclaimer in
- the documentation and/or other materials provided with the
- distribution.
- 3. Neither the name of the OpenBLAS project nor the names of
- its contributors may be used to endorse or promote products
- derived from this software without specific prior written permission.
- THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
- AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- ARE DISCLAIMED. IN NO EVENT SHALL THE OPENBLAS PROJECT OR CONTRIBUTORS BE
- LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
- SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
- CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
- OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
- USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- *******************************************************************************/
-
- #define ASSEMBLER
- #include "common.h"
-
- /* X0 X1 X2 s0 X3 x4 x5 x6 x7 */
- /*int CNAME(BLASLONG bm,BLASLONG bn,BLASLONG bk,FLOAT alpha,FLOAT* ba,FLOAT* bb,FLOAT* C,BLASLONG ldc, BLASLONG offset) */
-
- #define origM x0
- #define origN x1
- #define origK x2
- #define origPA x3
- #define origPB x4
- #define pC x5
- #define LDC x6
- #define offset x7
- #define counterL x8
- #define counterI x9
- #define counterJ x10
- #define pB x11
- #define pCRow0 x12
- #define pCRow1 x13
- #define pCRow2 x14
- #define pA x15
- #define temp x16
- #define tempOffset x17
- #define tempK x18
-
- #define alpha0 s10
- #define alphaV0 v10.s[0]
- #define alpha1 s11
- #define alphaV1 v11.s[0]
- #define alpha2 s14
- #define alphaV2 v14.s[0]
- #define alpha3 s15
- #define alphaV3 v15.s[0]
-
- // 00 origM
- // 01 origN
- // 02 origK
- // 03 origPA
- // 04 origPB
- // 05 pC
- // 06 origLDC -> LDC
- // 07 offset
- // 08 counterL
- // 09 counterI
- // 10 counterJ
- // 11 pB
- // 12 pCRow0
- // 13 pCRow1
- // 14 pCRow2
- // 15 pA
- // 16 temp
- // 17 tempOffset
- // 18 must save tempK
- // 19 must save
- // 20 must save
- // 21 must save
- // 22 must save
- // 23 must save
- // 24 must save
- // 25 must save
- // 26 must save
- // 27 must save
- // 28 must save
- // 29 frame
- // 30 link
- // 31 sp
-
- //v00 ALPHA -> pA0_00, pA0_01, pA0_02, pA0_03
- //v01 pA0_04, pA0_05, pA0_06, pA0_07
- //v02 pA0_08, pA0_09, pA0_10, pA0_11
- //v03 pA0_12, pA0_13, pA0_14, pA0_15
- //v04 pA1_00, pA1_01, pA1_02, pA1_03
- //v05 pA1_04, pA1_05, pA1_06, pA1_07
- //v06 pA1_08, pA1_09, pA1_10, pA1_11
- //v07 pA1_12, pA1_13, pA1_14, pA1_15
- //v08 must save pB00, pB01
- //v09 must save pB02, pB03
- //v10 must save ALPHA0
- //v11 must save ALPHA1
- //v12 must save pB10, pB11
- //v13 must save pB12, pB13
- //v14 must save ALPHA2
- //v15 must save ALPHA3
- //v16 must save C00, C01, C02, C03
- //v17 must save C04, C05, C06, C07
- //v18 C08, C09, C10, C11
- //v19 C12, C13, C14, C15
- //v20 C16, C17, C18, C19
- //v21 C20, C21, C22, C23
- //v22 C24, C25, C26, C27
- //v23 C28, C29, C30, C31
- //v24 C32, C33, C34, C35
- //v25 C36, C37, C38, C39
- //v26 C40, C41, C42, C43
- //v27 C44, C45, C46, C47
- //v28 C48, C49, C50, C51
- //v29 C52, C53, C54, C55
- //v30 C56, C57, C58, C59
- //v31 C60, C61, C62, C63
-
- /*******************************************************************************
- * Macro definitions
- *******************************************************************************/
-
- .macro INIT16x4
- fmov s16, wzr
- fmov s17, wzr
- fmov s18, s16
- fmov s19, s17
- fmov s20, wzr
- fmov s21, s16
- fmov s22, s17
- fmov s23, s18
- fmov s24, wzr
- fmov s25, s16
- fmov s26, s17
- fmov s27, s18
- fmov s28, wzr
- fmov s29, s16
- fmov s30, s17
- fmov s31, s18
- .endm
-
- .macro KERNEL16x4_I
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
- ld1 {v2.4s}, [pA]
- add pA, pA, #16
- ld1 {v3.4s}, [pA]
- add pA, pA, #16
-
- fmul v16.4s, v0.4s, v8.s[0]
- fmul v17.4s, v1.4s, v8.s[0]
- fmul v18.4s, v2.4s, v8.s[0]
- fmul v19.4s, v3.4s, v8.s[0]
-
- fmul v20.4s, v0.4s, v8.s[1]
- fmul v21.4s, v1.4s, v8.s[1]
- fmul v22.4s, v2.4s, v8.s[1]
- fmul v23.4s, v3.4s, v8.s[1]
-
- fmul v24.4s, v0.4s, v9.s[0]
- fmul v25.4s, v1.4s, v9.s[0]
- fmul v26.4s, v2.4s, v9.s[0]
- fmul v27.4s, v3.4s, v9.s[0]
-
- fmul v28.4s, v0.4s, v9.s[1]
- fmul v29.4s, v1.4s, v9.s[1]
- fmul v30.4s, v2.4s, v9.s[1]
- fmul v31.4s, v3.4s, v9.s[1]
-
- ld1 {v12.2s, v13.2s}, [pB]
- add pB, pB, #16
- ld1 {v4.4s}, [pA]
- add pA, pA, #16
- ld1 {v5.4s}, [pA]
- add pA, pA, #16
- ld1 {v6.4s}, [pA]
- add pA, pA, #16
- ld1 {v7.4s}, [pA]
- add pA, pA, #16
- .endm
-
- .macro KERNEL16x4_M1
- fmla v16.4s, v0.4s, v8.s[0]
- fmla v17.4s, v1.4s, v8.s[0]
- fmla v18.4s, v2.4s, v8.s[0]
- fmla v19.4s, v3.4s, v8.s[0]
-
- fmla v20.4s, v0.4s, v8.s[1]
- fmla v21.4s, v1.4s, v8.s[1]
- fmla v22.4s, v2.4s, v8.s[1]
- fmla v23.4s, v3.4s, v8.s[1]
-
- fmla v24.4s, v0.4s, v9.s[0]
- fmla v25.4s, v1.4s, v9.s[0]
- fmla v26.4s, v2.4s, v9.s[0]
- fmla v27.4s, v3.4s, v9.s[0]
-
- fmla v28.4s, v0.4s, v9.s[1]
- fmla v29.4s, v1.4s, v9.s[1]
- fmla v30.4s, v2.4s, v9.s[1]
- fmla v31.4s, v3.4s, v9.s[1]
-
- ld1 {v12.2s, v13.2s}, [pB]
- add pB, pB, #16
- ld1 {v4.4s}, [pA]
- add pA, pA, #16
- ld1 {v5.4s}, [pA]
- add pA, pA, #16
- ld1 {v6.4s}, [pA]
- add pA, pA, #16
- ld1 {v7.4s}, [pA]
- add pA, pA, #16
- .endm
-
- .macro KERNEL16x4_M2
- fmla v16.4s, v4.4s, v12.s[0]
- fmla v17.4s, v5.4s, v12.s[0]
- fmla v18.4s, v6.4s, v12.s[0]
- fmla v19.4s, v7.4s, v12.s[0]
-
- fmla v20.4s, v4.4s, v12.s[1]
- fmla v21.4s, v5.4s, v12.s[1]
- fmla v22.4s, v6.4s, v12.s[1]
- fmla v23.4s, v7.4s, v12.s[1]
-
- fmla v24.4s, v4.4s, v13.s[0]
- fmla v25.4s, v5.4s, v13.s[0]
- fmla v26.4s, v6.4s, v13.s[0]
- fmla v27.4s, v7.4s, v13.s[0]
-
- fmla v28.4s, v4.4s, v13.s[1]
- fmla v29.4s, v5.4s, v13.s[1]
- fmla v30.4s, v6.4s, v13.s[1]
- fmla v31.4s, v7.4s, v13.s[1]
-
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
- ld1 {v2.4s}, [pA]
- add pA, pA, #16
- ld1 {v3.4s}, [pA]
- add pA, pA, #16
- .endm
-
- .macro KERNEL16x4_E
- fmla v16.4s, v4.4s, v12.s[0]
- fmla v17.4s, v5.4s, v12.s[0]
- fmla v18.4s, v6.4s, v12.s[0]
- fmla v19.4s, v7.4s, v12.s[0]
-
- fmla v20.4s, v4.4s, v12.s[1]
- fmla v21.4s, v5.4s, v12.s[1]
- fmla v22.4s, v6.4s, v12.s[1]
- fmla v23.4s, v7.4s, v12.s[1]
-
- fmla v24.4s, v4.4s, v13.s[0]
- fmla v25.4s, v5.4s, v13.s[0]
- fmla v26.4s, v6.4s, v13.s[0]
- fmla v27.4s, v7.4s, v13.s[0]
-
- fmla v28.4s, v4.4s, v13.s[1]
- fmla v29.4s, v5.4s, v13.s[1]
- fmla v30.4s, v6.4s, v13.s[1]
- fmla v31.4s, v7.4s, v13.s[1]
- .endm
-
- .macro KERNEL16x4_SUB
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
- ld1 {v2.4s}, [pA]
- add pA, pA, #16
- ld1 {v3.4s}, [pA]
- add pA, pA, #16
-
- fmla v16.4s, v0.4s, v8.s[0]
- fmla v17.4s, v1.4s, v8.s[0]
- fmla v18.4s, v2.4s, v8.s[0]
- fmla v19.4s, v3.4s, v8.s[0]
-
- fmla v20.4s, v0.4s, v8.s[1]
- fmla v21.4s, v1.4s, v8.s[1]
- fmla v22.4s, v2.4s, v8.s[1]
- fmla v23.4s, v3.4s, v8.s[1]
-
- fmla v24.4s, v0.4s, v9.s[0]
- fmla v25.4s, v1.4s, v9.s[0]
- fmla v26.4s, v2.4s, v9.s[0]
- fmla v27.4s, v3.4s, v9.s[0]
-
- fmla v28.4s, v0.4s, v9.s[1]
- fmla v29.4s, v1.4s, v9.s[1]
- fmla v30.4s, v2.4s, v9.s[1]
- fmla v31.4s, v3.4s, v9.s[1]
- .endm
-
- .macro SAVE16x4
- add pCRow1, pCRow0, LDC
-
- fmul v0.4s, v16.4s, alphaV0
- fmul v1.4s, v17.4s, alphaV1
- fmul v2.4s, v18.4s, alphaV2
- fmul v3.4s, v19.4s, alphaV3
- st1 {v0.4s, v1.4s, v2.4s, v3.4s}, [pCRow0]
-
- add pCRow2, pCRow1, LDC
-
- fmul v4.4s, v20.4s, alphaV0
- fmul v5.4s, v21.4s, alphaV1
- fmul v6.4s, v22.4s, alphaV2
- fmul v7.4s, v23.4s, alphaV3
- st1 {v4.4s, v5.4s, v6.4s, v7.4s}, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
- fmul v0.4s, v24.4s, alphaV0
- fmul v1.4s, v25.4s, alphaV1
- fmul v2.4s, v26.4s, alphaV2
- fmul v3.4s, v27.4s, alphaV3
- st1 {v0.4s, v1.4s, v2.4s, v3.4s}, [pCRow2]
-
- fmul v4.4s, v28.4s, alphaV0
- fmul v5.4s, v29.4s, alphaV1
- fmul v6.4s, v30.4s, alphaV2
- fmul v7.4s, v31.4s, alphaV3
- st1 {v4.4s, v5.4s, v6.4s, v7.4s}, [pCRow1]
-
- add pCRow0, pCRow0, #64
- .endm
-
- /******************************************************************************/
-
- .macro INIT8x4
- fmov s16, wzr
- fmov s17, wzr
- fmov s20, wzr
- fmov s21, s16
- fmov s24, wzr
- fmov s25, s16
- fmov s28, wzr
- fmov s29, s16
- .endm
-
- .macro KERNEL8x4_I
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
-
- fmul v16.4s, v0.4s, v8.s[0]
- fmul v17.4s, v1.4s, v8.s[0]
- fmul v20.4s, v0.4s, v8.s[1]
- fmul v21.4s, v1.4s, v8.s[1]
- fmul v24.4s, v0.4s, v9.s[0]
- fmul v25.4s, v1.4s, v9.s[0]
- fmul v28.4s, v0.4s, v9.s[1]
- fmul v29.4s, v1.4s, v9.s[1]
-
- ld1 {v12.2s, v13.2s}, [pB]
- add pB, pB, #16
- ld1 {v4.4s}, [pA]
- add pA, pA, #16
- ld1 {v5.4s}, [pA]
- add pA, pA, #16
- .endm
-
- .macro KERNEL8x4_M1
- fmla v16.4s, v0.4s, v8.s[0]
- fmla v17.4s, v1.4s, v8.s[0]
- fmla v20.4s, v0.4s, v8.s[1]
- fmla v21.4s, v1.4s, v8.s[1]
- fmla v24.4s, v0.4s, v9.s[0]
- fmla v25.4s, v1.4s, v9.s[0]
- fmla v28.4s, v0.4s, v9.s[1]
- fmla v29.4s, v1.4s, v9.s[1]
-
- ld1 {v12.2s, v13.2s}, [pB]
- add pB, pB, #16
- ld1 {v4.4s}, [pA]
- add pA, pA, #16
- ld1 {v5.4s}, [pA]
- add pA, pA, #16
- .endm
-
- .macro KERNEL8x4_M2
- fmla v16.4s, v4.4s, v12.s[0]
- fmla v17.4s, v5.4s, v12.s[0]
- fmla v20.4s, v4.4s, v12.s[1]
- fmla v21.4s, v5.4s, v12.s[1]
- fmla v24.4s, v4.4s, v13.s[0]
- fmla v25.4s, v5.4s, v13.s[0]
- fmla v28.4s, v4.4s, v13.s[1]
- fmla v29.4s, v5.4s, v13.s[1]
-
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
- .endm
-
- .macro KERNEL8x4_E
- fmla v16.4s, v4.4s, v12.s[0]
- fmla v17.4s, v5.4s, v12.s[0]
- fmla v20.4s, v4.4s, v12.s[1]
- fmla v21.4s, v5.4s, v12.s[1]
- fmla v24.4s, v4.4s, v13.s[0]
- fmla v25.4s, v5.4s, v13.s[0]
- fmla v28.4s, v4.4s, v13.s[1]
- fmla v29.4s, v5.4s, v13.s[1]
- .endm
-
- .macro KERNEL8x4_SUB
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
-
- fmla v16.4s, v0.4s, v8.s[0]
- fmla v17.4s, v1.4s, v8.s[0]
- fmla v20.4s, v0.4s, v8.s[1]
- fmla v21.4s, v1.4s, v8.s[1]
- fmla v24.4s, v0.4s, v9.s[0]
- fmla v25.4s, v1.4s, v9.s[0]
- fmla v28.4s, v0.4s, v9.s[1]
- fmla v29.4s, v1.4s, v9.s[1]
- .endm
-
- .macro SAVE8x4
- add pCRow1, pCRow0, LDC
-
- fmul v0.4s, v16.4s, alphaV0
- fmul v1.4s, v17.4s, alphaV1
- st1 {v0.4s, v1.4s}, [pCRow0]
-
- add pCRow2, pCRow1, LDC
-
- fmul v4.4s, v20.4s, alphaV0
- fmul v5.4s, v21.4s, alphaV1
- st1 {v4.4s, v5.4s}, [pCRow1]
-
- add pCRow1, pCRow2, LDC
-
- fmul v0.4s, v24.4s, alphaV0
- fmul v1.4s, v25.4s, alphaV1
- st1 {v0.4s, v1.4s}, [pCRow2]
-
- fmul v4.4s, v28.4s, alphaV0
- fmul v5.4s, v29.4s, alphaV1
- st1 {v4.4s, v5.4s}, [pCRow1]
-
- add pCRow0, pCRow0, #32
- .endm
-
- /******************************************************************************/
-
- .macro INIT4x4
- fmov s16, wzr
- fmov s17, s16
- fmov s20, s17
- fmov s21, s16
- fmov s24, s17
- fmov s25, s16
- fmov s28, s17
- fmov s29, s16
- .endm
-
- .macro KERNEL4x4_I
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
- ld1 {v0.2s, v1.2s}, [pA]
- add pA, pA, #16
-
- fmul v16.2s, v0.2s, v8.s[0]
- fmul v29.2s, v1.2s, v9.s[1]
-
- fmul v20.2s, v0.2s, v8.s[1]
- fmul v25.2s, v1.2s, v9.s[0]
-
- fmul v24.2s, v0.2s, v9.s[0]
- fmul v21.2s, v1.2s, v8.s[1]
-
- fmul v28.2s, v0.2s, v9.s[1]
- fmul v17.2s, v1.2s, v8.s[0]
-
- ld1 {v12.2s, v13.2s}, [pB]
- add pB, pB, #16
- ld1 {v4.2s, v5.2s}, [pA]
- add pA, pA, #16
- .endm
-
- .macro KERNEL4x4_M1
- fmla v16.2s, v0.2s, v8.s[0]
- fmla v29.2s, v1.2s, v9.s[1]
-
- ld1 {v12.2s, v13.2s}, [pB] // For next round
- add pB, pB, #16
-
- fmla v20.2s, v0.2s, v8.s[1]
- fmla v25.2s, v1.2s, v9.s[0]
-
- ld1 {v4.2s, v5.2s}, [pA] // For next round
- add pA, pA, #16
-
- fmla v24.2s, v0.2s, v9.s[0]
- fmla v21.2s, v1.2s, v8.s[1]
-
- prfm PLDL1KEEP, [pB, #512]
-
- fmla v28.2s, v0.2s, v9.s[1]
- fmla v17.2s, v1.2s, v8.s[0]
- .endm
-
- .macro KERNEL4x4_M2
- fmla v16.2s, v4.2s, v12.s[0]
- fmla v29.2s, v5.2s, v13.s[1]
-
- ld1 {v8.2s, v9.2s}, [pB] // For next round
- add pB, pB, #16
-
- fmla v20.2s, v4.2s, v12.s[1]
- fmla v25.2s, v5.2s, v13.s[0]
-
- ld1 {v0.2s, v1.2s}, [pA] // For next round
- add pA, pA, #16
-
- fmla v24.2s, v4.2s, v13.s[0]
- fmla v21.2s, v5.2s, v12.s[1]
-
- prfm PLDL1KEEP, [pA, #512]
-
- fmla v28.2s, v4.2s, v13.s[1]
- fmla v17.2s, v5.2s, v12.s[0]
- .endm
-
- .macro KERNEL4x4_E
- fmla v16.2s, v4.2s, v12.s[0]
- fmla v29.2s, v5.2s, v13.s[1]
-
- fmla v20.2s, v4.2s, v12.s[1]
- fmla v25.2s, v5.2s, v13.s[0]
-
- fmla v24.2s, v4.2s, v13.s[0]
- fmla v21.2s, v5.2s, v12.s[1]
-
- fmla v28.2s, v4.2s, v13.s[1]
- fmla v17.2s, v5.2s, v12.s[0]
- .endm
-
- .macro KERNEL4x4_SUB
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
- ld1 {v0.2s, v1.2s}, [pA]
- add pA, pA, #16
-
- fmla v16.2s, v0.2s, v8.s[0]
- fmla v29.2s, v1.2s, v9.s[1]
-
- fmla v20.2s, v0.2s, v8.s[1]
- fmla v25.2s, v1.2s, v9.s[0]
-
- fmla v24.2s, v0.2s, v9.s[0]
- fmla v21.2s, v1.2s, v8.s[1]
-
- fmla v28.2s, v0.2s, v9.s[1]
- fmla v17.2s, v1.2s, v8.s[0]
- .endm
-
- .macro SAVE4x4
-
- fmul v8.2s, v16.2s, alphaV0
- fmul v9.2s, v17.2s, alphaV1
- st1 {v8.2s, v9.2s}, [pCRow0]
-
- add pCRow1, pCRow0, LDC
-
- fmul v12.2s, v20.2s, alphaV2
- fmul v13.2s, v21.2s, alphaV3
- st1 {v12.2s, v13.2s}, [pCRow1]
-
- add pCRow2, pCRow1, LDC
-
- fmul v8.2s, v24.2s, alphaV0
- fmul v9.2s, v25.2s, alphaV1
- st1 {v8.2s, v9.2s}, [pCRow2]
-
- add pCRow1, pCRow2, LDC
-
- fmul v12.2s, v28.2s, alphaV2
- fmul v13.2s, v29.2s, alphaV3
- st1 {v12.2s, v13.2s}, [pCRow1]
-
- add pCRow0, pCRow0, #16
- .endm
-
- /******************************************************************************/
-
- .macro INIT2x4
- fmov s16, wzr
- fmov s20, s16
- fmov s24, s20
- fmov s28, s16
- .endm
-
- .macro KERNEL2x4_SUB
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
- ld1 {v0.2s}, [pA]
- add pA, pA, #8
-
- fmla v16.2s, v0.2s, v8.s[0]
- fmla v20.2s, v0.2s, v8.s[1]
- fmla v24.2s, v0.2s, v9.s[0]
- fmla v28.2s, v0.2s, v9.s[1]
- .endm
-
- .macro SAVE2x4
- fmul v8.2s, v16.2s, alphaV0
- st1 {v8.2s}, [pCRow0]
-
- add pCRow1, pCRow0, LDC
- fmul v12.2s, v20.2s, alphaV1
- st1 {v12.2s}, [pCRow1]
-
- add pCRow2, pCRow1, LDC
- fmul v8.2s, v24.2s, alphaV2
- st1 {v8.2s}, [pCRow2]
-
- add pCRow1, pCRow2, LDC
- fmul v12.2s, v28.2s, alphaV3
- st1 {v12.2s}, [pCRow1]
-
- add pCRow0, pCRow0, #8
- .endm
-
- /******************************************************************************/
-
- .macro INIT1x4
- fmov s16, wzr
- fmov s20, s16
- .endm
-
- .macro KERNEL1x4_SUB
- ldr s0, [pA]
- add pA, pA, #4
-
- ld1 {v8.2s, v9.2s}, [pB]
- add pB, pB, #16
-
- fmla v16.2s, v8.2s, v0.s[0]
- fmla v20.2s, v9.2s, v0.s[0]
- .endm
-
- .macro SAVE1x4
- add pCRow1, pCRow0, LDC
-
- fmul v8.2s, v16.2s, alphaV0
- st1 {v8.s}[0], [pCRow0]
- st1 {v8.s}[1], [pCRow1]
-
- add pCRow2, pCRow1, LDC
- add pCRow1, pCRow2, LDC
-
- fmul v12.2s, v20.2s, alphaV1
- st1 {v12.s}[0], [pCRow2]
- st1 {v12.s}[1], [pCRow1]
-
- add pCRow0, pCRow0, #4
- .endm
-
- /******************************************************************************/
-
- .macro INIT16x2
- fmov s16, wzr
- fmov s17, wzr
- fmov s18, wzr
- fmov s19, s16
- fmov s20, wzr
- fmov s21, s16
- fmov s22, wzr
- fmov s23, s16
- .endm
-
- .macro KERNEL16x2_SUB
- ld1 {v8.2s}, [pB]
- add pB, pB, #8
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
- ld1 {v2.4s}, [pA]
- add pA, pA, #16
- ld1 {v3.4s}, [pA]
- add pA, pA, #16
-
- fmla v16.4s, v0.4s, v8.s[0]
- fmla v17.4s, v1.4s, v8.s[0]
- fmla v18.4s, v2.4s, v8.s[0]
- fmla v19.4s, v3.4s, v8.s[0]
-
- fmla v20.4s, v0.4s, v8.s[1]
- fmla v21.4s, v1.4s, v8.s[1]
- fmla v22.4s, v2.4s, v8.s[1]
- fmla v23.4s, v3.4s, v8.s[1]
- .endm
-
- .macro SAVE16x2
- add pCRow1, pCRow0, LDC
-
- fmul v0.4s, v16.4s, alphaV0
- fmul v1.4s, v17.4s, alphaV1
- fmul v2.4s, v18.4s, alphaV2
- fmul v3.4s, v19.4s, alphaV3
- st1 {v0.4s, v1.4s, v2.4s, v3.4s}, [pCRow0]
-
- fmul v4.4s, v20.4s, alphaV0
- fmul v5.4s, v21.4s, alphaV1
- fmul v6.4s, v22.4s, alphaV2
- fmul v7.4s, v23.4s, alphaV3
- st1 {v4.4s, v5.4s, v6.4s, v7.4s}, [pCRow1]
-
- add pCRow0, pCRow0, #64
- .endm
-
- /******************************************************************************/
-
- .macro INIT8x2
- fmov s16, wzr
- fmov s17, s16
- fmov s20, s17
- fmov s21, s16
- .endm
-
- .macro KERNEL8x2_SUB
- ld1 {v8.2s}, [pB]
- add pB, pB, #8
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
-
- fmla v16.4s, v0.4s, v8.s[0]
- fmla v17.4s, v1.4s, v8.s[0]
-
- fmla v20.4s, v0.4s, v8.s[1]
- fmla v21.4s, v1.4s, v8.s[1]
- .endm
-
- .macro SAVE8x2
- add pCRow1, pCRow0, LDC
-
- fmul v0.4s, v16.4s, alphaV0
- fmul v1.4s, v17.4s, alphaV1
- st1 {v0.4s, v1.4s}, [pCRow0]
-
- add pCRow2, pCRow1, LDC
-
- fmul v4.4s, v20.4s, alphaV0
- fmul v5.4s, v21.4s, alphaV1
- st1 {v4.4s, v5.4s}, [pCRow1]
-
- add pCRow0, pCRow0, #32
- .endm
-
- /******************************************************************************/
-
- .macro INIT4x2
- fmov s16, wzr
- fmov s17, s16
- fmov s20, s17
- fmov s21, s16
- .endm
-
- .macro KERNEL4x2_SUB
- ld1 {v8.2s}, [pB]
- add pB, pB, #8
- ld1 {v0.2s, v1.2s}, [pA]
- add pA, pA, #16
-
- fmla v16.2s, v0.2s, v8.s[0]
- fmla v17.2s, v1.2s, v8.s[0]
- fmla v20.2s, v0.2s, v8.s[1]
- fmla v21.2s, v1.2s, v8.s[1]
- .endm
-
- .macro SAVE4x2
-
- fmul v8.2s, v16.2s, alphaV0
- fmul v9.2s, v17.2s, alphaV1
- st1 {v8.2s, v9.2s}, [pCRow0]
-
- add pCRow1, pCRow0, LDC
-
- fmul v12.2s, v20.2s, alphaV2
- fmul v13.2s, v21.2s, alphaV3
- st1 {v12.2s, v13.2s}, [pCRow1]
-
- add pCRow0, pCRow0, #16
- .endm
-
- /******************************************************************************/
-
- .macro INIT2x2
- fmov s16, wzr
- fmov s20, s16
- .endm
-
- .macro KERNEL2x2_SUB
- ld1 {v8.2s}, [pB]
- add pB, pB, #8
-
- ld1 {v0.2s}, [pA]
- add pA, pA, #8
-
- fmla v16.2s, v0.2s, v8.s[0]
- fmla v20.2s, v0.2s, v8.s[1]
- .endm
-
- .macro SAVE2x2
- fmul v8.2s, v16.2s, alphaV0
- st1 {v8.2s}, [pCRow0]
-
- add pCRow1 , pCRow0, LDC
-
- fmul v12.2s, v20.2s, alphaV1
- st1 {v12.2s}, [pCRow1]
-
- add pCRow0, pCRow0, #8
- .endm
-
- /******************************************************************************/
-
- .macro INIT1x2
- fmov s16, wzr
- .endm
-
- .macro KERNEL1x2_SUB
- ld1 {v8.2s} , [pB]
- add pB , pB, #8
-
- ldr s0 , [pA]
- add pA, pA, #4
-
- fmla v16.2s, v8.2s, v0.s[0]
- .endm
-
- .macro SAVE1x2
- add pCRow1 , pCRow0, LDC
-
- fmul v8.2s, v16.2s, alphaV0
- st1 {v8.s}[0], [pCRow0]
- st1 {v8.s}[1], [pCRow1]
-
- add pCRow0, pCRow0, #4
- .endm
-
- /******************************************************************************/
-
- .macro INIT16x1
- fmov s16, wzr
- fmov s17, wzr
- fmov s18, wzr
- fmov s19, s16
- .endm
-
- .macro KERNEL16x1_SUB
- ldr s8, [pB]
- add pB , pB, #4
-
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
- ld1 {v2.4s}, [pA]
- add pA, pA, #16
- ld1 {v3.4s}, [pA]
- add pA, pA, #16
-
- fmla v16.4s, v0.4s, v8.s[0]
- fmla v17.4s, v1.4s, v8.s[0]
- fmla v18.4s, v2.4s, v8.s[0]
- fmla v19.4s, v3.4s, v8.s[0]
- .endm
-
- .macro SAVE16x1
-
- fmul v0.4s, v16.4s, alphaV0
- fmul v1.4s, v17.4s, alphaV1
- fmul v2.4s, v18.4s, alphaV2
- fmul v3.4s, v19.4s, alphaV3
- st1 {v0.4s, v1.4s, v2.4s, v3.4s}, [pCRow0]
-
- add pCRow0, pCRow0, #64
- .endm
-
- /******************************************************************************/
-
- .macro INIT8x1
- fmov s16, wzr
- fmov s17, wzr
- .endm
-
- .macro KERNEL8x1_SUB
- ldr s8, [pB]
- add pB , pB, #4
-
- ld1 {v0.4s}, [pA]
- add pA, pA, #16
- ld1 {v1.4s}, [pA]
- add pA, pA, #16
-
- fmla v16.4s, v0.4s, v8.s[0]
- fmla v17.4s, v1.4s, v8.s[0]
- .endm
-
- .macro SAVE8x1
-
- fmul v0.4s, v16.4s, alphaV0
- fmul v1.4s, v17.4s, alphaV1
- st1 {v0.4s, v1.4s}, [pCRow0]
-
- add pCRow0, pCRow0, #32
- .endm
-
- /******************************************************************************/
-
- .macro INIT4x1
- fmov s16, wzr
- fmov s17, s16
- .endm
-
- .macro KERNEL4x1_SUB
- ldr s8, [pB]
- add pB , pB, #4
-
- ld1 {v0.2s, v1.2s}, [pA]
- add pA , pA, #16
-
- fmla v16.2s, v0.2s, v8.s[0]
- fmla v17.2s, v1.2s, v8.s[0]
- .endm
-
- .macro SAVE4x1
-
- fmul v8.2s, v16.2s, alphaV0
- fmul v9.2s, v17.2s, alphaV1
- st1 {v8.2s, v9.2s}, [pCRow0]
-
- add pCRow0, pCRow0, #16
- .endm
-
- /******************************************************************************/
-
- .macro INIT2x1
- fmov s16, wzr
- .endm
-
- .macro KERNEL2x1_SUB
- ldr s8, [pB]
- add pB , pB, #4
-
- ld1 {v0.2s}, [pA]
- add pA , pA, #8
-
- fmla v16.2s, v0.2s, v8.s[0]
- .endm
-
- .macro SAVE2x1
-
- fmul v8.2s, v16.2s, alphaV0
- st1 {v8.2s}, [pCRow0]
-
- add pCRow0, pCRow0, #8
- .endm
-
- /******************************************************************************/
-
- .macro INIT1x1
- fmov s16, wzr
- .endm
-
- .macro KERNEL1x1_SUB
- ldr s8, [pB]
- add pB , pB, #4
-
- ldr s0, [pA]
- add pA , pA, #4
-
- fmadd s16, s0, s8, s16
- .endm
-
- .macro SAVE1x1
- fmul s8, s16, alpha0
- str s8, [pCRow0]
-
- add pCRow0, pCRow0, #4
- .endm
-
- /*******************************************************************************
- * End of macro definitions
- *******************************************************************************/
-
- PROLOGUE
-
- strmm_kernel_begin:
-
- .align 5
- add sp, sp, #-(11 * 16)
- stp d8, d9, [sp, #(0 * 16)]
- stp d10, d11, [sp, #(1 * 16)]
- stp d12, d13, [sp, #(2 * 16)]
- stp d14, d15, [sp, #(3 * 16)]
- stp d16, d17, [sp, #(4 * 16)]
- stp x18, x19, [sp, #(5 * 16)]
- stp x20, x21, [sp, #(6 * 16)]
- stp x22, x23, [sp, #(7 * 16)]
- stp x24, x25, [sp, #(8 * 16)]
- stp x26, x27, [sp, #(9 * 16)]
- str x28, [sp, #(10 * 16)]
-
- fmov alpha0, s0
- fmov alpha1, s0
- fmov alpha2, s0
- fmov alpha3, s0
-
- lsl LDC, LDC, #2 // ldc = ldc * 4
-
- #if !defined(LEFT)
- neg tempOffset, offset
- #endif
- mov pB, origPB
-
- mov counterJ, origN
- asr counterJ, counterJ, #2 // J = J / 4
- cmp counterJ, #0
- ble strmm_kernel_L2_BEGIN
-
- /******************************************************************************/
-
- strmm_kernel_L4_BEGIN:
- mov pCRow0, pC // pCRow0 = C
- add pC, pC, LDC, lsl #2
-
- #if defined(LEFT)
- mov tempOffset, offset
- #endif
- mov pA, origPA // pA = start of A array
-
- strmm_kernel_L4_M16_BEGIN:
-
- mov counterI, origM
- asr counterI, counterI, #4 // counterI = counterI / 16
- cmp counterI, #0
- ble strmm_kernel_L4_M8_BEGIN
-
- strmm_kernel_L4_M16_20:
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #6
- add pA, pA, temp
- lsl temp, tempOffset, #4
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #16
- #else
- add tempK, tempOffset, #4
- #endif
-
- asr counterL , tempK, #1 // L = K / 2
- cmp counterL , #2 // is there at least 4 to do?
- blt strmm_kernel_L4_M16_32
-
- KERNEL16x4_I // do one in the K
- KERNEL16x4_M2 // do another in the K
-
- subs counterL, counterL, #2
- ble strmm_kernel_L4_M16_22a
- .align 5
-
- strmm_kernel_L4_M16_22:
-
- KERNEL16x4_M1
- KERNEL16x4_M2
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L4_M16_22
-
- strmm_kernel_L4_M16_22a:
-
- KERNEL16x4_M1
- KERNEL16x4_E
-
- b strmm_kernel_L4_M16_44
-
- strmm_kernel_L4_M16_32:
-
- tst counterL, #1
- ble strmm_kernel_L4_M16_40
-
- KERNEL16x4_I
- KERNEL16x4_E
-
- b strmm_kernel_L4_M16_44
-
- strmm_kernel_L4_M16_40:
-
- INIT16x4
-
- strmm_kernel_L4_M16_44:
-
- ands counterL , tempK, #1
- ble strmm_kernel_L4_M16_100
-
- strmm_kernel_L4_M16_46:
-
- KERNEL16x4_SUB
-
- strmm_kernel_L4_M16_100:
-
- SAVE16x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #16
- #else
- sub tempK, tempK, #4
- #endif
- lsl temp, tempK, #6
- add pA, pA, temp
- lsl temp, tempK, #4
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #16
- #endif
-
- strmm_kernel_L4_M16_END:
- subs counterI, counterI, #1
- bne strmm_kernel_L4_M16_20
-
- //------------------------------------------------------------------------------
-
- strmm_kernel_L4_M8_BEGIN:
-
- mov counterI, origM
- tst counterI , #15
- ble strmm_kernel_L4_END
-
- tst counterI, #8
- ble strmm_kernel_L4_M4_BEGIN
-
- strmm_kernel_L4_M8_20:
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #5
- add pA, pA, temp
- lsl temp, tempOffset, #4
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #8
- #else
- add tempK, tempOffset, #4
- #endif
-
- asr counterL , tempK, #1 // L = K / 2
- cmp counterL , #2 // is there at least 4 to do?
- blt strmm_kernel_L4_M8_32
-
- KERNEL8x4_I // do one in the K
- KERNEL8x4_M2 // do another in the K
-
- subs counterL, counterL, #2
- ble strmm_kernel_L4_M8_22a
- .align 5
-
- strmm_kernel_L4_M8_22:
-
- KERNEL8x4_M1
- KERNEL8x4_M2
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L4_M8_22
-
- strmm_kernel_L4_M8_22a:
-
- KERNEL8x4_M1
- KERNEL8x4_E
-
- b strmm_kernel_L4_M8_44
-
- strmm_kernel_L4_M8_32:
-
- tst counterL, #1
- ble strmm_kernel_L4_M8_40
-
- KERNEL8x4_I
- KERNEL8x4_E
-
- b strmm_kernel_L4_M8_44
-
- strmm_kernel_L4_M8_40:
-
- INIT8x4
-
- strmm_kernel_L4_M8_44:
-
- ands counterL , tempK, #1
- ble strmm_kernel_L4_M8_100
-
- strmm_kernel_L4_M8_46:
-
- KERNEL8x4_SUB
-
- strmm_kernel_L4_M8_100:
-
- SAVE8x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #8
- #else
- sub tempK, tempK, #4
- #endif
- lsl temp, tempK, #5
- add pA, pA, temp
- lsl temp, tempK, #4
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #8
- #endif
-
- strmm_kernel_L4_M8_END:
-
- //------------------------------------------------------------------------------
-
- strmm_kernel_L4_M4_BEGIN:
-
- mov counterI, origM
- tst counterI , #7
- ble strmm_kernel_L4_END
-
- tst counterI, #4
- ble strmm_kernel_L4_M2_BEGIN
-
- strmm_kernel_L4_M4_20:
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #4
- add pB, pB, temp
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #4
- #else
- add tempK, tempOffset, #4
- #endif
- asr counterL , tempK, #1 // L = K / 2
- cmp counterL , #2 // is there at least 4 to do?
- blt strmm_kernel_L4_M4_32
-
- KERNEL4x4_I // do one in the K
- KERNEL4x4_M2 // do another in the K
-
- subs counterL, counterL, #2
- ble strmm_kernel_L4_M4_22a
- .align 5
-
- strmm_kernel_L4_M4_22:
-
- KERNEL4x4_M1
- KERNEL4x4_M2
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L4_M4_22
-
- strmm_kernel_L4_M4_22a:
-
- KERNEL4x4_M1
- KERNEL4x4_E
-
- b strmm_kernel_L4_M4_44
-
- strmm_kernel_L4_M4_32:
-
- tst counterL, #1
- ble strmm_kernel_L4_M4_40
-
- KERNEL4x4_I
- KERNEL4x4_E
-
- b strmm_kernel_L4_M4_44
-
- strmm_kernel_L4_M4_40:
-
- INIT4x4
-
- strmm_kernel_L4_M4_44:
-
- ands counterL , tempK, #1
- ble strmm_kernel_L4_M4_100
-
- strmm_kernel_L4_M4_46:
-
- KERNEL4x4_SUB
-
- strmm_kernel_L4_M4_100:
-
- SAVE4x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #4
- #else
- sub tempK, tempK, #4
- #endif
- lsl temp, tempK, #4
- add pA, pA, temp
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #4
- #endif
- strmm_kernel_L4_M4_END:
-
- //------------------------------------------------------------------------------
-
- strmm_kernel_L4_M2_BEGIN:
-
- mov counterI, origM
- tst counterI , #3
- ble strmm_kernel_L4_END
-
- tst counterI, #2 // counterI = counterI / 2
- ble strmm_kernel_L4_M1_BEGIN
-
- strmm_kernel_L4_M2_20:
-
- INIT2x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #3
- add pA, pA, temp
- lsl temp, tempOffset, #4
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #2
- #else
- add tempK, tempOffset, #4
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble strmm_kernel_L4_M2_40
-
- strmm_kernel_L4_M2_22:
-
- KERNEL2x4_SUB
- KERNEL2x4_SUB
- KERNEL2x4_SUB
- KERNEL2x4_SUB
-
- KERNEL2x4_SUB
- KERNEL2x4_SUB
- KERNEL2x4_SUB
- KERNEL2x4_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L4_M2_22
-
-
- strmm_kernel_L4_M2_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble strmm_kernel_L4_M2_100
-
- strmm_kernel_L4_M2_42:
-
- KERNEL2x4_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L4_M2_42
-
- strmm_kernel_L4_M2_100:
-
- SAVE2x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #2
- #else
- sub tempK, tempK, #4
- #endif
- lsl temp, tempK, #3
- add pA, pA, temp
- lsl temp, tempK, #4
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #2
- #endif
- strmm_kernel_L4_M2_END:
-
-
- strmm_kernel_L4_M1_BEGIN:
-
- tst counterI, #1 // counterI = counterI % 2
- ble strmm_kernel_L4_END
-
- strmm_kernel_L4_M1_20:
-
- INIT1x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #4
- add pB, pB, temp
- lsl temp, tempOffset, #2
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #1
- #else
- add tempK, tempOffset, #4
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble strmm_kernel_L4_M1_40
-
- strmm_kernel_L4_M1_22:
- KERNEL1x4_SUB
- KERNEL1x4_SUB
- KERNEL1x4_SUB
- KERNEL1x4_SUB
-
- KERNEL1x4_SUB
- KERNEL1x4_SUB
- KERNEL1x4_SUB
- KERNEL1x4_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L4_M1_22
-
-
- strmm_kernel_L4_M1_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble strmm_kernel_L4_M1_100
-
- strmm_kernel_L4_M1_42:
-
- KERNEL1x4_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L4_M1_42
-
- strmm_kernel_L4_M1_100:
-
- SAVE1x4
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #1
- #else
- sub tempK, tempK, #4
- #endif
- lsl temp, tempK, #2
- add pA, pA, temp
- lsl temp, tempK, #4
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #1
- #endif
- strmm_kernel_L4_END:
- add origPB, origPB, origK, lsl #4 // B = B + K * 4 * 4
- #if !defined(LEFT)
- add tempOffset, tempOffset, #4
- #endif
-
- subs counterJ, counterJ , #1 // j--
- bgt strmm_kernel_L4_BEGIN
-
-
- /******************************************************************************/
-
- strmm_kernel_L2_BEGIN: // less than 2 left in N direction
-
- mov counterJ , origN
- tst counterJ , #3
- ble strmm_kernel_L999
-
- tst counterJ , #2
- ble strmm_kernel_L1_BEGIN
-
- mov pCRow0, pC // pCRow0 = pC
-
- add pC,pC,LDC, lsl #1
-
- #if defined(LEFT)
- mov tempOffset, offset
- #endif
- mov pA, origPA // pA = A
-
- strmm_kernel_L2_M16_BEGIN:
-
- mov counterI, origM
- asr counterI, counterI, #4 // counterI = counterI / 16
- cmp counterI,#0
- ble strmm_kernel_L2_M8_BEGIN
-
- strmm_kernel_L2_M16_20:
-
- INIT16x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #6
- add pA, pA, temp
- lsl temp, tempOffset, #3
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #16
- #else
- add tempK, tempOffset, #2
- #endif
-
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL,#0
- ble strmm_kernel_L2_M16_40
- .align 5
-
- strmm_kernel_L2_M16_22:
- KERNEL16x2_SUB
- KERNEL16x2_SUB
- KERNEL16x2_SUB
- KERNEL16x2_SUB
-
- KERNEL16x2_SUB
- KERNEL16x2_SUB
- KERNEL16x2_SUB
- KERNEL16x2_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L2_M16_22
-
-
- strmm_kernel_L2_M16_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble strmm_kernel_L2_M16_100
-
- strmm_kernel_L2_M16_42:
-
- KERNEL16x2_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L2_M16_42
-
- strmm_kernel_L2_M16_100:
-
- SAVE16x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #16
- #else
- sub tempK, tempK, #2
- #endif
- lsl temp, tempK, #6
- add pA, pA, temp
- lsl temp, tempK, #3
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #16
- #endif
-
- strmm_kernel_L2_M16_END:
-
- subs counterI, counterI, #1
- bgt strmm_kernel_L2_M16_20
-
- //------------------------------------------------------------------------------
-
- strmm_kernel_L2_M8_BEGIN:
- mov counterI, origM
- tst counterI , #15
- ble strmm_kernel_L2_END
-
- tst counterI, #8
- ble strmm_kernel_L2_M4_BEGIN
-
- strmm_kernel_L2_M8_20:
-
- INIT8x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #5
- add pA, pA, temp
- lsl temp, tempOffset, #3
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #8
- #else
- add tempK, tempOffset, #2
- #endif
-
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL,#0
- ble strmm_kernel_L2_M8_40
- .align 5
-
- strmm_kernel_L2_M8_22:
- KERNEL8x2_SUB
- KERNEL8x2_SUB
- KERNEL8x2_SUB
- KERNEL8x2_SUB
-
- KERNEL8x2_SUB
- KERNEL8x2_SUB
- KERNEL8x2_SUB
- KERNEL8x2_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L2_M8_22
-
-
- strmm_kernel_L2_M8_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble strmm_kernel_L2_M8_100
-
- strmm_kernel_L2_M8_42:
-
- KERNEL8x2_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L2_M8_42
-
- strmm_kernel_L2_M8_100:
-
- SAVE8x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #8
- #else
- sub tempK, tempK, #2
- #endif
- lsl temp, tempK, #5
- add pA, pA, temp
- lsl temp, tempK, #3
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #8
- #endif
-
- strmm_kernel_L2_M8_END:
-
- //------------------------------------------------------------------------------
-
- strmm_kernel_L2_M4_BEGIN:
- mov counterI, origM
- tst counterI , #7
- ble strmm_kernel_L2_END
-
- tst counterI, #4
- ble strmm_kernel_L2_M2_BEGIN
-
- strmm_kernel_L2_M4_20:
-
- INIT4x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #3
- add pB, pB, temp
- lsl temp, tempOffset, #4
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #4
- #else
- add tempK, tempOffset, #2
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL,#0
- ble strmm_kernel_L2_M4_40
- .align 5
-
- strmm_kernel_L2_M4_22:
- KERNEL4x2_SUB
- KERNEL4x2_SUB
- KERNEL4x2_SUB
- KERNEL4x2_SUB
-
- KERNEL4x2_SUB
- KERNEL4x2_SUB
- KERNEL4x2_SUB
- KERNEL4x2_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L2_M4_22
-
-
- strmm_kernel_L2_M4_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble strmm_kernel_L2_M4_100
-
- strmm_kernel_L2_M4_42:
-
- KERNEL4x2_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L2_M4_42
-
- strmm_kernel_L2_M4_100:
-
- SAVE4x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #4
- #else
- sub tempK, tempK, #2
- #endif
- lsl temp, tempK, #4
- add pA, pA, temp
- lsl temp, tempK, #3
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #4
- #endif
- strmm_kernel_L2_M4_END:
-
- //------------------------------------------------------------------------------
-
-
- strmm_kernel_L2_M2_BEGIN:
-
- mov counterI, origM
- tst counterI , #3
- ble strmm_kernel_L2_END
-
- tst counterI, #2 // counterI = counterI / 2
- ble strmm_kernel_L2_M1_BEGIN
-
- strmm_kernel_L2_M2_20:
-
- INIT2x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #3
- add pB, pB, temp
- lsl temp, tempOffset, #3
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #2
- #else
- add tempK, tempOffset, #2
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL,#0
- ble strmm_kernel_L2_M2_40
-
- strmm_kernel_L2_M2_22:
-
- KERNEL2x2_SUB
- KERNEL2x2_SUB
- KERNEL2x2_SUB
- KERNEL2x2_SUB
-
- KERNEL2x2_SUB
- KERNEL2x2_SUB
- KERNEL2x2_SUB
- KERNEL2x2_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L2_M2_22
-
-
- strmm_kernel_L2_M2_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble strmm_kernel_L2_M2_100
-
- strmm_kernel_L2_M2_42:
-
- KERNEL2x2_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L2_M2_42
-
- strmm_kernel_L2_M2_100:
-
- SAVE2x2
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #2
- #else
- sub tempK, tempK, #2
- #endif
- lsl temp, tempK, #3
- add pA, pA, temp
- lsl temp, tempK, #3
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #2
- #endif
-
- strmm_kernel_L2_M2_END:
-
-
- strmm_kernel_L2_M1_BEGIN:
-
- tst counterI, #1 // counterI = counterI % 2
- ble strmm_kernel_L2_END
-
- strmm_kernel_L2_M1_20:
-
- INIT1x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #3
- add pB, pB, temp
- lsl temp, tempOffset, #2
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #1
- #else
- add tempK, tempOffset, #2
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL, #0
- ble strmm_kernel_L2_M1_40
-
- strmm_kernel_L2_M1_22:
- KERNEL1x2_SUB
- KERNEL1x2_SUB
- KERNEL1x2_SUB
- KERNEL1x2_SUB
-
- KERNEL1x2_SUB
- KERNEL1x2_SUB
- KERNEL1x2_SUB
- KERNEL1x2_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L2_M1_22
-
-
- strmm_kernel_L2_M1_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble strmm_kernel_L2_M1_100
-
- strmm_kernel_L2_M1_42:
-
- KERNEL1x2_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L2_M1_42
-
- strmm_kernel_L2_M1_100:
-
- SAVE1x2
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #1
- #else
- sub tempK, tempK, #2
- #endif
- lsl temp, tempK, #2
- add pA, pA, temp
- lsl temp, tempK, #3
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #1
- #endif
- strmm_kernel_L2_END:
- #if !defined(LEFT)
- add tempOffset, tempOffset, #2
- #endif
- add origPB, origPB, origK, lsl #3 // B = B + K * 2 * 4
-
- /******************************************************************************/
-
- strmm_kernel_L1_BEGIN:
-
- mov counterJ , origN
- tst counterJ , #1
- ble strmm_kernel_L999 // done
-
-
- mov pCRow0, pC // pCRow0 = C
- add pC , pC , LDC // Update pC to point to next
-
- #if defined(LEFT)
- mov tempOffset, offset
- #endif
- mov pA, origPA // pA = A
-
- strmm_kernel_L1_M16_BEGIN:
-
- mov counterI, origM
- asr counterI, counterI, #4 // counterI = counterI / 16
- cmp counterI, #0
- ble strmm_kernel_L1_M8_BEGIN
-
- strmm_kernel_L1_M16_20:
-
- INIT16x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #6
- add pA, pA, temp
- lsl temp, tempOffset, #2
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #16
- #else
- add tempK, tempOffset, #1
- #endif
-
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble strmm_kernel_L1_M16_40
- .align 5
-
- strmm_kernel_L1_M16_22:
- KERNEL16x1_SUB
- KERNEL16x1_SUB
- KERNEL16x1_SUB
- KERNEL16x1_SUB
-
- KERNEL16x1_SUB
- KERNEL16x1_SUB
- KERNEL16x1_SUB
- KERNEL16x1_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L1_M16_22
-
-
- strmm_kernel_L1_M16_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble strmm_kernel_L1_M16_100
-
- strmm_kernel_L1_M16_42:
-
- KERNEL16x1_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L1_M16_42
-
- strmm_kernel_L1_M16_100:
-
- SAVE16x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #16
- #else
- sub tempK, tempK, #1
- #endif
- lsl temp, tempK, #6
- add pA, pA, temp
- lsl temp, tempK, #2
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #16
- #endif
-
- strmm_kernel_L1_M16_END:
-
- subs counterI, counterI, #1
- bgt strmm_kernel_L1_M16_20
-
- //------------------------------------------------------------------------------
-
- strmm_kernel_L1_M8_BEGIN:
-
- mov counterI, origM
- tst counterI , #15
- ble strmm_kernel_L1_END
-
- tst counterI, #8
- ble strmm_kernel_L1_M4_BEGIN
-
- strmm_kernel_L1_M8_20:
-
- INIT8x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #5
- add pA, pA, temp
- lsl temp, tempOffset, #2
- add pB, pB, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #8
- #else
- add tempK, tempOffset, #1
- #endif
-
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble strmm_kernel_L1_M8_40
- .align 5
-
- strmm_kernel_L1_M8_22:
- KERNEL8x1_SUB
- KERNEL8x1_SUB
- KERNEL8x1_SUB
- KERNEL8x1_SUB
-
- KERNEL8x1_SUB
- KERNEL8x1_SUB
- KERNEL8x1_SUB
- KERNEL8x1_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L1_M8_22
-
-
- strmm_kernel_L1_M8_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble strmm_kernel_L1_M8_100
-
- strmm_kernel_L1_M8_42:
-
- KERNEL8x1_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L1_M8_42
-
- strmm_kernel_L1_M8_100:
-
- SAVE8x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #8
- #else
- sub tempK, tempK, #1
- #endif
- lsl temp, tempK, #5
- add pA, pA, temp
- lsl temp, tempK, #2
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #8
- #endif
-
- strmm_kernel_L1_M8_END:
-
- //------------------------------------------------------------------------------
-
- strmm_kernel_L1_M4_BEGIN:
- mov counterI, origM
- tst counterI , #7
- ble strmm_kernel_L1_END
-
- tst counterI, #4
- ble strmm_kernel_L1_M2_BEGIN
-
- strmm_kernel_L1_M4_20:
-
- INIT4x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #2
- add pB, pB, temp
- lsl temp, tempOffset, #4
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #4
- #else
- add tempK, tempOffset, #1
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble strmm_kernel_L1_M4_40
- .align 5
-
- strmm_kernel_L1_M4_22:
- KERNEL4x1_SUB
- KERNEL4x1_SUB
- KERNEL4x1_SUB
- KERNEL4x1_SUB
-
- KERNEL4x1_SUB
- KERNEL4x1_SUB
- KERNEL4x1_SUB
- KERNEL4x1_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L1_M4_22
-
-
- strmm_kernel_L1_M4_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble strmm_kernel_L1_M4_100
-
- strmm_kernel_L1_M4_42:
-
- KERNEL4x1_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L1_M4_42
-
- strmm_kernel_L1_M4_100:
-
- SAVE4x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #4
- #else
- sub tempK, tempK, #1
- #endif
- lsl temp, tempK, #4
- add pA, pA, temp
- lsl temp, tempK, #2
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #4
- #endif
- strmm_kernel_L1_M4_END:
-
- //------------------------------------------------------------------------------
-
- strmm_kernel_L1_M2_BEGIN:
-
- mov counterI, origM
- tst counterI , #3
- ble strmm_kernel_L1_END
-
- tst counterI, #2 // counterI = counterI / 2
- ble strmm_kernel_L1_M1_BEGIN
-
- strmm_kernel_L1_M2_20:
-
- INIT2x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #2
- add pB, pB, temp
- lsl temp, tempOffset, #3
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #2
- #else
- add tempK, tempOffset, #1
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble strmm_kernel_L1_M2_40
-
- strmm_kernel_L1_M2_22:
-
- KERNEL2x1_SUB
- KERNEL2x1_SUB
- KERNEL2x1_SUB
- KERNEL2x1_SUB
-
- KERNEL2x1_SUB
- KERNEL2x1_SUB
- KERNEL2x1_SUB
- KERNEL2x1_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L1_M2_22
-
-
- strmm_kernel_L1_M2_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble strmm_kernel_L1_M2_100
-
- strmm_kernel_L1_M2_42:
-
- KERNEL2x1_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L1_M2_42
-
- strmm_kernel_L1_M2_100:
-
- SAVE2x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- sub tempK, origK, tempOffset
- #if defined(LEFT)
- sub tempK, tempK, #2
- #else
- sub tempK, tempK, #1
- #endif
- lsl temp, tempK, #3
- add pA, pA, temp
- lsl temp, tempK, #2
- add pB, pB, temp
- #endif
- #if defined(LEFT)
- add tempOffset, tempOffset, #2
- #endif
- strmm_kernel_L1_M2_END:
-
-
- strmm_kernel_L1_M1_BEGIN:
-
- tst counterI, #1 // counterI = counterI % 2
- ble strmm_kernel_L1_END
-
- strmm_kernel_L1_M1_20:
-
- INIT1x1
-
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- mov pB, origPB
- #else
- mov pB, origPB
- lsl temp, tempOffset, #2
- add pB, pB, temp
- lsl temp, tempOffset, #2
- add pA, pA, temp
- #endif
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- sub tempK, origK, tempOffset
- #elif defined(LEFT)
- add tempK, tempOffset, #1
- #else
- add tempK, tempOffset, #1
- #endif
- asr counterL , tempK, #3 // counterL = counterL / 8
- cmp counterL , #0
- ble strmm_kernel_L1_M1_40
-
- strmm_kernel_L1_M1_22:
- KERNEL1x1_SUB
- KERNEL1x1_SUB
- KERNEL1x1_SUB
- KERNEL1x1_SUB
-
- KERNEL1x1_SUB
- KERNEL1x1_SUB
- KERNEL1x1_SUB
- KERNEL1x1_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L1_M1_22
-
-
- strmm_kernel_L1_M1_40:
-
- ands counterL , tempK, #7 // counterL = counterL % 8
- ble strmm_kernel_L1_M1_100
-
- strmm_kernel_L1_M1_42:
-
- KERNEL1x1_SUB
-
- subs counterL, counterL, #1
- bgt strmm_kernel_L1_M1_42
-
- strmm_kernel_L1_M1_100:
-
- SAVE1x1
-
- strmm_kernel_L1_END:
-
- strmm_kernel_L999:
- mov x0, #0 // set return value
- ldp d8, d9, [sp, #(0 * 16)]
- ldp d10, d11, [sp, #(1 * 16)]
- ldp d12, d13, [sp, #(2 * 16)]
- ldp d14, d15, [sp, #(3 * 16)]
- ldp d16, d17, [sp, #(4 * 16)]
- ldp x18, x19, [sp, #(5 * 16)]
- ldp x20, x21, [sp, #(6 * 16)]
- ldp x22, x23, [sp, #(7 * 16)]
- ldp x24, x25, [sp, #(8 * 16)]
- ldp x26, x27, [sp, #(9 * 16)]
- ldr x28, [sp, #(10 * 16)]
- add sp, sp, #(11*16)
- ret
-
- EPILOGUE
-
|