|
- #define ASSEMBLER
- #include "common.h"
-
- #define FETCH ld
- #define gsLQC1(base,fq,ft,offset) .word(0x32<<26|base<<21|ft<<16|0x1<<15|offset<<6|0x1<<5|fq)
- #define gsSQC1(base,fq,ft,offset) .word(0x3A<<26|base<<21|ft<<16|0x1<<15|offset<<6|0x1<<5|fq)
-
-
- #define STACKSIZE 160
- #define M $4
- #define N $5
- #define K $6
- #define A $9
- #define B $10
- #define C $11
- #define LDC $8
-
- #define AO $12
- #define BO $13
-
- #define R12 12
- #define R13 13
-
- #define I $2
- #define J $3
- #define L $7
-
- #define CO1 $14
- #define CO2 $15
- #define PREA $16
- #define PREB $17
-
- #if defined(TRMMKERNEL)
- #define OFFSET $18
- #define KK $19
- #define TEMP $20
- #endif
-
- #define a1 $f0
- #define a2 $f1
- #define a3 $f2
- #define a4 $f3
-
- #define b1 $f4
- #define b2 $f5
- #define b3 $f6
- #define b4 $f7
-
- #define a5 $f8
- #define a6 $f9
- #define a7 $f10
- #define a8 $f11
-
- #define b5 $f12
- #define b6 $f13
- #define b7 $f15
- #define b8 $f16
-
- #define c11 $f14
- #define c12 $f17
- #define c13 $f18
- #define c14 $f19
- #define c21 $f20
- #define c22 $f21
- #define c23 $f22
- #define c24 $f23
- #define c31 $f24
- #define c32 $f25
- #define c33 $f26
- #define c34 $f27
- #define c41 $f28
- #define c42 $f29
- #define c43 $f30
- #define c44 $f31
-
- #define F0 0
- #define F1 1
- #define F2 2
- #define F3 3
- #define F4 4
- #define F5 5
- #define F6 6
- #define F7 7
- #define F8 8
- #define F9 9
- #define F10 10
- #define F11 11
- #define F12 12
- #define F13 13
- #define F14 14
- #define F15 15
- #define F16 16
- #define F17 17
- #define F18 18
- #define F19 19
- #define F20 20
- #define F21 21
- #define F22 22
- #define F23 23
- #define F24 24
- #define F25 25
- #define F26 26
- #define F27 27
- #define F28 28
- #define F29 29
- #define F30 30
- #define F31 31
-
- #define ALPHA_R $f15
- #define ALPHA_I $f16
-
- #################################
- ## MADD1 a*c
- ## MADD2 b*c
- ## MADD3 a*d
- ## MADD4 d*b
- ##################################
- #if defined(NN) || defined(NT) || defined(TN) || defined(TT)
- #define MADD1 MADD
- #define MADD2 MADD
- #define MADD3 MADD
- #define MADD4 NMSUB
- #endif
-
- #if defined(NR) || defined(NC) || defined(TR) || defined(TC)
- #define MADD1 MADD
- #define MADD2 MADD
- #define MADD3 NMSUB
- #define MADD4 MADD
- #endif
-
- #if defined(RN) || defined(RT) || defined(CN) || defined(CT)
- #define MADD1 MADD
- #define MADD2 NMSUB
- #define MADD3 MADD
- #define MADD4 MADD
- #endif
-
- #if defined(RR) || defined(RC) || defined(CR) || defined(CC)
- #define MADD1 MADD
- #define MADD2 NMSUB
- #define MADD3 NMSUB
- #define MADD4 NMSUB
- #endif
-
- PROLOGUE
-
- LDARG LDC, 0($sp)
- daddiu $sp, $sp, -STACKSIZE
-
- SDARG $16, 0($sp)
- SDARG $17, 8($sp)
- sdc1 $f24, 16($sp)
- sdc1 $f25, 24($sp)
- sdc1 $f26, 32($sp)
- sdc1 $f27, 40($sp)
- sdc1 $f28, 48($sp)
- sdc1 $f29, 56($sp)
-
- #if defined(TRMMKERNEL)
- SDARG $18, 64($sp)
- SDARG $19, 72($sp)
- SDARG $20, 80($sp)
-
- LDARG OFFSET, STACKSIZE + 8($sp)
- #endif
-
- #ifndef __64BIT__
- sdc1 $f20, 88($sp)
- sdc1 $f21, 96($sp)
- sdc1 $f22,104($sp)
- sdc1 $f23,112($sp)
- #endif
-
- dsra J, N, 1 # J=N/2
- ST ALPHA_R, 128($sp) # store alpha_r & alpha_i
-
- #if defined(TRMMKERNEL) && !defined(LEFT)
- neg KK, OFFSET
- #endif
-
- dsll LDC, LDC, ZBASE_SHIFT # LDC*SIZE*COMPSIZE
- blez J, .L20
- ST ALPHA_I, 136($sp)
-
-
- .align 5
- .L10:
- #if defined(TRMMKERNEL) && defined(LEFT)
- move KK, OFFSET
- #endif
-
- daddiu J, J, -1
- dsra I, M, 1 # I=M/2
-
- dsll PREB, K, 1 + ZBASE_SHIFT # PREA=K*2*2^4
- dsll PREA, K, 1 + ZBASE_SHIFT # PREA=K*2*2^4
-
- move CO1, C # Fix pointer Cx
- daddu CO2, C, LDC
-
- move AO, A # Reset AO
- blez I, .L30
- daddu PREA, PREA, A # PREA=A+panel size
-
- .L11:
- #if defined(TRMMKERNEL)
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- move BO, B
- #else
- dsll TEMP, KK, 1 + ZBASE_SHIFT
-
- daddu AO, AO, TEMP
- daddu BO, B, TEMP
- #endif
-
- MTC $0, c11 # Clear results regs
- LD a1, 0 * SIZE(AO)
- MOV c12, c11
- LD a2, 1 * SIZE(AO)
-
- MOV c13, c11
- LD b1, 0 * SIZE(BO)
- MOV c14, c11
- LD b2, 1 * SIZE(BO)
-
- MOV c21, c11
- LD a3, 2 * SIZE(AO)
- MOV c22, c11
- LD a4, 3 * SIZE(AO)
-
- MOV c23, c11
- LD b3, 2 * SIZE(BO)
- MOV c24, c11
- LD b4, 3 * SIZE(BO)
-
- FETCH $0, 0 * SIZE(CO2)
- MOV c31, c11
- MOV c32, c11
-
- FETCH $0, 0 * SIZE(CO1)
- MOV c33, c11
- MOV c34, c11
-
- FETCH $0, 4 * SIZE(CO2)
- MOV c41, c11
- MOV c42, c11
-
- FETCH $0, 4 * SIZE(CO1)
- MOV c43, c11
- MOV c44, c11
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- dsubu TEMP, K, KK
- #elif defined(LEFT)
- daddiu TEMP, KK, 2
- #else
- daddiu TEMP, KK, 2
- #endif
- dsra L, TEMP, 2
- daddu PREB, PREB, B # PREA=A+panel size
- blez L, .L15
- NOP
-
- #else
-
- dsra L, K, 2 # Unroll K 4 times
- move BO, B
-
- MTC $0, c11 # Clear results regs
- LD a1, 0 * SIZE(AO)
- MOV c12, c11
- LD a2, 1 * SIZE(AO)
-
- MOV c13, c11
- LD b1, 0 * SIZE(BO)
- MOV c14, c11
- LD b2, 1 * SIZE(BO)
-
- MOV c21, c11
- LD a3, 2 * SIZE(AO)
- MOV c22, c11
- LD a4, 3 * SIZE(AO)
-
- MOV c23, c11
- LD b3, 2 * SIZE(BO)
- MOV c24, c11
- LD b4, 3 * SIZE(BO)
-
- MOV c31, c11
- MOV c32, c11
- FETCH $0, 0 * SIZE(CO2)
-
- MOV c33, c11
- MOV c34, c11
- FETCH $0, 0 * SIZE(CO1)
-
- MOV c41, c11
- MOV c42, c11
- FETCH $0, 4 * SIZE(CO2)
-
- MOV c43, c11
- NOP
- FETCH $0, 4 * SIZE(CO1)
-
- daddu PREB, PREB, B # PREA=A+panel size
- blez L, .L15
- MOV c44, c11
- #endif
-
- .align 5
-
- .L12:
- LD a5, 4 * SIZE(AO)
- LD a6, 5 * SIZE(AO)
- MADD1 c11, c11, a1, b1 # axc A1xB1
- MADD3 c13, c13, a1, b2 # axd
-
- LD b5, 4 * SIZE(BO)
- LD b6, 5 * SIZE(BO)
- MADD2 c12, c12, a2, b1 # bxc
- MADD4 c14, c14, a2, b2 # bxd
-
- LD a7, 6 * SIZE(AO)
- LD a8, 7 * SIZE(AO)
- MADD1 c21, c21, a3, b1 # A2xB1
- MADD3 c23, c23, a3, b2
-
- LD b7, 6 * SIZE(BO)
- LD b8, 7 * SIZE(BO)
- MADD2 c22, c22, a4, b1
- MADD4 c24, c24, a4, b2
-
- FETCH $0, 4 * SIZE(PREA)
- FETCH $0, 4 * SIZE(PREB)
- MADD1 c31, c31, a1, b3 # A1xB2
- MADD3 c33, c33, a1, b4
-
- MADD2 c32, c32, a2, b3
- MADD4 c34, c34, a2, b4
-
- MADD1 c41, c41, a3, b3 # A2xB2
- MADD3 c43, c43, a3, b4
- MADD2 c42, c42, a4, b3
- MADD4 c44, c44, a4, b4
-
- LD a1, 8 * SIZE(AO)
- LD a2, 9 * SIZE(AO)
- MADD1 c11, c11, a5, b5 # axc A1xB1
- MADD3 c13, c13, a5, b6 # axd
-
- LD b1, 8 * SIZE(BO)
- LD b2, 9 * SIZE(BO)
- MADD2 c12, c12, a6, b5 # bxc
- MADD4 c14, c14, a6, b6 # bxd
-
- LD a3, 10 * SIZE(AO)
- LD a4, 11 * SIZE(AO)
- MADD1 c21, c21, a7, b5 # A2xB1
- MADD3 c23, c23, a7, b6
-
- LD b3, 10 * SIZE(BO)
- LD b4, 11 * SIZE(BO)
- MADD2 c22, c22, a8, b5
- MADD4 c24, c24, a8, b6
-
- FETCH $0, 8 * SIZE(PREA)
- FETCH $0, 8 * SIZE(PREB)
- MADD1 c31, c31, a5, b7 # A1xB2
- MADD3 c33, c33, a5, b8
-
- MADD2 c32, c32, a6, b7
- MADD4 c34, c34, a6, b8
-
- MADD1 c41, c41, a7, b7 # A2xB2
- MADD3 c43, c43, a7, b8
- MADD2 c42, c42, a8, b7
- MADD4 c44, c44, a8, b8
-
- LD a5, 12 * SIZE(AO)
- LD a6, 13 * SIZE(AO)
- MADD1 c11, c11, a1, b1 # axc A1xB1
- MADD3 c13, c13, a1, b2 # axd
-
- LD b5, 12 * SIZE(BO)
- LD b6, 13 * SIZE(BO)
- MADD2 c12, c12, a2, b1 # bxc
- MADD4 c14, c14, a2, b2 # bxd
-
- LD a7, 14 * SIZE(AO)
- LD a8, 15 * SIZE(AO)
- MADD1 c21, c21, a3, b1 # A2xB1
- MADD3 c23, c23, a3, b2
-
- LD b7, 14 * SIZE(BO)
- LD b8, 15 * SIZE(BO)
- MADD2 c22, c22, a4, b1
- MADD4 c24, c24, a4, b2
-
- FETCH $0, 12 * SIZE(PREA)
- MADD1 c31, c31, a1, b3 # A1xB2
- MADD3 c33, c33, a1, b4
- daddiu L, L, -1
-
- FETCH $0, 12 * SIZE(PREB)
- MADD2 c32, c32, a2, b3
- MADD4 c34, c34, a2, b4
- daddiu AO, AO, 16 * SIZE
-
- daddiu BO, BO, 16 * SIZE # 2nr*4kr*cmpx
- MADD1 c41, c41, a3, b3 # A2xB2
- MADD3 c43, c43, a3, b4
- daddu PREA, PREA, 16 * SIZE
-
- MADD2 c42, c42, a4, b3
- MADD4 c44, c44, a4, b4
- daddu PREB, PREB, 16 * SIZE
-
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- MADD1 c11, c11, a5, b5 # axc A1xB1
- MADD3 c13, c13, a5, b6 # axd
-
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- MADD2 c12, c12, a6, b5 # bxc
- MADD4 c14, c14, a6, b6 # bxd
-
- LD a3, 2 * SIZE(AO)
- LD a4, 3 * SIZE(AO)
- MADD1 c21, c21, a7, b5 # A2xB1
- MADD3 c23, c23, a7, b6
-
- LD b3, 2 * SIZE(BO)
- LD b4, 3 * SIZE(BO)
- MADD2 c22, c22, a8, b5
- MADD4 c24, c24, a8, b6
-
- FETCH $0, 0 * SIZE(PREA)
- FETCH $0, 0 * SIZE(PREB)
- MADD1 c31, c31, a5, b7 # A1xB2
- MADD3 c33, c33, a5, b8
-
- MADD2 c32, c32, a6, b7
- MADD4 c34, c34, a6, b8
-
- MADD1 c41, c41, a7, b7 # A2xB2
- MADD3 c43, c43, a7, b8
-
- MADD2 c42, c42, a8, b7
- bgtz L, .L12
- MADD4 c44, c44, a8, b8
-
- .align 5
-
- .L15:
- #ifndef TRMMKERNEL
- andi L, K, 3
- LD ALPHA_R, 128($sp)
- #else
- andi L, TEMP, 3
- LD ALPHA_R, 128($sp)
- #endif
- blez L, .L18
- LD ALPHA_I, 136($sp)
-
- .align 5
-
- .L16:
- daddiu BO, BO, 4 * SIZE # 2nr*1kr*cmpx
- daddiu AO, AO, 4 * SIZE # 2mr*1kr*cmpx
- MADD1 c11, c11, a1, b1 # axc A1xB1
- MADD3 c13, c13, a1, b2 # axd
-
- daddiu PREA, PREA, 4 * SIZE
- daddiu PREB, PREB, 4 * SIZE
- MADD2 c12, c12, a2, b1 # bxc
- MADD4 c14, c14, a2, b2 # bxd
-
- MADD1 c21, c21, a3, b1 # A2xB1
- MADD3 c23, c23, a3, b2
-
- MADD2 c22, c22, a4, b1
- MADD4 c24, c24, a4, b2
-
- FETCH $0, 0 * SIZE(PREA)
- MADD1 c31, c31, a1, b3 # A1xB2
- MADD3 c33, c33, a1, b4
- daddiu L, L, -1
-
- MADD2 c32, c32, a2, b3
- MADD4 c34, c34, a2, b4
-
- FETCH $0, 0 * SIZE(PREB)
- MADD1 c41, c41, a3, b3 # A2xB2
- MADD3 c43, c43, a3, b4
-
- MADD2 c42, c42, a4, b3
- MADD4 c44, c44, a4, b4
-
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- LD a3, 2 * SIZE(AO)
- LD a4, 3 * SIZE(AO)
- LD b3, 2 * SIZE(BO)
- LD b4, 3 * SIZE(BO)
- bgtz L, .L16
- NOP
-
- .L18:
- #ifndef TRMMKERNEL
- ADD c11, c14, c11
- LD a1, 0 * SIZE(CO1)
- ADD c12, c13, c12
- LD a2, 1 * SIZE(CO1)
- ADD c21, c24, c21
- LD b1, 2 * SIZE(CO1)
- ADD c22, c23, c22
- LD b2, 3 * SIZE(CO1)
-
- ADD c31, c34, c31
- LD a3, 0 * SIZE(CO2)
- ADD c32, c33, c32
- LD a4, 1 * SIZE(CO2)
- ADD c41, c44, c41
- LD b3, 2 * SIZE(CO2)
- ADD c42, c43, c42
- LD b4, 3 * SIZE(CO2)
-
- daddiu I, I, -1
- MADD a1, a1, ALPHA_R, c11
- MADD a2, a2, ALPHA_R, c12
- MADD b1, b1, ALPHA_R, c21
- MADD b2, b2, ALPHA_R, c22
-
- NMSUB a1, a1, ALPHA_I, c12
- MADD a2, a2, ALPHA_I, c11
- NMSUB b1, b1, ALPHA_I, c22
- MADD b2, b2, ALPHA_I, c21
-
- MADD a3, a3, ALPHA_R, c31
- MADD a4, a4, ALPHA_R, c32
- ST a1, 0 * SIZE(CO1)
- MADD b3, b3, ALPHA_R, c41
- MADD b4, b4, ALPHA_R, c42
- ST a2, 1 * SIZE(CO1)
-
- NMSUB a3, a3, ALPHA_I, c32
- MADD a4, a4, ALPHA_I, c31
- ST b1, 2 * SIZE(CO1)
-
- NMSUB b3, b3, ALPHA_I, c42
- MADD b4, b4, ALPHA_I, c41
- ST b2, 3 * SIZE(CO1)
-
- ST a3, 0 * SIZE(CO2)
- ST a4, 1 * SIZE(CO2)
- ST b3, 2 * SIZE(CO2)
- ST b4, 3 * SIZE(CO2)
-
- #else
- ADD c11, c14, c11
- ADD c12, c13, c12
- ADD c21, c24, c21
- ADD c22, c23, c22
-
- ADD c31, c34, c31
- ADD c32, c33, c32
- ADD c41, c44, c41
- ADD c42, c43, c42
-
- daddiu I, I, -1
- MUL a1, ALPHA_R, c11
- MUL a2, ALPHA_R, c12
- MUL b1, ALPHA_R, c21
- MUL b2, ALPHA_R, c22
-
- NMSUB a1, a1, ALPHA_I, c12
- MADD a2, a2, ALPHA_I, c11
- NMSUB b1, b1, ALPHA_I, c22
- MADD b2, b2, ALPHA_I, c21
-
- MUL a3, ALPHA_R, c31
- MUL a4, ALPHA_R, c32
- MUL b3, ALPHA_R, c41
- MUL b4, ALPHA_R, c42
-
- NMSUB a3, a3, ALPHA_I, c32
- MADD a4, a4, ALPHA_I, c31
- NMSUB b3, b3, ALPHA_I, c42
- MADD b4, b4, ALPHA_I, c41
-
- ST a1, 0 * SIZE(CO1)
- ST a2, 1 * SIZE(CO1)
- ST b1, 2 * SIZE(CO1)
- ST b2, 3 * SIZE(CO1)
-
- ST a3, 0 * SIZE(CO2)
- ST a4, 1 * SIZE(CO2)
- ST b3, 2 * SIZE(CO2)
- ST b4, 3 * SIZE(CO2)
-
-
- #if ( defined(LEFT) && defined(TRANSA)) || \
- (!defined(LEFT) && !defined(TRANSA))
- dsubu TEMP, K, KK
- #ifdef LEFT
- daddiu TEMP, TEMP, -2
- #else
- daddiu TEMP, TEMP, -2
- #endif
-
- dsll TEMP, TEMP, 1 + ZBASE_SHIFT
-
- daddu AO, AO, TEMP
- daddu BO, BO, TEMP
- #endif
-
- #ifdef LEFT
- daddiu KK, KK, 2
- #endif
- #endif
-
- dsll PREB, K, 1 + ZBASE_SHIFT # PREA=K*2*2^4
- daddiu CO1,CO1, 4 * SIZE
- bgtz I, .L11
- daddiu CO2,CO2, 4 * SIZE
-
- .align 5
- .L30:
- andi I, M, 1
- daddu C, C, LDC # Change C to next panel
-
- daddu PREB, PREB, B # PREA=A+panel size
- blez I, .L19
- daddu C, C, LDC # Change C to next panel
-
- #if defined(TRMMKERNEL)
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- move BO, B
- #else
- dsll L, KK, ZBASE_SHIFT # MR=1
- dsll TEMP, KK, 1 + ZBASE_SHIFT # NR=2
-
- daddu AO, AO, L
- daddu BO, B, TEMP
- #endif
-
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- MTC $0, c11 # Clear results regs
- MOV c12, c11
-
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- MOV c13, c11
- MOV c14, c11
-
- LD b3, 2 * SIZE(BO)
- LD b4, 3 * SIZE(BO)
- MOV c31, c11
- MOV c32, c11
-
- FETCH $0, 0 * SIZE(PREB)
- MOV c33, c11
- MOV c34, c11
-
- FETCH $0, 0 * SIZE(CO1)
- FETCH $0, 0 * SIZE(CO2)
- FETCH $0, 4 * SIZE(CO1)
- FETCH $0, 4 * SIZE(CO2)
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- dsubu TEMP, K, KK
- #elif defined(LEFT)
- daddiu TEMP, KK, 1 # MR=1
- #else
- daddiu TEMP, KK, 2 # NR=2
- #endif
- dsra L, TEMP, 2
- blez L, .L35
- NOP
-
- #else
-
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- dsra L, K, 2 # Unroll K 4 times
- move BO, B
-
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- MTC $0, c11 # Clear results regs
- MOV c12, c11
-
- LD b3, 2 * SIZE(BO)
- LD b4, 3 * SIZE(BO)
- MOV c13, c11
- MOV c14, c11
-
- FETCH $0, 0 * SIZE(PREB)
- MOV c31, c11
- MOV c32, c11
-
- FETCH $0, 0 * SIZE(CO1)
- FETCH $0, 0 * SIZE(CO2)
- FETCH $0, 4 * SIZE(CO1)
- FETCH $0, 4 * SIZE(CO2)
-
- MOV c33, c11
- blez L, .L35
- MOV c34, c11
- #endif
-
- .align 5
-
- .L32:
- LD a3, 2 * SIZE(AO)
- LD a4, 3 * SIZE(AO)
- MADD1 c11, c11, a1, b1 # axc A1xB1
- MADD3 c13, c13, a1, b2 # axd
-
- LD b5, 4 * SIZE(BO)
- LD b6, 5 * SIZE(BO)
- MADD2 c12, c12, a2, b1 # bxc
- MADD4 c14, c14, a2, b2 # bxd
-
- LD b7, 6 * SIZE(BO)
- LD b8, 7 * SIZE(BO)
- MADD1 c31, c31, a1, b3 # A1xB2
- MADD3 c33, c33, a1, b4
-
- FETCH $0, 4 * SIZE(PREB)
- MADD2 c32, c32, a2, b3
- MADD4 c34, c34, a2, b4
- NOP
-
- LD a5, 4 * SIZE(AO)
- LD a6, 5 * SIZE(AO)
- MADD1 c11, c11, a3, b5 # axc A1xB1
- MADD3 c13, c13, a3, b6 # axd
-
- LD b1, 8 * SIZE(BO)
- LD b2, 9 * SIZE(BO)
- MADD2 c12, c12, a4, b5 # bxc
- MADD4 c14, c14, a4, b6 # bxd
-
- LD b3, 10 * SIZE(BO)
- LD b4, 11 * SIZE(BO)
- MADD1 c31, c31, a3, b7 # A1xB2
- MADD3 c33, c33, a3, b8
-
- FETCH $0, 8 * SIZE(PREB)
- MADD2 c32, c32, a4, b7
- MADD4 c34, c34, a4, b8
- daddiu L, L, -1
-
- LD a7, 6 * SIZE(AO)
- LD a8, 7 * SIZE(AO)
- MADD1 c11, c11, a5, b1 # axc A1xB1
- MADD3 c13, c13, a5, b2 # axd
-
- LD b5, 12 * SIZE(BO)
- LD b6, 13 * SIZE(BO)
- MADD2 c12, c12, a6, b1 # bxc
- MADD4 c14, c14, a6, b2 # bxd
-
- LD b7, 14 * SIZE(BO)
- LD b8, 15 * SIZE(BO)
- MADD1 c31, c31, a5, b3 # A1xB2
- MADD3 c33, c33, a5, b4
-
- daddiu AO, AO, 8 * SIZE # 2mr*4kr*cmpx
- daddiu BO, BO, 16 * SIZE # 2nr*4kr*cmpx
-
- FETCH $0, 12 * SIZE(PREB)
- MADD2 c32, c32, a6, b3
- MADD4 c34, c34, a6, b4
- NOP
-
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- MADD1 c11, c11, a7, b5 # axc A1xB1
- MADD3 c13, c13, a7, b6 # axd
-
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- MADD2 c12, c12, a8, b5 # bxc
- MADD4 c14, c14, a8, b6 # bxd
-
- LD b3, 2 * SIZE(BO)
- LD b4, 3 * SIZE(BO)
- MADD1 c31, c31, a7, b7 # A1xB2
- NOP
-
- MADD3 c33, c33, a7, b8
- daddiu PREB, PREB, 16 * SIZE
-
- FETCH $0, 0 * SIZE(PREB)
- MADD2 c32, c32, a8, b7
- bgtz L, .L32
- MADD4 c34, c34, a8, b8
-
-
- .L35:
- #ifndef TRMMKERNEL
- andi L, K, 3
- LD ALPHA_R, 128($sp)
- #else
- andi L, TEMP, 3
- LD ALPHA_R, 128($sp)
- #endif
- blez L, .L38
- LD ALPHA_I, 136($sp)
- .align 5
-
- .L36:
- daddiu L, L, -1
- MADD1 c11, c11, a1, b1 # axc A1xB1
- MADD3 c13, c13, a1, b2 # axd
-
- daddiu BO, BO, 4 * SIZE # 2nr*1kr*cmpx
- MADD2 c12, c12, a2, b1 # bxc
- MADD4 c14, c14, a2, b2 # bxd
-
- daddiu AO, AO, 2 * SIZE # 2mr*1kr*cmpx
- MADD1 c31, c31, a1, b3 # A1xB2
- MADD3 c33, c33, a1, b4
-
- daddiu PREB, PREB, 4 * SIZE
- MADD2 c32, c32, a2, b3
- MADD4 c34, c34, a2, b4
-
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
-
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- LD b3, 2 * SIZE(BO)
- LD b4, 3 * SIZE(BO)
- bgtz L, .L36
- NOP
-
- .L38:
- #ifndef TRMMKERNEL
- ADD c11, c14, c11
- LD a1, 0 * SIZE(CO1)
- ADD c12, c13, c12
- LD a2, 1 * SIZE(CO1)
-
- ADD c31, c34, c31
- LD a3, 0 * SIZE(CO2)
- ADD c32, c33, c32
- LD a4, 1 * SIZE(CO2)
-
- MADD a1, a1, ALPHA_R, c11
- MADD a2, a2, ALPHA_R, c12
-
- MADD a3, a3, ALPHA_R, c31
- MADD a4, a4, ALPHA_R, c32
-
- NMSUB a1, a1, ALPHA_I, c12
- MADD a2, a2, ALPHA_I, c11
-
- NMSUB a3, a3, ALPHA_I, c32
- MADD a4, a4, ALPHA_I, c31
-
- ST a1, 0 * SIZE(CO1)
- ST a2, 1 * SIZE(CO1)
-
- ST a3, 0 * SIZE(CO2)
- ST a4, 1 * SIZE(CO2)
-
- daddiu CO1,CO1, 2 * SIZE
- daddiu CO2,CO2, 2 * SIZE
-
- #else
- ADD c11, c14, c11
- ADD c12, c13, c12
-
- ADD c31, c34, c31
- ADD c32, c33, c32
-
- MUL a1, ALPHA_R, c11
- MUL a2, ALPHA_R, c12
-
- MUL a3, ALPHA_R, c31
- MUL a4, ALPHA_R, c32
-
- NMSUB a1, a1, ALPHA_I, c12
- MADD a2, a2, ALPHA_I, c11
-
- NMSUB a3, a3, ALPHA_I, c32
- MADD a4, a4, ALPHA_I, c31
-
- ST a1, 0 * SIZE(CO1)
- ST a2, 1 * SIZE(CO1)
-
- ST a3, 0 * SIZE(CO2)
- ST a4, 1 * SIZE(CO2)
-
- daddiu CO1,CO1, 2 * SIZE
- daddiu CO2,CO2, 2 * SIZE
-
- #if ( defined(LEFT) && defined(TRANSA)) || \
- (!defined(LEFT) && !defined(TRANSA))
- dsubu TEMP, K, KK
- #ifdef LEFT
- daddiu TEMP, TEMP, -1
- #else
- daddiu TEMP, TEMP, -2
- #endif
- dsll L, TEMP, ZBASE_SHIFT
- dsll TEMP, TEMP, 1 + ZBASE_SHIFT
-
- daddu AO, AO, L
- daddu BO, BO, TEMP
- #endif
-
- #ifdef LEFT
- daddiu KK, KK, 1
- #endif
- #endif
-
- .align 5
-
- .L19:
- #if defined(TRMMKERNEL) && !defined(LEFT)
- daddiu KK, KK, 2
- #endif
-
- bgtz J, .L10
- move B, BO
-
- .align 5
-
- .L20:
- andi J, N, 1
- blez J, .L999
- dsll PREA, K, 1+ZBASE_SHIFT # PREA=K*2*2^4
-
- dsra I, M, 1 # I=M/2
- move CO1, C
-
- #if defined(TRMMKERNEL) && defined(LEFT)
- move KK, OFFSET
- #endif
-
- move AO, A # Reset AO
- blez I, .L29
- daddu PREA, PREA, A
-
- .L21:
- #if defined(TRMMKERNEL)
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- move BO, B
- #else
- dsll L, KK, 1 + ZBASE_SHIFT
- dsll TEMP, KK, ZBASE_SHIFT
-
- daddu AO, AO, L
- daddu BO, B, TEMP
- #endif
-
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- MTC $0, c11 # Clear results regs
- MOV c12, c11
-
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- MOV c13, c11
- MOV c14, c11
-
- LD a3, 2 * SIZE(AO)
- LD a4, 3 * SIZE(AO)
- MOV c21, c11
- MOV c22, c11
-
- FETCH $0, 0 * SIZE(PREA)
- MOV c23, c11
- MOV c24, c11
-
- FETCH $0, 0 * SIZE(CO1)
- FETCH $0, 4 * SIZE(CO1)
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- dsubu TEMP, K, KK
- #elif defined(LEFT)
- daddiu TEMP, KK, 2 # define Mr=2
- #else
- daddiu TEMP, KK, 1 # define NR=1
- #endif
- dsra L, TEMP, 2
- blez L, .L25
- NOP
-
- #else
- dsra L, K, 2 # Unroll K 4 times
- move BO, B
-
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- MTC $0, c11 # Clear results regs
- MOV c12, c11
-
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- MOV c13, c11
- MOV c14, c11
-
- LD a3, 2 * SIZE(AO)
- LD a4, 3 * SIZE(AO)
- MOV c21, c11
- MOV c22, c11
-
- FETCH $0, 0 * SIZE(PREA)
- MOV c23, c11
- MOV c24, c11
-
- FETCH $0, 0 * SIZE(CO1)
- FETCH $0, 4 * SIZE(CO1)
-
- blez L, .L25
- NOP
- #endif
-
- .align 5
-
- .L22:
- LD a5, 4 * SIZE(AO)
- LD a6, 5 * SIZE(AO)
- MADD1 c11, c11, a1, b1 # axc A1xB1
- MADD3 c13, c13, a1, b2 # axd
-
- LD b3, 2 * SIZE(BO)
- LD b4, 3 * SIZE(BO)
- MADD2 c12, c12, a2, b1 # bxc
- MADD4 c14, c14, a2, b2 # bxd
-
- LD a7, 6 * SIZE(AO)
- LD a8, 7 * SIZE(AO)
- MADD1 c21, c21, a3, b1 # A2xB1
- MADD3 c23, c23, a3, b2
-
- FETCH $0, 4 * SIZE(PREA)
- MADD2 c22, c22, a4, b1
- MADD4 c24, c24, a4, b2
-
- LD a1, 8 * SIZE(AO)
- LD a2, 9 * SIZE(AO)
- MADD1 c11, c11, a5, b3 # axc A1xB1
- MADD3 c13, c13, a5, b4 # axd
-
- LD b5, 4 * SIZE(BO)
- LD b6, 5 * SIZE(BO)
- MADD2 c12, c12, a6, b3 # bxc
- MADD4 c14, c14, a6, b4 # bxd
-
- LD a3, 10 * SIZE(AO)
- LD a4, 11 * SIZE(AO)
- MADD1 c21, c21, a7, b3 # A2xB1
- MADD3 c23, c23, a7, b4
-
- FETCH $0, 8 * SIZE(PREA)
- MADD2 c22, c22, a8, b3
- MADD4 c24, c24, a8, b4
- daddiu L, L, -1
-
- LD a5, 12 * SIZE(AO)
- LD a6, 13 * SIZE(AO)
- MADD1 c11, c11, a1, b5 # axc A1xB1
- MADD3 c13, c13, a1, b6 # axd
-
- LD b7, 6 * SIZE(BO)
- LD b8, 7 * SIZE(BO)
- MADD2 c12, c12, a2, b5 # bxc
- MADD4 c14, c14, a2, b6 # bxd
-
- LD a7, 14 * SIZE(AO)
- LD a8, 15 * SIZE(AO)
- MADD1 c21, c21, a3, b5 # A2xB1
- MADD3 c23, c23, a3, b6
-
- daddiu BO, BO, 8 * SIZE # 1nr*4kr*cmpx
- daddiu AO, AO, 16 * SIZE # 2mr*4kr*cmpx
-
- FETCH $0, 12 * SIZE(PREA)
- MADD2 c22, c22, a4, b5
- MADD4 c24, c24, a4, b6
- daddiu PREA, PREA, 16 * SIZE
-
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- MADD1 c11, c11, a5, b7 # axc A1xB1
- MADD3 c13, c13, a5, b8 # axd
-
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- MADD2 c12, c12, a6, b7 # bxc
- MADD4 c14, c14, a6, b8 # bxd
-
- LD a3, 2 * SIZE(AO)
- LD a4, 3 * SIZE(AO)
- MADD1 c21, c21, a7, b7 # A2xB1
- MADD3 c23, c23, a7, b8
-
- FETCH $0, 0 * SIZE(PREA)
- MADD2 c22, c22, a8, b7
- bgtz L, .L22
- MADD4 c24, c24, a8, b8
-
-
- .L25:
- #ifndef TRMMKERNEL
- andi L, K, 3
- LD ALPHA_R, 128($sp)
- #else
- andi L, TEMP, 3
- LD ALPHA_R, 128($sp)
- #endif
- blez L, .L28
- LD ALPHA_I, 136($sp)
- .align 3
-
- .L26:
- daddiu L, L, -1
- MADD1 c11, c11, a1, b1 # axc A1xB1
- MADD3 c13, c13, a1, b2 # axd
-
- daddiu BO, BO, 2 * SIZE # 2nr*1kr*cmpx
- MADD2 c12, c12, a2, b1 # bxc
- MADD4 c14, c14, a2, b2 # bxd
-
- daddiu AO, AO, 4 * SIZE # 2mr*1kr*cmpx
- MADD1 c21, c21, a3, b1 # A2xB1
- MADD3 c23, c23, a3, b2
-
- daddiu PREA, PREA, 4 * SIZE # 2mr*1kr*cmpx
- MADD2 c22, c22, a4, b1
- MADD4 c24, c24, a4, b2
-
- # gsLQC1(R12, F1, F0, 0) # R:a1 I:a2 Unroll K=4
- # gsLQC1(R12, F3, F2, 1) # R:a3 I:a4
- # gsLQC1(R13, F5, F4, 0) # R:b1 I:b2
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- LD a3, 2 * SIZE(AO)
- LD a4, 3 * SIZE(AO)
-
- bgtz L, .L26
- FETCH $0, 0 * SIZE(PREA)
-
- .L28:
- #ifndef TRMMKERNEL
- ADD c11, c14, c11
- LD a1, 0 * SIZE(CO1)
- ADD c12, c13, c12
- LD a2, 1 * SIZE(CO1)
- ADD c21, c24, c21
- LD b1, 2 * SIZE(CO1)
- ADD c22, c23, c22
- LD b2, 3 * SIZE(CO1)
-
- daddiu I, I, -1
- MADD a1, a1, ALPHA_R, c11
- MADD a2, a2, ALPHA_R, c12
- MADD b1, b1, ALPHA_R, c21
- MADD b2, b2, ALPHA_R, c22
-
- NMSUB a1, a1, ALPHA_I, c12
- MADD a2, a2, ALPHA_I, c11
- NMSUB b1, b1, ALPHA_I, c22
- MADD b2, b2, ALPHA_I, c21
-
- ST a1, 0 * SIZE(CO1)
- ST a2, 1 * SIZE(CO1)
- ST b1, 2 * SIZE(CO1)
- ST b2, 3 * SIZE(CO1)
-
- #else
- ADD c11, c14, c11
- ADD c12, c13, c12
- ADD c21, c24, c21
- ADD c22, c23, c22
-
- daddiu I, I, -1
- MUL a1, ALPHA_R, c11
- MUL a2, ALPHA_R, c12
- MUL b1, ALPHA_R, c21
- MUL b2, ALPHA_R, c22
-
- NMSUB a1, a1, ALPHA_I, c12
- MADD a2, a2, ALPHA_I, c11
- NMSUB b1, b1, ALPHA_I, c22
- MADD b2, b2, ALPHA_I, c21
-
- ST a1, 0 * SIZE(CO1)
- ST a2, 1 * SIZE(CO1)
- ST b1, 2 * SIZE(CO1)
- ST b2, 3 * SIZE(CO1)
-
- #if ( defined(LEFT) && defined(TRANSA)) || \
- (!defined(LEFT) && !defined(TRANSA))
- dsubu TEMP, K, KK
- #ifdef LEFT
- daddiu TEMP, TEMP, -2
- #else
- daddiu TEMP, TEMP, -1
- #endif
-
- dsll L, TEMP, 1 + ZBASE_SHIFT
- dsll TEMP, TEMP, ZBASE_SHIFT
-
- daddu AO, AO, L
- daddu BO, BO, TEMP
- #endif
-
- #ifdef LEFT
- daddiu KK, KK, 2
- #endif
- #endif
- daddiu CO1,CO1, 4 * SIZE
- bgtz I, .L21
- NOP
-
- .L29:
- andi I, M, 1
- blez I, .L999
- NOP
-
- #if defined(TRMMKERNEL)
- #if (defined(LEFT) && defined(TRANSA)) || (!defined(LEFT) && !defined(TRANSA))
- move BO, B
- #else
- dsll TEMP, KK, ZBASE_SHIFT
-
- daddu AO, AO, TEMP
- daddu BO, B, TEMP
- #endif
-
- # gsLQC1(R12, F1, F0, 0) # R:a1 I:a2
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- MTC $0, c11 # Clear results regs
- MOV c12, c11
-
- # gsLQC1(R13, F5, F4, 0) # R:b1 I:b2
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- MOV c13, c11
- MOV c14, c11
-
- FETCH $0, 0 * SIZE(PREA)
- FETCH $0, 4 * SIZE(PREA)
-
- #if (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA))
- dsubu TEMP, K, KK
- #elif defined(LEFT)
- daddiu TEMP, KK, 1
- #else
- daddiu TEMP, KK, 1
- #endif
- dsra L, TEMP, 2
- blez L, .L45
- NOP
-
- #else
- dsra L, K, 2 # Unroll K 4 times
- move BO, B
-
- # gsLQC1(R12, F1, F0, 0) # R:a1 I:a2
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- MTC $0, c11 # Clear results regs
- MOV c12, c11
-
- # gsLQC1(R13, F5, F4, 0) # R:b1 I:b2
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- MOV c13, c11
- MOV c14, c11
-
- FETCH $0, 0 * SIZE(PREA)
- FETCH $0, 4 * SIZE(PREA)
- blez L, .L45
- NOP
- #endif
-
- .align 3
-
- .L42:
- # gsLQC1(R12, F3, F2, 1) # R:a3 I:a4
- LD a3, 2 * SIZE(AO)
- LD a4, 3 * SIZE(AO)
- MADD1 c11, c11, a1, b1 # axc A1xB1
- MADD3 c13, c13, a1, b2 # axd
-
- # gsLQC1(R13, F7, F6, 1) # R:b2 I:b3
- LD b3, 2 * SIZE(BO)
- LD b4, 3 * SIZE(BO)
- MADD2 c12, c12, a2, b1 # bxc
- MADD4 c14, c14, a2, b2 # bxd
-
- # gsLQC1(R12, F9, F8, 2) # Unroll K=1
- LD a5, 4 * SIZE(AO)
- LD a6, 5 * SIZE(AO)
- MADD1 c11, c11, a3, b3 # axc A1xB1
- MADD3 c13, c13, a3, b4 # axd
-
- # gsLQC1(R13, F13, F12, 2)
- LD b5, 4 * SIZE(BO)
- LD b6, 5 * SIZE(BO)
- MADD2 c12, c12, a4, b3 # bxc
- MADD4 c14, c14, a4, b4 # bxd
-
- # gsLQC1(R12, F11, F10, 3)
- LD a7, 6 * SIZE(AO)
- LD a8, 7 * SIZE(AO)
- MADD1 c11, c11, a5, b5 # axc A1xB1
- MADD3 c13, c13, a5, b6 # axd
-
- daddiu L, L, -1
-
- # gsLQC1(R13, F16, F15, 3)
- LD b7, 6 * SIZE(BO)
- LD b8, 7 * SIZE(BO)
- MADD2 c12, c12, a6, b5 # bxc
- MADD4 c14, c14, a6, b6 # bxd
-
- daddiu AO, AO, 8 * SIZE # 2mr*4kr*cmpx
- daddiu BO, BO, 8 * SIZE # 2nr*4kr*cmpx
-
- # gsLQC1(R12, F1, F0, 0) # R:a1 I:a2 Unroll K=4
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- MADD1 c11, c11, a7, b7 # axc A1xB1
- MADD3 c13, c13, a7, b8 # axd
-
- # gsLQC1(R13, F5, F4, 0) # R:b1 I:b2
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- MADD2 c12, c12, a8, b7 # bxc
- MADD4 c14, c14, a8, b8 # bxd
-
- bgtz L, .L42
- NOP
-
-
- .align 5
-
- .L45:
- #ifndef TRMMKERNEL
- andi L, K, 3
- LD ALPHA_R, 128($sp)
- #else
- andi L, TEMP, 3
- LD ALPHA_R, 128($sp)
- #endif
- blez L, .L48
- LD ALPHA_I, 136($sp)
-
- .L46:
- daddiu L, L, -1
- daddiu BO, BO, 1 * SIZE * COMPSIZE # 2nr*1kr*cmpx
- daddiu AO, AO, 1 * SIZE * COMPSIZE # 2mr*1kr*cmpx
-
- MADD1 c11, c11, a1, b1 # axc A1xB1
- MADD3 c13, c13, a1, b2 # axd
- MADD2 c12, c12, a2, b1 # bxc
- MADD4 c14, c14, a2, b2 # bxd
-
- # gsLQC1(R12, F1, F0, 0) # R:a1 I:a2 Unroll K=4
- # gsLQC1(R13, F5, F4, 0) # R:b1 I:b2
- LD a1, 0 * SIZE(AO)
- LD a2, 1 * SIZE(AO)
- LD b1, 0 * SIZE(BO)
- LD b2, 1 * SIZE(BO)
- bgtz L, .L46
- NOP
-
- .L48:
- #ifndef TRMMKERNEL
- ADD c11, c14, c11
- ADD c12, c13, c12
-
- LD a1, 0 * SIZE(CO1)
- LD a2, 1 * SIZE(CO1)
-
- MADD a1, a1, ALPHA_R, c11
- MADD a2, a2, ALPHA_R, c12
-
- NMSUB a1, a1, ALPHA_I, c12
- MADD a2, a2, ALPHA_I, c11
-
- ST a1, 0 * SIZE(CO1)
- ST a2, 1 * SIZE(CO1)
-
- #else
- ADD c11, c14, c11
- ADD c12, c13, c12
-
- MUL a1, ALPHA_R, c11
- MUL a2, ALPHA_R, c12
-
- NMSUB a1, a1, ALPHA_I, c12
- MADD a2, a2, ALPHA_I, c11
-
- ST a1, 0 * SIZE(CO1)
- ST a2, 1 * SIZE(CO1)
-
- #if ( defined(LEFT) && defined(TRANSA)) || \
- (!defined(LEFT) && !defined(TRANSA))
- dsubu TEMP, K, KK
- #ifdef LEFT
- daddiu TEMP, TEMP, -1
- #else
- daddiu TEMP, TEMP, -1
- #endif
-
- dsll TEMP, TEMP, ZBASE_SHIFT
-
- daddu AO, AO, TEMP
- daddu BO, BO, TEMP
- #endif
-
- #ifdef LEFT
- daddiu KK, KK, 1
- #endif
-
- daddiu CO1,CO1, 2 * SIZE
- #endif
-
-
-
- .align 5
-
- .L999:
- LDARG $16, 0($sp)
- LDARG $17, 8($sp)
- ldc1 $f24, 16($sp)
- ldc1 $f25, 24($sp)
- ldc1 $f26, 32($sp)
- ldc1 $f27, 40($sp)
- ldc1 $f28, 48($sp)
- ldc1 $f29, 56($sp)
-
- #if defined(TRMMKERNEL)
- LDARG $18, 64($sp)
- LDARG $19, 72($sp)
- LDARG $20, 80($sp)
- #endif
-
- #ifndef __64BIT__
- ldc1 $f20, 88($sp)
- ldc1 $f21, 96($sp)
- ldc1 $f22,104($sp)
- ldc1 $f23,112($sp)
- #endif
-
- j $31
- daddiu $sp, $sp, STACKSIZE
-
- EPILOGUE
|