added optimized sgemm_tcopy for power8
This commit is contained in:
Werner Saar
176
kernel/power/sgemm_tcopy_16_power8.S
Normal file
176
kernel/power/sgemm_tcopy_16_power8.S
Normal file
@@ -0,0 +1,176 @@
|
||||
/*********************************************************************/
|
||||
/* Copyright 2009, 2010 The University of Texas at Austin. */
|
||||
/* 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. */
|
||||
/* */
|
||||
/* THIS SOFTWARE IS PROVIDED BY THE UNIVERSITY OF TEXAS AT */
|
||||
/* AUSTIN ``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 UNIVERSITY OF TEXAS AT */
|
||||
/* AUSTIN 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. */
|
||||
/* */
|
||||
/* The views and conclusions contained in the software and */
|
||||
/* documentation are those of the authors and should not be */
|
||||
/* interpreted as representing official policies, either expressed */
|
||||
/* or implied, of The University of Texas at Austin. */
|
||||
/*********************************************************************/
|
||||
|
||||
#define ASSEMBLER
|
||||
#include "common.h"
|
||||
#include "def_vsx.h"
|
||||
|
||||
#define M r3
|
||||
#define N r4
|
||||
#define A r5
|
||||
#define LDA r6
|
||||
#define B r7
|
||||
|
||||
#define A0 r8
|
||||
#define A1 r9
|
||||
#define A2 r10
|
||||
#define A3 r11
|
||||
|
||||
#define J r12
|
||||
|
||||
#define PREA r14
|
||||
#define PREB r15
|
||||
#define BO r16
|
||||
#define B8 r17
|
||||
#define B4 r18
|
||||
#define B2 r19
|
||||
#define B1 r20
|
||||
#define o4 r21
|
||||
#define T2 r22
|
||||
#define I r23
|
||||
#define o16 r24
|
||||
#define o32 r25
|
||||
#define o48 r26
|
||||
#define B16 r29
|
||||
#define M16 r30
|
||||
#define T1 r31
|
||||
|
||||
#define o0 0
|
||||
|
||||
#include "sgemm_tcopy_macros_16_power8.S"
|
||||
|
||||
#define STACKSIZE 384
|
||||
|
||||
|
||||
PROLOGUE
|
||||
PROFCODE
|
||||
|
||||
addi SP, SP, -STACKSIZE
|
||||
li r0, 0
|
||||
|
||||
std r31, 144(SP)
|
||||
std r30, 152(SP)
|
||||
std r29, 160(SP)
|
||||
std r28, 168(SP)
|
||||
std r27, 176(SP)
|
||||
std r26, 184(SP)
|
||||
std r25, 192(SP)
|
||||
std r24, 200(SP)
|
||||
std r23, 208(SP)
|
||||
std r22, 216(SP)
|
||||
std r21, 224(SP)
|
||||
std r20, 232(SP)
|
||||
std r19, 240(SP)
|
||||
std r18, 248(SP)
|
||||
std r17, 256(SP)
|
||||
std r16, 264(SP)
|
||||
std r15, 272(SP)
|
||||
std r14, 280(SP)
|
||||
|
||||
cmpwi cr0, M, 0
|
||||
ble- L999
|
||||
cmpwi cr0, N, 0
|
||||
ble- L999
|
||||
|
||||
slwi LDA, LDA, BASE_SHIFT
|
||||
slwi M16, M, 4 + BASE_SHIFT
|
||||
|
||||
li T1, -16
|
||||
li T2, -8
|
||||
li PREA, -4
|
||||
li PREB, -2
|
||||
|
||||
and B8, N, T1
|
||||
and B4, N, T2
|
||||
and B2, N, PREA
|
||||
and B1, N, PREB
|
||||
|
||||
mullw B8, B8, M
|
||||
mullw B4, B4, M
|
||||
mullw B2, B2, M
|
||||
mullw B1, B1, M
|
||||
|
||||
slwi B8, B8, BASE_SHIFT
|
||||
slwi B4, B4, BASE_SHIFT
|
||||
slwi B2, B2, BASE_SHIFT
|
||||
slwi B1, B1, BASE_SHIFT
|
||||
|
||||
add B8, B8, B
|
||||
add B4, B4, B
|
||||
add B2, B2, B
|
||||
add B1, B1, B
|
||||
|
||||
li PREA, 384
|
||||
li PREB, 384
|
||||
|
||||
li o4, 4
|
||||
li o16, 16
|
||||
li o32, 32
|
||||
li o48, 48
|
||||
|
||||
#include "sgemm_tcopy_logic_16_power8.S"
|
||||
|
||||
L999:
|
||||
|
||||
li r3, 0
|
||||
|
||||
ld r31, 144(SP)
|
||||
ld r30, 152(SP)
|
||||
ld r29, 160(SP)
|
||||
ld r28, 168(SP)
|
||||
ld r27, 176(SP)
|
||||
ld r26, 184(SP)
|
||||
ld r25, 192(SP)
|
||||
ld r24, 200(SP)
|
||||
ld r23, 208(SP)
|
||||
ld r22, 216(SP)
|
||||
ld r21, 224(SP)
|
||||
ld r20, 232(SP)
|
||||
ld r19, 240(SP)
|
||||
ld r18, 248(SP)
|
||||
ld r17, 256(SP)
|
||||
ld r16, 264(SP)
|
||||
ld r15, 272(SP)
|
||||
ld r14, 280(SP)
|
||||
|
||||
addi SP, SP, STACKSIZE
|
||||
|
||||
blr
|
||||
EPILOGUE
|
||||
|
||||
|
||||
Reference in New Issue
Block a user