blob: e883072dfb7ea16e726bd2d9db11766b42df32f2 [file] [log] [blame]
/**************************************************************************************************
* *
* This file is part of BLASFEO. *
* *
* BLASFEO -- BLAS For Embedded Optimization. *
* Copyright (C) 2016-2017 by Gianluca Frison. *
* Developed at IMTEK (University of Freiburg) under the supervision of Moritz Diehl. *
* All rights reserved. *
* *
* HPMPC is free software; you can redistribute it and/or *
* modify it under the terms of the GNU Lesser General Public *
* License as published by the Free Software Foundation; either *
* version 2.1 of the License, or (at your option) any later version. *
* *
* HPMPC is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. *
* See the GNU Lesser General Public License for more details. *
* *
* You should have received a copy of the GNU Lesser General Public *
* License along with HPMPC; if not, write to the Free Software *
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA *
* *
* Author: Gianluca Frison, giaf (at) dtu.dk *
* gianluca.frison (at) imtek.uni-freiburg.de *
* *
**************************************************************************************************/
// both A and B are aligned to 256-bit boundaries
void kernel_dgecp_4_0_lib4(int tri, int kmax, double alpha, double *A, double *B)
{
if(tri==1)
{
// A and C are lower triangular
// kmax+1 4-wide + end 3x3 triangle
kmax += 1;
}
if(kmax<=0)
return;
const int bs = 4;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] = alpha*A[0+bs*0];
B[1+bs*0] = alpha*A[1+bs*0];
B[2+bs*0] = alpha*A[2+bs*0];
B[3+bs*0] = alpha*A[3+bs*0];
B[0+bs*1] = alpha*A[0+bs*1];
B[1+bs*1] = alpha*A[1+bs*1];
B[2+bs*1] = alpha*A[2+bs*1];
B[3+bs*1] = alpha*A[3+bs*1];
B[0+bs*2] = alpha*A[0+bs*2];
B[1+bs*2] = alpha*A[1+bs*2];
B[2+bs*2] = alpha*A[2+bs*2];
B[3+bs*2] = alpha*A[3+bs*2];
B[0+bs*3] = alpha*A[0+bs*3];
B[1+bs*3] = alpha*A[1+bs*3];
B[2+bs*3] = alpha*A[2+bs*3];
B[3+bs*3] = alpha*A[3+bs*3];
A += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] = alpha*A[0+bs*0];
B[1+bs*0] = alpha*A[1+bs*0];
B[2+bs*0] = alpha*A[2+bs*0];
B[3+bs*0] = alpha*A[3+bs*0];
A += 4;
B += 4;
}
if(tri==1)
{
// 3x3 triangle
B[1+bs*0] = alpha*A[1+bs*0];
B[2+bs*0] = alpha*A[2+bs*0];
B[3+bs*0] = alpha*A[3+bs*0];
B[2+bs*1] = alpha*A[2+bs*1];
B[3+bs*1] = alpha*A[3+bs*1];
B[3+bs*2] = alpha*A[3+bs*2];
}
}
// both A and B are aligned to 256-bit boundaries, 1 element of A must be skipped
void kernel_dgecp_4_1_lib4(int tri, int kmax, double alpha, double *A0, int sda, double *B)
{
if(tri==1)
{
// A and C are lower triangular
// kmax+1 4-wide + end 3x3 triangle
kmax += 1;
}
if(kmax<=0)
return;
const int bs = 4;
double *A1 = A0 + bs*sda;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] = alpha*A0[1+bs*0];
B[1+bs*0] = alpha*A0[2+bs*0];
B[2+bs*0] = alpha*A0[3+bs*0];
B[3+bs*0] = alpha*A1[0+bs*0];
B[0+bs*1] = alpha*A0[1+bs*1];
B[1+bs*1] = alpha*A0[2+bs*1];
B[2+bs*1] = alpha*A0[3+bs*1];
B[3+bs*1] = alpha*A1[0+bs*1];
B[0+bs*2] = alpha*A0[1+bs*2];
B[1+bs*2] = alpha*A0[2+bs*2];
B[2+bs*2] = alpha*A0[3+bs*2];
B[3+bs*2] = alpha*A1[0+bs*2];
B[0+bs*3] = alpha*A0[1+bs*3];
B[1+bs*3] = alpha*A0[2+bs*3];
B[2+bs*3] = alpha*A0[3+bs*3];
B[3+bs*3] = alpha*A1[0+bs*3];
A0 += 16;
A1 += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] = alpha*A0[1+bs*0];
B[1+bs*0] = alpha*A0[2+bs*0];
B[2+bs*0] = alpha*A0[3+bs*0];
B[3+bs*0] = alpha*A1[0+bs*0];
A0 += 4;
A1 += 4;
B += 4;
}
if(tri==1)
{
// 3x3 triangle
B[1+0*bs] = alpha*A0[2+0*bs];
B[2+0*bs] = alpha*A0[3+0*bs];
B[3+0*bs] = alpha*A1[0+0*bs];
B[2+1*bs] = alpha*A0[3+1*bs];
B[3+1*bs] = alpha*A1[0+1*bs];
B[3+2*bs] = alpha*A1[0+2*bs];
}
}
// both A and B are aligned to 256-bit boundaries, 2 elements of A must be skipped
void kernel_dgecp_4_2_lib4(int tri, int kmax, double alpha, double *A0, int sda, double *B)
{
if(tri==1)
{
// A and C are lower triangular
// kmax+1 4-wide + end 3x3 triangle
kmax += 1;
}
if(kmax<=0)
return;
const int bs = 4;
double *A1 = A0 + bs*sda;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] = alpha*A0[2+bs*0];
B[1+bs*0] = alpha*A0[3+bs*0];
B[2+bs*0] = alpha*A1[0+bs*0];
B[3+bs*0] = alpha*A1[1+bs*0];
B[0+bs*1] = alpha*A0[2+bs*1];
B[1+bs*1] = alpha*A0[3+bs*1];
B[2+bs*1] = alpha*A1[0+bs*1];
B[3+bs*1] = alpha*A1[1+bs*1];
B[0+bs*2] = alpha*A0[2+bs*2];
B[1+bs*2] = alpha*A0[3+bs*2];
B[2+bs*2] = alpha*A1[0+bs*2];
B[3+bs*2] = alpha*A1[1+bs*2];
B[0+bs*3] = alpha*A0[2+bs*3];
B[1+bs*3] = alpha*A0[3+bs*3];
B[2+bs*3] = alpha*A1[0+bs*3];
B[3+bs*3] = alpha*A1[1+bs*3];
A0 += 16;
A1 += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] = alpha*A0[2+bs*0];
B[1+bs*0] = alpha*A0[3+bs*0];
B[2+bs*0] = alpha*A1[0+bs*0];
B[3+bs*0] = alpha*A1[1+bs*0];
A0 += 4;
A1 += 4;
B += 4;
}
if(tri==1)
{
// 3x3 triangle}
B[1+bs*0] = alpha*A0[3+bs*0];
B[2+bs*0] = alpha*A1[0+bs*0];
B[3+bs*0] = alpha*A1[1+bs*0];
B[2+bs*1] = alpha*A1[0+bs*1];
B[3+bs*1] = alpha*A1[1+bs*1];
B[3+bs*2] = alpha*A1[1+bs*2];
}
}
// both A and B are aligned to 256-bit boundaries, 3 elements of A must be skipped
void kernel_dgecp_4_3_lib4(int tri, int kmax, double alpha, double *A0, int sda, double *B)
{
if(tri==1)
{
// A and C are lower triangular
// kmax+1 4-wide + end 3x3 triangle
kmax += 1;
}
if(kmax<=0)
return;
const int bs = 4;
double *A1 = A0 + bs*sda;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] = alpha*A0[3+bs*0];
B[1+bs*0] = alpha*A1[0+bs*0];
B[2+bs*0] = alpha*A1[1+bs*0];
B[3+bs*0] = alpha*A1[2+bs*0];
B[0+bs*1] = alpha*A0[3+bs*1];
B[1+bs*1] = alpha*A1[0+bs*1];
B[2+bs*1] = alpha*A1[1+bs*1];
B[3+bs*1] = alpha*A1[2+bs*1];
B[0+bs*2] = alpha*A0[3+bs*2];
B[1+bs*2] = alpha*A1[0+bs*2];
B[2+bs*2] = alpha*A1[1+bs*2];
B[3+bs*2] = alpha*A1[2+bs*2];
B[0+bs*3] = alpha*A0[3+bs*3];
B[1+bs*3] = alpha*A1[0+bs*3];
B[2+bs*3] = alpha*A1[1+bs*3];
B[3+bs*3] = alpha*A1[2+bs*3];
A0 += 16;
A1 += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] = alpha*A0[3+bs*0];
B[1+bs*0] = alpha*A1[0+bs*0];
B[2+bs*0] = alpha*A1[1+bs*0];
B[3+bs*0] = alpha*A1[2+bs*0];
A0 += 4;
A1 += 4;
B += 4;
}
if(tri==1)
{
// 3x3 triangle
B[1+bs*0] = alpha*A1[0+bs*0];
B[2+bs*0] = alpha*A1[1+bs*0];
B[3+bs*0] = alpha*A1[2+bs*0];
B[2+bs*1] = alpha*A1[1+bs*1];
B[3+bs*1] = alpha*A1[2+bs*1];
B[3+bs*2] = alpha*A1[2+bs*2];
}
}
// both A and B are aligned to 64-bit boundaries
void kernel_dgecp_3_0_lib4(int tri, int kmax, double alpha, double *A, double *B)
{
if(tri==1)
{
// A and C are lower triangular
// kmax+1 3-wide + end 2x2 triangle
kmax += 1;
}
if(kmax<=0)
return;
const int bs = 4;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] = alpha*A[0+bs*0];
B[1+bs*0] = alpha*A[1+bs*0];
B[2+bs*0] = alpha*A[2+bs*0];
B[0+bs*1] = alpha*A[0+bs*1];
B[1+bs*1] = alpha*A[1+bs*1];
B[2+bs*1] = alpha*A[2+bs*1];
B[0+bs*2] = alpha*A[0+bs*2];
B[1+bs*2] = alpha*A[1+bs*2];
B[2+bs*2] = alpha*A[2+bs*2];
B[0+bs*3] = alpha*A[0+bs*3];
B[1+bs*3] = alpha*A[1+bs*3];
B[2+bs*3] = alpha*A[2+bs*3];
A += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] = alpha*A[0+bs*0];
B[1+bs*0] = alpha*A[1+bs*0];
B[2+bs*0] = alpha*A[2+bs*0];
A += 4;
B += 4;
}
if(tri==1)
{
// 2x2 triangle
B[1+bs*0] = alpha*A[1+bs*0];
B[2+bs*0] = alpha*A[2+bs*0];
B[2+bs*1] = alpha*A[2+bs*1];
}
}
// both A and B are aligned to 256-bit boundaries, 2 elements of A must be skipped
void kernel_dgecp_3_2_lib4(int tri, int kmax, double alpha, double *A0, int sda, double *B)
{
if(tri==1)
{
// A and C are lower triangular
// kmax+1 3-wide + end 2x2 triangle
kmax += 1;
}
if(kmax<=0)
return;
const int bs = 4;
double *A1 = A0 + bs*sda;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] = alpha*A0[2+bs*0];
B[1+bs*0] = alpha*A0[3+bs*0];
B[2+bs*0] = alpha*A1[0+bs*0];
B[0+bs*1] = alpha*A0[2+bs*1];
B[1+bs*1] = alpha*A0[3+bs*1];
B[2+bs*1] = alpha*A1[0+bs*1];
B[0+bs*2] = alpha*A0[2+bs*2];
B[1+bs*2] = alpha*A0[3+bs*2];
B[2+bs*2] = alpha*A1[0+bs*2];
B[0+bs*3] = alpha*A0[2+bs*3];
B[1+bs*3] = alpha*A0[3+bs*3];
B[2+bs*3] = alpha*A1[0+bs*3];
A0 += 16;
A1 += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] = alpha*A0[2+bs*0];
B[1+bs*0] = alpha*A0[3+bs*0];
B[2+bs*0] = alpha*A1[0+bs*0];
A0 += 4;
A1 += 4;
B += 4;
}
if(tri==1)
{
// 2x2 triangle
B[1+bs*0] = alpha*A0[3+bs*0];
B[2+bs*0] = alpha*A1[0+bs*0];
B[2+bs*1] = alpha*A1[0+bs*1];
}
}
// both A and B are aligned to 256-bit boundaries, 3 elements of A must be skipped
void kernel_dgecp_3_3_lib4(int tri, int kmax, double alpha, double *A0, int sda, double *B)
{
if(tri==1)
{
// A and C are lower triangular
// kmax+1 3-wide + end 2x2 triangle
kmax += 1;
}
if(kmax<=0)
return;
const int bs = 4;
double *A1 = A0 + bs*sda;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] = alpha*A0[3+bs*0];
B[1+bs*0] = alpha*A1[0+bs*0];
B[2+bs*0] = alpha*A1[1+bs*0];
B[0+bs*1] = alpha*A0[3+bs*1];
B[1+bs*1] = alpha*A1[0+bs*1];
B[2+bs*1] = alpha*A1[1+bs*1];
B[0+bs*2] = alpha*A0[3+bs*2];
B[1+bs*2] = alpha*A1[0+bs*2];
B[2+bs*2] = alpha*A1[1+bs*2];
B[0+bs*3] = alpha*A0[3+bs*3];
B[1+bs*3] = alpha*A1[0+bs*3];
B[2+bs*3] = alpha*A1[1+bs*3];
A0 += 16;
A1 += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] = alpha*A0[3+bs*0];
B[1+bs*0] = alpha*A1[0+bs*0];
B[2+bs*0] = alpha*A1[1+bs*0];
A0 += 4;
A1 += 4;
B += 4;
}
if(tri==1)
{
// 2x2 triangle
B[1+bs*0] = alpha*A1[0+bs*0];
B[2+bs*0] = alpha*A1[1+bs*0];
B[2+bs*1] = alpha*A1[1+bs*1];
}
}
// both A and B are aligned to 64-bit boundaries
void kernel_dgecp_2_0_lib4(int tri, int kmax, double alpha, double *A, double *B)
{
if(tri==1)
{
// A and C are lower triangular
// kmax+1 2-wide + end 1x1 triangle
kmax += 1;
}
if(kmax<=0)
return;
const int bs = 4;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] = alpha*A[0+bs*0];
B[1+bs*0] = alpha*A[1+bs*0];
B[0+bs*1] = alpha*A[0+bs*1];
B[1+bs*1] = alpha*A[1+bs*1];
B[0+bs*2] = alpha*A[0+bs*2];
B[1+bs*2] = alpha*A[1+bs*2];
B[0+bs*3] = alpha*A[0+bs*3];
B[1+bs*3] = alpha*A[1+bs*3];
A += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] = alpha*A[0+bs*0];
B[1+bs*0] = alpha*A[1+bs*0];
A += 4;
B += 4;
}
if(tri==1)
{
// 1x1 triangle
B[1+bs*0] = alpha*A[1+bs*0];
}
}
// both A and B are aligned to 128-bit boundaries, 3 elements of A must be skipped
void kernel_dgecp_2_3_lib4(int tri, int kmax, double alpha, double *A0, int sda, double *B)
{
if(tri==1)
{
// A and C are lower triangular
// kmax+1 2-wide + end 1x1 triangle
kmax += 1;
}
if(kmax<=0)
return;
const int bs = 4;
double *A1 = A0 + bs*sda;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] = alpha*A0[3+bs*0];
B[1+bs*0] = alpha*A1[0+bs*0];
B[0+bs*1] = alpha*A0[3+bs*1];
B[1+bs*1] = alpha*A1[0+bs*1];
B[0+bs*2] = alpha*A0[3+bs*2];
B[1+bs*2] = alpha*A1[0+bs*2];
B[0+bs*3] = alpha*A0[3+bs*3];
B[1+bs*3] = alpha*A1[0+bs*3];
A0 += 16;
A1 += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] = alpha*A0[3+bs*0];
B[1+bs*0] = alpha*A1[0+bs*0];
A0 += 4;
A1 += 4;
B += 4;
}
if(tri==1)
{
// 1x1 triangle
B[1+bs*0] = alpha*A1[0+bs*0];
}
}
// both A and B are aligned 64-bit boundaries
void kernel_dgecp_1_0_lib4(int tri, int kmax, double alpha, double *A, double *B)
{
if(tri==1)
{
// A and C are lower triangular
// kmax+1 1-wide
kmax += 1;
}
if(kmax<=0)
return;
const int bs = 4;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] = alpha*A[0+bs*0];
B[0+bs*1] = alpha*A[0+bs*1];
B[0+bs*2] = alpha*A[0+bs*2];
B[0+bs*3] = alpha*A[0+bs*3];
A += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] = alpha*A[0+bs*0];
A += 4;
B += 4;
}
}
// both A and B are aligned to 256-bit boundaries
void kernel_dgead_4_0_lib4(int kmax, double alpha, double *A, double *B)
{
if(kmax<=0)
return;
const int bs = 4;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] += alpha * A[0+bs*0];
B[1+bs*0] += alpha * A[1+bs*0];
B[2+bs*0] += alpha * A[2+bs*0];
B[3+bs*0] += alpha * A[3+bs*0];
B[0+bs*1] += alpha * A[0+bs*1];
B[1+bs*1] += alpha * A[1+bs*1];
B[2+bs*1] += alpha * A[2+bs*1];
B[3+bs*1] += alpha * A[3+bs*1];
B[0+bs*2] += alpha * A[0+bs*2];
B[1+bs*2] += alpha * A[1+bs*2];
B[2+bs*2] += alpha * A[2+bs*2];
B[3+bs*2] += alpha * A[3+bs*2];
B[0+bs*3] += alpha * A[0+bs*3];
B[1+bs*3] += alpha * A[1+bs*3];
B[2+bs*3] += alpha * A[2+bs*3];
B[3+bs*3] += alpha * A[3+bs*3];
A += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] += alpha * A[0+bs*0];
B[1+bs*0] += alpha * A[1+bs*0];
B[2+bs*0] += alpha * A[2+bs*0];
B[3+bs*0] += alpha * A[3+bs*0];
A += 4;
B += 4;
}
}
// both A and B are aligned to 256-bit boundaries, 1 element of A must be skipped
void kernel_dgead_4_1_lib4(int kmax, double alpha, double *A0, int sda, double *B)
{
if(kmax<=0)
return;
const int bs = 4;
double *A1 = A0 + bs*sda;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] += alpha * A0[1+bs*0];
B[1+bs*0] += alpha * A0[2+bs*0];
B[2+bs*0] += alpha * A0[3+bs*0];
B[3+bs*0] += alpha * A1[0+bs*0];
B[0+bs*1] += alpha * A0[1+bs*1];
B[1+bs*1] += alpha * A0[2+bs*1];
B[2+bs*1] += alpha * A0[3+bs*1];
B[3+bs*1] += alpha * A1[0+bs*1];
B[0+bs*2] += alpha * A0[1+bs*2];
B[1+bs*2] += alpha * A0[2+bs*2];
B[2+bs*2] += alpha * A0[3+bs*2];
B[3+bs*2] += alpha * A1[0+bs*2];
B[0+bs*3] += alpha * A0[1+bs*3];
B[1+bs*3] += alpha * A0[2+bs*3];
B[2+bs*3] += alpha * A0[3+bs*3];
B[3+bs*3] += alpha * A1[0+bs*3];
A0 += 16;
A1 += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] += alpha * A0[1+bs*0];
B[1+bs*0] += alpha * A0[2+bs*0];
B[2+bs*0] += alpha * A0[3+bs*0];
B[3+bs*0] += alpha * A1[0+bs*0];
A0 += 4;
A1 += 4;
B += 4;
}
}
// both A and B are aligned to 256-bit boundaries, 2 elements of A must be skipped
void kernel_dgead_4_2_lib4(int kmax, double alpha, double *A0, int sda, double *B)
{
if(kmax<=0)
return;
const int bs = 4;
double *A1 = A0 + bs*sda;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] += alpha * A0[2+bs*0];
B[1+bs*0] += alpha * A0[3+bs*0];
B[2+bs*0] += alpha * A1[0+bs*0];
B[3+bs*0] += alpha * A1[1+bs*0];
B[0+bs*1] += alpha * A0[2+bs*1];
B[1+bs*1] += alpha * A0[3+bs*1];
B[2+bs*1] += alpha * A1[0+bs*1];
B[3+bs*1] += alpha * A1[1+bs*1];
B[0+bs*2] += alpha * A0[2+bs*2];
B[1+bs*2] += alpha * A0[3+bs*2];
B[2+bs*2] += alpha * A1[0+bs*2];
B[3+bs*2] += alpha * A1[1+bs*2];
B[0+bs*3] += alpha * A0[2+bs*3];
B[1+bs*3] += alpha * A0[3+bs*3];
B[2+bs*3] += alpha * A1[0+bs*3];
B[3+bs*3] += alpha * A1[1+bs*3];
A0 += 16;
A1 += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] += alpha * A0[2+bs*0];
B[1+bs*0] += alpha * A0[3+bs*0];
B[2+bs*0] += alpha * A1[0+bs*0];
B[3+bs*0] += alpha * A1[1+bs*0];
A0 += 4;
A1 += 4;
B += 4;
}
}
// both A and B are aligned to 256-bit boundaries, 3 elements of A must be skipped
void kernel_dgead_4_3_lib4(int kmax, double alpha, double *A0, int sda, double *B)
{
if(kmax<=0)
return;
const int bs = 4;
double *A1 = A0 + bs*sda;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] += alpha * A0[3+bs*0];
B[1+bs*0] += alpha * A1[0+bs*0];
B[2+bs*0] += alpha * A1[1+bs*0];
B[3+bs*0] += alpha * A1[2+bs*0];
B[0+bs*1] += alpha * A0[3+bs*1];
B[1+bs*1] += alpha * A1[0+bs*1];
B[2+bs*1] += alpha * A1[1+bs*1];
B[3+bs*1] += alpha * A1[2+bs*1];
B[0+bs*2] += alpha * A0[3+bs*2];
B[1+bs*2] += alpha * A1[0+bs*2];
B[2+bs*2] += alpha * A1[1+bs*2];
B[3+bs*2] += alpha * A1[2+bs*2];
B[0+bs*3] += alpha * A0[3+bs*3];
B[1+bs*3] += alpha * A1[0+bs*3];
B[2+bs*3] += alpha * A1[1+bs*3];
B[3+bs*3] += alpha * A1[2+bs*3];
A0 += 16;
A1 += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] += alpha * A0[3+bs*0];
B[1+bs*0] += alpha * A1[0+bs*0];
B[2+bs*0] += alpha * A1[1+bs*0];
B[3+bs*0] += alpha * A1[2+bs*0];
A0 += 4;
A1 += 4;
B += 4;
}
}
// both A and B are aligned to 64-bit boundaries
void kernel_dgead_3_0_lib4(int kmax, double alpha, double *A, double *B)
{
if(kmax<=0)
return;
const int bs = 4;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] += alpha * A[0+bs*0];
B[1+bs*0] += alpha * A[1+bs*0];
B[2+bs*0] += alpha * A[2+bs*0];
B[0+bs*1] += alpha * A[0+bs*1];
B[1+bs*1] += alpha * A[1+bs*1];
B[2+bs*1] += alpha * A[2+bs*1];
B[0+bs*2] += alpha * A[0+bs*2];
B[1+bs*2] += alpha * A[1+bs*2];
B[2+bs*2] += alpha * A[2+bs*2];
B[0+bs*3] += alpha * A[0+bs*3];
B[1+bs*3] += alpha * A[1+bs*3];
B[2+bs*3] += alpha * A[2+bs*3];
A += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] += alpha * A[0+bs*0];
B[1+bs*0] += alpha * A[1+bs*0];
B[2+bs*0] += alpha * A[2+bs*0];
A += 4;
B += 4;
}
}
// both A and B are aligned to 256-bit boundaries, 2 elements of A must be skipped
void kernel_dgead_3_2_lib4(int kmax, double alpha, double *A0, int sda, double *B)
{
if(kmax<=0)
return;
const int bs = 4;
double *A1 = A0 + bs*sda;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] += alpha * A0[2+bs*0];
B[1+bs*0] += alpha * A0[3+bs*0];
B[2+bs*0] += alpha * A1[0+bs*0];
B[0+bs*1] += alpha * A0[2+bs*1];
B[1+bs*1] += alpha * A0[3+bs*1];
B[2+bs*1] += alpha * A1[0+bs*1];
B[0+bs*2] += alpha * A0[2+bs*2];
B[1+bs*2] += alpha * A0[3+bs*2];
B[2+bs*2] += alpha * A1[0+bs*2];
B[0+bs*3] += alpha * A0[2+bs*3];
B[1+bs*3] += alpha * A0[3+bs*3];
B[2+bs*3] += alpha * A1[0+bs*3];
A0 += 16;
A1 += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] += alpha * A0[2+bs*0];
B[1+bs*0] += alpha * A0[3+bs*0];
B[2+bs*0] += alpha * A1[0+bs*0];
A0 += 4;
A1 += 4;
B += 4;
}
}
// both A and B are aligned to 256-bit boundaries, 3 elements of A must be skipped
void kernel_dgead_3_3_lib4(int kmax, double alpha, double *A0, int sda, double *B)
{
if(kmax<=0)
return;
const int bs = 4;
double *A1 = A0 + bs*sda;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] += alpha * A0[3+bs*0];
B[1+bs*0] += alpha * A1[0+bs*0];
B[2+bs*0] += alpha * A1[1+bs*0];
B[0+bs*1] += alpha * A0[3+bs*1];
B[1+bs*1] += alpha * A1[0+bs*1];
B[2+bs*1] += alpha * A1[1+bs*1];
B[0+bs*2] += alpha * A0[3+bs*2];
B[1+bs*2] += alpha * A1[0+bs*2];
B[2+bs*2] += alpha * A1[1+bs*2];
B[0+bs*3] += alpha * A0[3+bs*3];
B[1+bs*3] += alpha * A1[0+bs*3];
B[2+bs*3] += alpha * A1[1+bs*3];
A0 += 16;
A1 += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] += alpha * A0[3+bs*0];
B[1+bs*0] += alpha * A1[0+bs*0];
B[2+bs*0] += alpha * A1[1+bs*0];
A0 += 4;
A1 += 4;
B += 4;
}
}
// both A and B are aligned to 64-bit boundaries
void kernel_dgead_2_0_lib4(int kmax, double alpha, double *A, double *B)
{
if(kmax<=0)
return;
const int bs = 4;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] += alpha * A[0+bs*0];
B[1+bs*0] += alpha * A[1+bs*0];
B[0+bs*1] += alpha * A[0+bs*1];
B[1+bs*1] += alpha * A[1+bs*1];
B[0+bs*2] += alpha * A[0+bs*2];
B[1+bs*2] += alpha * A[1+bs*2];
B[0+bs*3] += alpha * A[0+bs*3];
B[1+bs*3] += alpha * A[1+bs*3];
A += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] += alpha * A[0+bs*0];
B[1+bs*0] += alpha * A[1+bs*0];
A += 4;
B += 4;
}
}
// both A and B are aligned to 128-bit boundaries, 3 elements of A must be skipped
void kernel_dgead_2_3_lib4(int kmax, double alpha, double *A0, int sda, double *B)
{
if(kmax<=0)
return;
const int bs = 4;
double *A1 = A0 + bs*sda;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] += alpha * A0[3+bs*0];
B[1+bs*0] += alpha * A1[0+bs*0];
B[0+bs*1] += alpha * A0[3+bs*1];
B[1+bs*1] += alpha * A1[0+bs*1];
B[0+bs*2] += alpha * A0[3+bs*2];
B[1+bs*2] += alpha * A1[0+bs*2];
B[0+bs*3] += alpha * A0[3+bs*3];
B[1+bs*3] += alpha * A1[0+bs*3];
A0 += 16;
A1 += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] += alpha * A0[3+bs*0];
B[1+bs*0] += alpha * A1[0+bs*0];
A0 += 4;
A1 += 4;
B += 4;
}
}
// both A and B are aligned 64-bit boundaries
void kernel_dgead_1_0_lib4(int kmax, double alpha, double *A, double *B)
{
if(kmax<=0)
return;
const int bs = 4;
int k;
for(k=0; k<kmax-3; k+=4)
{
B[0+bs*0] += alpha * A[0+bs*0];
B[0+bs*1] += alpha * A[0+bs*1];
B[0+bs*2] += alpha * A[0+bs*2];
B[0+bs*3] += alpha * A[0+bs*3];
A += 16;
B += 16;
}
for(; k<kmax; k++)
{
B[0+bs*0] += alpha * A[0+bs*0];
A += 4;
B += 4;
}
}