|
kusano |
2b45e8 |
/*********************************************************************/
|
|
kusano |
2b45e8 |
/* Copyright 2009, 2010 The University of Texas at Austin. */
|
|
kusano |
2b45e8 |
/* All rights reserved. */
|
|
kusano |
2b45e8 |
/* */
|
|
kusano |
2b45e8 |
/* Redistribution and use in source and binary forms, with or */
|
|
kusano |
2b45e8 |
/* without modification, are permitted provided that the following */
|
|
kusano |
2b45e8 |
/* conditions are met: */
|
|
kusano |
2b45e8 |
/* */
|
|
kusano |
2b45e8 |
/* 1. Redistributions of source code must retain the above */
|
|
kusano |
2b45e8 |
/* copyright notice, this list of conditions and the following */
|
|
kusano |
2b45e8 |
/* disclaimer. */
|
|
kusano |
2b45e8 |
/* */
|
|
kusano |
2b45e8 |
/* 2. Redistributions in binary form must reproduce the above */
|
|
kusano |
2b45e8 |
/* copyright notice, this list of conditions and the following */
|
|
kusano |
2b45e8 |
/* disclaimer in the documentation and/or other materials */
|
|
kusano |
2b45e8 |
/* provided with the distribution. */
|
|
kusano |
2b45e8 |
/* */
|
|
kusano |
2b45e8 |
/* THIS SOFTWARE IS PROVIDED BY THE UNIVERSITY OF TEXAS AT */
|
|
kusano |
2b45e8 |
/* AUSTIN ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, */
|
|
kusano |
2b45e8 |
/* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF */
|
|
kusano |
2b45e8 |
/* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE */
|
|
kusano |
2b45e8 |
/* DISCLAIMED. IN NO EVENT SHALL THE UNIVERSITY OF TEXAS AT */
|
|
kusano |
2b45e8 |
/* AUSTIN OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, */
|
|
kusano |
2b45e8 |
/* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES */
|
|
kusano |
2b45e8 |
/* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE */
|
|
kusano |
2b45e8 |
/* GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR */
|
|
kusano |
2b45e8 |
/* BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF */
|
|
kusano |
2b45e8 |
/* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT */
|
|
kusano |
2b45e8 |
/* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT */
|
|
kusano |
2b45e8 |
/* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE */
|
|
kusano |
2b45e8 |
/* POSSIBILITY OF SUCH DAMAGE. */
|
|
kusano |
2b45e8 |
/* */
|
|
kusano |
2b45e8 |
/* The views and conclusions contained in the software and */
|
|
kusano |
2b45e8 |
/* documentation are those of the authors and should not be */
|
|
kusano |
2b45e8 |
/* interpreted as representing official policies, either expressed */
|
|
kusano |
2b45e8 |
/* or implied, of The University of Texas at Austin. */
|
|
kusano |
2b45e8 |
/*********************************************************************/
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
/* This file is a template for level 3 operation */
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#ifndef BETA_OPERATION
|
|
kusano |
2b45e8 |
#if !defined(XDOUBLE) || !defined(QUAD_PRECISION)
|
|
kusano |
2b45e8 |
#ifndef COMPLEX
|
|
kusano |
2b45e8 |
#define BETA_OPERATION(M_FROM, M_TO, N_FROM, N_TO, BETA, C, LDC) \
|
|
kusano |
2b45e8 |
GEMM_BETA((M_TO) - (M_FROM), (N_TO - N_FROM), 0, \
|
|
kusano |
2b45e8 |
BETA[0], NULL, 0, NULL, 0, \
|
|
kusano |
2b45e8 |
(FLOAT *)(C) + ((M_FROM) + (N_FROM) * (LDC)) * COMPSIZE, LDC)
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
#define BETA_OPERATION(M_FROM, M_TO, N_FROM, N_TO, BETA, C, LDC) \
|
|
kusano |
2b45e8 |
GEMM_BETA((M_TO) - (M_FROM), (N_TO - N_FROM), 0, \
|
|
kusano |
2b45e8 |
BETA[0], BETA[1], NULL, 0, NULL, 0, \
|
|
kusano |
2b45e8 |
(FLOAT *)(C) + ((M_FROM) + (N_FROM) * (LDC)) * COMPSIZE, LDC)
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
#define BETA_OPERATION(M_FROM, M_TO, N_FROM, N_TO, BETA, C, LDC) \
|
|
kusano |
2b45e8 |
GEMM_BETA((M_TO) - (M_FROM), (N_TO - N_FROM), 0, \
|
|
kusano |
2b45e8 |
BETA, NULL, 0, NULL, 0, \
|
|
kusano |
2b45e8 |
(FLOAT *)(C) + ((M_FROM) + (N_FROM) * (LDC)) * COMPSIZE, LDC)
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#ifndef ICOPY_OPERATION
|
|
kusano |
2b45e8 |
#if defined(NN) || defined(NT) || defined(NC) || defined(NR) || \
|
|
kusano |
2b45e8 |
defined(RN) || defined(RT) || defined(RC) || defined(RR)
|
|
kusano |
2b45e8 |
#define ICOPY_OPERATION(M, N, A, LDA, X, Y, BUFFER) GEMM_ITCOPY(M, N, (FLOAT *)(A) + ((Y) + (X) * (LDA)) * COMPSIZE, LDA, BUFFER);
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
#define ICOPY_OPERATION(M, N, A, LDA, X, Y, BUFFER) GEMM_INCOPY(M, N, (FLOAT *)(A) + ((X) + (Y) * (LDA)) * COMPSIZE, LDA, BUFFER);
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#ifndef OCOPY_OPERATION
|
|
kusano |
2b45e8 |
#if defined(NN) || defined(TN) || defined(CN) || defined(RN) || \
|
|
kusano |
2b45e8 |
defined(NR) || defined(TR) || defined(CR) || defined(RR)
|
|
kusano |
2b45e8 |
#define OCOPY_OPERATION(M, N, A, LDA, X, Y, BUFFER) GEMM_ONCOPY(M, N, (FLOAT *)(A) + ((X) + (Y) * (LDA)) * COMPSIZE, LDA, BUFFER);
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
#define OCOPY_OPERATION(M, N, A, LDA, X, Y, BUFFER) GEMM_OTCOPY(M, N, (FLOAT *)(A) + ((Y) + (X) * (LDA)) * COMPSIZE, LDA, BUFFER);
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#ifndef KERNEL_FUNC
|
|
kusano |
2b45e8 |
#if defined(NN) || defined(NT) || defined(TN) || defined(TT)
|
|
kusano |
2b45e8 |
#define KERNEL_FUNC GEMM_KERNEL_N
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#if defined(CN) || defined(CT) || defined(RN) || defined(RT)
|
|
kusano |
2b45e8 |
#define KERNEL_FUNC GEMM_KERNEL_L
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#if defined(NC) || defined(TC) || defined(NR) || defined(TR)
|
|
kusano |
2b45e8 |
#define KERNEL_FUNC GEMM_KERNEL_R
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#if defined(CC) || defined(CR) || defined(RC) || defined(RR)
|
|
kusano |
2b45e8 |
#define KERNEL_FUNC GEMM_KERNEL_B
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#ifndef KERNEL_OPERATION
|
|
kusano |
2b45e8 |
#if !defined(XDOUBLE) || !defined(QUAD_PRECISION)
|
|
kusano |
2b45e8 |
#ifndef COMPLEX
|
|
kusano |
2b45e8 |
#define KERNEL_OPERATION(M, N, K, ALPHA, SA, SB, C, LDC, X, Y) \
|
|
kusano |
2b45e8 |
KERNEL_FUNC(M, N, K, ALPHA[0], SA, SB, (FLOAT *)(C) + ((X) + (Y) * LDC) * COMPSIZE, LDC)
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
#define KERNEL_OPERATION(M, N, K, ALPHA, SA, SB, C, LDC, X, Y) \
|
|
kusano |
2b45e8 |
KERNEL_FUNC(M, N, K, ALPHA[0], ALPHA[1], SA, SB, (FLOAT *)(C) + ((X) + (Y) * LDC) * COMPSIZE, LDC)
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
#define KERNEL_OPERATION(M, N, K, ALPHA, SA, SB, C, LDC, X, Y) \
|
|
kusano |
2b45e8 |
KERNEL_FUNC(M, N, K, ALPHA, SA, SB, (FLOAT *)(C) + ((X) + (Y) * LDC) * COMPSIZE, LDC)
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#ifndef FUSED_KERNEL_OPERATION
|
|
kusano |
2b45e8 |
#if defined(NN) || defined(TN) || defined(CN) || defined(RN) || \
|
|
kusano |
2b45e8 |
defined(NR) || defined(TR) || defined(CR) || defined(RR)
|
|
kusano |
2b45e8 |
#ifndef COMPLEX
|
|
kusano |
2b45e8 |
#define FUSED_KERNEL_OPERATION(M, N, K, ALPHA, SA, SB, B, LDB, C, LDC, I, J, L) \
|
|
kusano |
2b45e8 |
FUSED_GEMM_KERNEL_N(M, N, K, ALPHA[0], SA, SB, \
|
|
kusano |
2b45e8 |
(FLOAT *)(B) + ((L) + (J) * LDB) * COMPSIZE, LDB, (FLOAT *)(C) + ((I) + (J) * LDC) * COMPSIZE, LDC)
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
#define FUSED_KERNEL_OPERATION(M, N, K, ALPHA, SA, SB, B, LDB, C, LDC, I, J, L) \
|
|
kusano |
2b45e8 |
FUSED_GEMM_KERNEL_N(M, N, K, ALPHA[0], ALPHA[1], SA, SB, \
|
|
kusano |
2b45e8 |
(FLOAT *)(B) + ((L) + (J) * LDB) * COMPSIZE, LDB, (FLOAT *)(C) + ((I) + (J) * LDC) * COMPSIZE, LDC)
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
#ifndef COMPLEX
|
|
kusano |
2b45e8 |
#define FUSED_KERNEL_OPERATION(M, N, K, ALPHA, SA, SB, B, LDB, C, LDC, I, J, L) \
|
|
kusano |
2b45e8 |
FUSED_GEMM_KERNEL_T(M, N, K, ALPHA[0], SA, SB, \
|
|
kusano |
2b45e8 |
(FLOAT *)(B) + ((J) + (L) * LDB) * COMPSIZE, LDB, (FLOAT *)(C) + ((I) + (J) * LDC) * COMPSIZE, LDC)
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
#define FUSED_KERNEL_OPERATION(M, N, K, ALPHA, SA, SB, B, LDB, C, LDC, I, J, L) \
|
|
kusano |
2b45e8 |
FUSED_GEMM_KERNEL_T(M, N, K, ALPHA[0], ALPHA[1], SA, SB, \
|
|
kusano |
2b45e8 |
(FLOAT *)(B) + ((J) + (L) * LDB) * COMPSIZE, LDB, (FLOAT *)(C) + ((I) + (J) * LDC) * COMPSIZE, LDC)
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#ifndef A
|
|
kusano |
2b45e8 |
#define A args -> a
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#ifndef LDA
|
|
kusano |
2b45e8 |
#define LDA args -> lda
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#ifndef B
|
|
kusano |
2b45e8 |
#define B args -> b
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#ifndef LDB
|
|
kusano |
2b45e8 |
#define LDB args -> ldb
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#ifndef C
|
|
kusano |
2b45e8 |
#define C args -> c
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#ifndef LDC
|
|
kusano |
2b45e8 |
#define LDC args -> ldc
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#ifndef M
|
|
kusano |
2b45e8 |
#define M args -> m
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#ifndef N
|
|
kusano |
2b45e8 |
#define N args -> n
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#ifndef K
|
|
kusano |
2b45e8 |
#define K args -> k
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#ifdef TIMING
|
|
kusano |
2b45e8 |
#define START_RPCC() rpcc_counter = rpcc()
|
|
kusano |
2b45e8 |
#define STOP_RPCC(COUNTER) COUNTER += rpcc() - rpcc_counter
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
#define START_RPCC()
|
|
kusano |
2b45e8 |
#define STOP_RPCC(COUNTER)
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
int CNAME(blas_arg_t *args, BLASLONG *range_m, BLASLONG *range_n,
|
|
kusano |
2b45e8 |
XFLOAT *sa, XFLOAT *sb, BLASLONG dummy){
|
|
kusano |
2b45e8 |
BLASLONG k, lda, ldb, ldc;
|
|
kusano |
2b45e8 |
FLOAT *alpha, *beta;
|
|
kusano |
2b45e8 |
FLOAT *a, *b, *c;
|
|
kusano |
2b45e8 |
BLASLONG m_from, m_to, n_from, n_to;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
BLASLONG ls, is, js;
|
|
kusano |
2b45e8 |
BLASLONG min_l, min_i, min_j;
|
|
kusano |
2b45e8 |
#if !defined(FUSED_GEMM) || defined(TIMING)
|
|
kusano |
2b45e8 |
BLASLONG jjs, min_jj;
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
BLASLONG l1stride, gemm_p, l2size;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#if defined(XDOUBLE) && defined(QUAD_PRECISION)
|
|
kusano |
2b45e8 |
xidouble xalpha;
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#ifdef TIMING
|
|
kusano |
2b45e8 |
unsigned long long rpcc_counter;
|
|
kusano |
2b45e8 |
unsigned long long innercost = 0;
|
|
kusano |
2b45e8 |
unsigned long long outercost = 0;
|
|
kusano |
2b45e8 |
unsigned long long kernelcost = 0;
|
|
kusano |
2b45e8 |
double total;
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
k = K;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
a = (FLOAT *)A;
|
|
kusano |
2b45e8 |
b = (FLOAT *)B;
|
|
kusano |
2b45e8 |
c = (FLOAT *)C;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
lda = LDA;
|
|
kusano |
2b45e8 |
ldb = LDB;
|
|
kusano |
2b45e8 |
ldc = LDC;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
alpha = (FLOAT *)args -> alpha;
|
|
kusano |
2b45e8 |
beta = (FLOAT *)args -> beta;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
m_from = 0;
|
|
kusano |
2b45e8 |
m_to = M;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
if (range_m) {
|
|
kusano |
2b45e8 |
m_from = *(((BLASLONG *)range_m) + 0);
|
|
kusano |
2b45e8 |
m_to = *(((BLASLONG *)range_m) + 1);
|
|
kusano |
2b45e8 |
}
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
n_from = 0;
|
|
kusano |
2b45e8 |
n_to = N;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
if (range_n) {
|
|
kusano |
2b45e8 |
n_from = *(((BLASLONG *)range_n) + 0);
|
|
kusano |
2b45e8 |
n_to = *(((BLASLONG *)range_n) + 1);
|
|
kusano |
2b45e8 |
}
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
if (beta) {
|
|
kusano |
2b45e8 |
#if !defined(XDOUBLE) || !defined(QUAD_PRECISION)
|
|
kusano |
2b45e8 |
#ifndef COMPLEX
|
|
kusano |
2b45e8 |
if (beta[0] != ONE
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
if ((beta[0] != ONE) || (beta[1] != ZERO)
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
if (((beta[0].x[1] != 0x3fff000000000000UL) || beta[0].x[0] != 0)
|
|
kusano |
2b45e8 |
#ifdef COMPLEX
|
|
kusano |
2b45e8 |
&&(((beta[1].x[0] | beta[1].x[1]) << 1) != 0)
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
) {
|
|
kusano |
2b45e8 |
#if defined(XDOUBLE) && defined(QUAD_PRECISION)
|
|
kusano |
2b45e8 |
xidouble xbeta;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
qtox(&xbeta, beta);
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
BETA_OPERATION(m_from, m_to, n_from, n_to, beta, c, ldc);
|
|
kusano |
2b45e8 |
}
|
|
kusano |
2b45e8 |
}
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
if ((k == 0) || (alpha == NULL)) return 0;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#if !defined(XDOUBLE) || !defined(QUAD_PRECISION)
|
|
kusano |
2b45e8 |
if ((alpha[0] == ZERO)
|
|
kusano |
2b45e8 |
#ifdef COMPLEX
|
|
kusano |
2b45e8 |
&& (alpha[1] == ZERO)
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
) return 0;
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
if (((alpha[0].x[0] | alpha[0].x[1]
|
|
kusano |
2b45e8 |
#ifdef COMPLEX
|
|
kusano |
2b45e8 |
| alpha[1].x[0] | alpha[1].x[1]
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
) << 1) == 0) return 0;
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#if defined(XDOUBLE) && defined(QUAD_PRECISION)
|
|
kusano |
2b45e8 |
qtox(&xalpha, alpha);
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
l2size = GEMM_P * GEMM_Q;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#if 0
|
|
kusano |
2b45e8 |
fprintf(stderr, "GEMM(Single): M_from : %ld M_to : %ld N_from : %ld N_to : %ld k : %ld\n", m_from, m_to, n_from, n_to, k);
|
|
kusano |
2b45e8 |
fprintf(stderr, "GEMM(Single):: P = %4ld Q = %4ld R = %4ld\n", (BLASLONG)GEMM_P, (BLASLONG)GEMM_Q, (BLASLONG)GEMM_R);
|
|
kusano |
2b45e8 |
// fprintf(stderr, "GEMM: SA .. %p SB .. %p\n", sa, sb);
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
// fprintf(stderr, "A = %p B = %p C = %p\n\tlda = %ld ldb = %ld ldc = %ld\n", a, b, c, lda, ldb, ldc);
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#ifdef TIMING
|
|
kusano |
2b45e8 |
innercost = 0;
|
|
kusano |
2b45e8 |
outercost = 0;
|
|
kusano |
2b45e8 |
kernelcost = 0;
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
for(js = n_from; js < n_to; js += GEMM_R){
|
|
kusano |
2b45e8 |
min_j = n_to - js;
|
|
kusano |
2b45e8 |
if (min_j > GEMM_R) min_j = GEMM_R;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
for(ls = 0; ls < k; ls += min_l){
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
min_l = k - ls;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
if (min_l >= GEMM_Q * 2) {
|
|
kusano |
2b45e8 |
gemm_p = GEMM_P;
|
|
kusano |
2b45e8 |
min_l = GEMM_Q;
|
|
kusano |
2b45e8 |
} else {
|
|
kusano |
2b45e8 |
if (min_l > GEMM_Q) {
|
|
kusano |
2b45e8 |
min_l = (min_l / 2 + GEMM_UNROLL_M - 1) & ~(GEMM_UNROLL_M - 1);
|
|
kusano |
2b45e8 |
}
|
|
kusano |
2b45e8 |
gemm_p = ((l2size / min_l + GEMM_UNROLL_M - 1) & ~(GEMM_UNROLL_M - 1));
|
|
kusano |
2b45e8 |
while (gemm_p * min_l > l2size) gemm_p -= GEMM_UNROLL_M;
|
|
kusano |
2b45e8 |
}
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
/* First, we have to move data A to L2 cache */
|
|
kusano |
2b45e8 |
min_i = m_to - m_from;
|
|
kusano |
2b45e8 |
l1stride = 1;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
if (min_i >= GEMM_P * 2) {
|
|
kusano |
2b45e8 |
min_i = GEMM_P;
|
|
kusano |
2b45e8 |
} else {
|
|
kusano |
2b45e8 |
if (min_i > GEMM_P) {
|
|
kusano |
2b45e8 |
min_i = (min_i / 2 + GEMM_UNROLL_M - 1) & ~(GEMM_UNROLL_M - 1);
|
|
kusano |
2b45e8 |
} else {
|
|
kusano |
2b45e8 |
l1stride = 0;
|
|
kusano |
2b45e8 |
}
|
|
kusano |
2b45e8 |
}
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
START_RPCC();
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
ICOPY_OPERATION(min_l, min_i, a, lda, ls, m_from, sa);
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
STOP_RPCC(innercost);
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#if defined(FUSED_GEMM) && !defined(TIMING)
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
FUSED_KERNEL_OPERATION(min_i, min_j, min_l, alpha,
|
|
kusano |
2b45e8 |
sa, sb, b, ldb, c, ldc, m_from, js, ls);
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
for(jjs = js; jjs < js + min_j; jjs += min_jj){
|
|
kusano |
2b45e8 |
min_jj = min_j + js - jjs;
|
|
kusano |
2b45e8 |
if (min_jj > GEMM_UNROLL_N) min_jj = GEMM_UNROLL_N;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
START_RPCC();
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
OCOPY_OPERATION(min_l, min_jj, b, ldb, ls, jjs,
|
|
kusano |
2b45e8 |
sb + min_l * (jjs - js) * COMPSIZE * l1stride);
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
STOP_RPCC(outercost);
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
START_RPCC();
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#if !defined(XDOUBLE) || !defined(QUAD_PRECISION)
|
|
kusano |
2b45e8 |
KERNEL_OPERATION(min_i, min_jj, min_l, alpha,
|
|
kusano |
2b45e8 |
sa, sb + min_l * (jjs - js) * COMPSIZE * l1stride, c, ldc, m_from, jjs);
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
KERNEL_OPERATION(min_i, min_jj, min_l, (void *)&xalpha,
|
|
kusano |
2b45e8 |
sa, sb + min_l * (jjs - js) * COMPSIZE * l1stride, c, ldc, m_from, jjs);
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
STOP_RPCC(kernelcost);
|
|
kusano |
2b45e8 |
}
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
for(is = m_from + min_i; is < m_to; is += min_i){
|
|
kusano |
2b45e8 |
min_i = m_to - is;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
if (min_i >= GEMM_P * 2) {
|
|
kusano |
2b45e8 |
min_i = GEMM_P;
|
|
kusano |
2b45e8 |
} else
|
|
kusano |
2b45e8 |
if (min_i > GEMM_P) {
|
|
kusano |
2b45e8 |
min_i = (min_i / 2 + GEMM_UNROLL_M - 1) & ~(GEMM_UNROLL_M - 1);
|
|
kusano |
2b45e8 |
}
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
START_RPCC();
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
ICOPY_OPERATION(min_l, min_i, a, lda, ls, is, sa);
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
STOP_RPCC(innercost);
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
START_RPCC();
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#if !defined(XDOUBLE) || !defined(QUAD_PRECISION)
|
|
kusano |
2b45e8 |
KERNEL_OPERATION(min_i, min_j, min_l, alpha, sa, sb, c, ldc, is, js);
|
|
kusano |
2b45e8 |
#else
|
|
kusano |
2b45e8 |
KERNEL_OPERATION(min_i, min_j, min_l, (void *)&xalpha, sa, sb, c, ldc, is, js);
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
STOP_RPCC(kernelcost);
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
} /* end of is */
|
|
kusano |
2b45e8 |
} /* end of js */
|
|
kusano |
2b45e8 |
} /* end of ls */
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#ifdef TIMING
|
|
kusano |
2b45e8 |
total = (double)outercost + (double)innercost + (double)kernelcost;
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
printf( "Copy A : %5.2f Copy B: %5.2f Kernel : %5.2f kernel Effi. : %5.2f Total Effi. : %5.2f\n",
|
|
kusano |
2b45e8 |
innercost / total * 100., outercost / total * 100.,
|
|
kusano |
2b45e8 |
kernelcost / total * 100.,
|
|
kusano |
2b45e8 |
(double)(m_to - m_from) * (double)(n_to - n_from) * (double)k / (double)kernelcost * 100. * (double)COMPSIZE / 2.,
|
|
kusano |
2b45e8 |
(double)(m_to - m_from) * (double)(n_to - n_from) * (double)k / total * 100. * (double)COMPSIZE / 2.);
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
#endif
|
|
kusano |
2b45e8 |
|
|
kusano |
2b45e8 |
return 0;
|
|
kusano |
2b45e8 |
}
|