273 lines
9.0 KiB
C
273 lines
9.0 KiB
C
/*****************************************************************************
|
|
Copyright (c) 2023, 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 COPYRIGHT OWNER 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.
|
|
|
|
**********************************************************************************/
|
|
|
|
#include "utest/openblas_utest.h"
|
|
#include <cblas.h>
|
|
#include "common.h"
|
|
|
|
#define DATASIZE 100
|
|
#define INCREMENT 2
|
|
|
|
struct DATA_ZGEMM {
|
|
double a_test[DATASIZE * DATASIZE * 2];
|
|
double a_verify[DATASIZE * DATASIZE * 2];
|
|
double b_test[DATASIZE * DATASIZE * 2];
|
|
double b_verify[DATASIZE * DATASIZE * 2];
|
|
double c_test[DATASIZE * DATASIZE * 2];
|
|
double c_verify[DATASIZE * DATASIZE * 2];
|
|
};
|
|
|
|
#ifdef BUILD_COMPLEX16
|
|
static struct DATA_ZGEMM data_zgemm;
|
|
|
|
/**
|
|
* Test zgemm with the conjugate matrices by conjugating and not transposed matrices
|
|
* and comparing it with the non-conjugate zgemm.
|
|
*
|
|
* param transa specifies op(A), the transposition (conjugation) operation applied to A
|
|
* param transb specifies op(B), the transposition (conjugation) operation applied to B
|
|
* param m specifies the number of rows of the matrix op(A) and of the matrix C
|
|
* param n specifies the number of columns of the matrix op(B) and the number of columns of the matrix C
|
|
* param k specifies the number of columns of the matrix op(A) and the number of rows of the matrix op(B)
|
|
* param alpha - scaling factor for the matrix-matrix product
|
|
* param lda - leading dimension of matrix A
|
|
* param ldb - leading dimension of matrix B
|
|
* param beta - scaling factor for matrix C
|
|
* param ldc - leading dimension of matrix C
|
|
* return norm of difference
|
|
*/
|
|
static double check_zgemm(char transa, char transb, blasint m, blasint n, blasint k,
|
|
double *alpha, blasint lda, blasint ldb, double *beta, blasint ldc)
|
|
{
|
|
blasint i;
|
|
double alpha_conj[] = {1.0, 0.0};
|
|
char transa_verify = transa;
|
|
char transb_verify = transb;
|
|
|
|
int arows = k, acols = m;
|
|
int brows = n, bcols = k;
|
|
|
|
if (transa == 'T' || transa == 'C'){
|
|
arows = m; acols = k;
|
|
}
|
|
|
|
if (transb == 'T' || transb == 'C'){
|
|
brows = k; bcols = n;
|
|
}
|
|
|
|
drand_generate(data_zgemm.a_test, arows * lda * 2);
|
|
drand_generate(data_zgemm.b_test, brows * ldb * 2);
|
|
drand_generate(data_zgemm.c_test, n * ldc * 2);
|
|
|
|
for (i = 0; i < arows * lda * 2; i++)
|
|
data_zgemm.a_verify[i] = data_zgemm.a_test[i];
|
|
|
|
for (i = 0; i < brows * ldb * 2; i++)
|
|
data_zgemm.b_verify[i] = data_zgemm.b_test[i];
|
|
|
|
for (i = 0; i < n * ldc * 2; i++)
|
|
data_zgemm.c_verify[i] = data_zgemm.c_test[i];
|
|
|
|
if (transa == 'R'){
|
|
cblas_zimatcopy(CblasColMajor, CblasConjNoTrans, arows, acols, alpha_conj, data_zgemm.a_verify, lda, lda);
|
|
transa_verify = 'N';
|
|
}
|
|
|
|
if (transb == 'R'){
|
|
cblas_zimatcopy(CblasColMajor, CblasConjNoTrans, brows, bcols, alpha_conj, data_zgemm.b_verify, ldb, ldb);
|
|
transb_verify = 'N';
|
|
}
|
|
|
|
BLASFUNC(zgemm)(&transa_verify, &transb_verify, &m, &n, &k, alpha, data_zgemm.a_verify, &lda,
|
|
data_zgemm.b_verify, &ldb, beta, data_zgemm.c_verify, &ldc);
|
|
|
|
BLASFUNC(zgemm)(&transa, &transb, &m, &n, &k, alpha, data_zgemm.a_test, &lda,
|
|
data_zgemm.b_test, &ldb, beta, data_zgemm.c_test, &ldc);
|
|
|
|
return dmatrix_difference(data_zgemm.c_test, data_zgemm.c_verify, m, n, ldc*2);
|
|
}
|
|
|
|
/**
|
|
* Test zgemm with the conjugate matrices by conjugating and not transposed matrices
|
|
* and comparing it with the non-conjugate zgemm.
|
|
* Test with the following options:
|
|
*
|
|
* matrix A is conjugate and transposed
|
|
* matrix B is conjugate and not transposed
|
|
*/
|
|
CTEST(zgemm, conjtransa_conjnotransb)
|
|
{
|
|
blasint n = DATASIZE, m = DATASIZE, k = DATASIZE;
|
|
blasint lda = DATASIZE, ldb = DATASIZE, ldc = DATASIZE;
|
|
char transa = 'C';
|
|
char transb = 'R';
|
|
double alpha[] = {-2.0, 1.0};
|
|
double beta[] = {1.0, -1.0};
|
|
|
|
double norm = check_zgemm(transa, transb, m, n, k, alpha, lda, ldb, beta, ldc);
|
|
|
|
ASSERT_DBL_NEAR_TOL(0.0, norm, DOUBLE_EPS);
|
|
}
|
|
|
|
/**
|
|
* Test zgemm with the conjugate matrices by conjugating and not transposed matrices
|
|
* and comparing it with the non-conjugate zgemm.
|
|
* Test with the following options:
|
|
*
|
|
* matrix A is not conjugate and not transposed
|
|
* matrix B is conjugate and not transposed
|
|
*/
|
|
CTEST(zgemm, notransa_conjnotransb)
|
|
{
|
|
blasint n = DATASIZE, m = DATASIZE, k = DATASIZE;
|
|
blasint lda = DATASIZE, ldb = DATASIZE, ldc = DATASIZE;
|
|
char transa = 'N';
|
|
char transb = 'R';
|
|
double alpha[] = {-2.0, 1.0};
|
|
double beta[] = {1.0, -1.0};
|
|
|
|
double norm = check_zgemm(transa, transb, m, n, k, alpha, lda, ldb, beta, ldc);
|
|
|
|
ASSERT_DBL_NEAR_TOL(0.0, norm, DOUBLE_EPS);
|
|
}
|
|
|
|
/**
|
|
* Test zgemm with the conjugate matrices by conjugating and not transposed matrices
|
|
* and comparing it with the non-conjugate zgemm.
|
|
* Test with the following options:
|
|
*
|
|
* matrix A is conjugate and not transposed
|
|
* matrix B is conjugate and transposed
|
|
*/
|
|
CTEST(zgemm, conjnotransa_conjtransb)
|
|
{
|
|
blasint n = DATASIZE, m = DATASIZE, k = DATASIZE;
|
|
blasint lda = DATASIZE, ldb = DATASIZE, ldc = DATASIZE;
|
|
char transa = 'R';
|
|
char transb = 'C';
|
|
double alpha[] = {-2.0, 1.0};
|
|
double beta[] = {1.0, -1.0};
|
|
|
|
double norm = check_zgemm(transa, transb, m, n, k, alpha, lda, ldb, beta, ldc);
|
|
|
|
ASSERT_DBL_NEAR_TOL(0.0, norm, DOUBLE_EPS);
|
|
}
|
|
|
|
/**
|
|
* Test zgemm with the conjugate matrices by conjugating and not transposed matrices
|
|
* and comparing it with the non-conjugate zgemm.
|
|
* Test with the following options:
|
|
*
|
|
* matrix A is conjugate and not transposed
|
|
* matrix B is not conjugate and not transposed
|
|
*/
|
|
CTEST(zgemm, conjnotransa_notransb)
|
|
{
|
|
blasint n = DATASIZE, m = DATASIZE, k = DATASIZE;
|
|
blasint lda = DATASIZE, ldb = DATASIZE, ldc = DATASIZE;
|
|
char transa = 'R';
|
|
char transb = 'N';
|
|
double alpha[] = {-2.0, 1.0};
|
|
double beta[] = {1.0, -1.0};
|
|
|
|
double norm = check_zgemm(transa, transb, m, n, k, alpha, lda, ldb, beta, ldc);
|
|
|
|
ASSERT_DBL_NEAR_TOL(0.0, norm, DOUBLE_EPS);
|
|
}
|
|
|
|
/**
|
|
* Test zgemm with the conjugate matrices by conjugating and not transposed matrices
|
|
* and comparing it with the non-conjugate zgemm.
|
|
* Test with the following options:
|
|
*
|
|
* matrix A is conjugate and not transposed
|
|
* matrix B is conjugate and not transposed
|
|
*/
|
|
CTEST(zgemm, conjnotransa_conjnotransb)
|
|
{
|
|
blasint n = DATASIZE, m = DATASIZE, k = DATASIZE;
|
|
blasint lda = DATASIZE, ldb = DATASIZE, ldc = DATASIZE;
|
|
char transa = 'R';
|
|
char transb = 'R';
|
|
double alpha[] = {-2.0, 1.0};
|
|
double beta[] = {1.0, -1.0};
|
|
|
|
double norm = check_zgemm(transa, transb, m, n, k, alpha, lda, ldb, beta, ldc);
|
|
|
|
ASSERT_DBL_NEAR_TOL(0.0, norm, DOUBLE_EPS);
|
|
}
|
|
|
|
/**
|
|
* Test zgemm with the conjugate matrices by conjugating and not transposed matrices
|
|
* and comparing it with the non-conjugate zgemm.
|
|
* Test with the following options:
|
|
*
|
|
* matrix A is conjugate and not transposed
|
|
* matrix B is transposed
|
|
*/
|
|
CTEST(zgemm, conjnotransa_transb)
|
|
{
|
|
blasint n = DATASIZE, m = DATASIZE, k = DATASIZE;
|
|
blasint lda = DATASIZE, ldb = DATASIZE, ldc = DATASIZE;
|
|
char transa = 'R';
|
|
char transb = 'T';
|
|
double alpha[] = {-2.0, 1.0};
|
|
double beta[] = {1.0, -1.0};
|
|
|
|
double norm = check_zgemm(transa, transb, m, n, k, alpha, lda, ldb, beta, ldc);
|
|
|
|
ASSERT_DBL_NEAR_TOL(0.0, norm, DOUBLE_EPS);
|
|
}
|
|
|
|
/**
|
|
* Test zgemm with the conjugate matrices by conjugating and not transposed matrices
|
|
* and comparing it with the non-conjugate zgemm.
|
|
* Test with the following options:
|
|
*
|
|
* matrix A is transposed
|
|
* matrix B is conjugate and not transposed
|
|
*/
|
|
CTEST(zgemm, transa_conjnotransb)
|
|
{
|
|
blasint n = DATASIZE, m = DATASIZE, k = DATASIZE;
|
|
blasint lda = DATASIZE, ldb = DATASIZE, ldc = DATASIZE;
|
|
char transa = 'T';
|
|
char transb = 'R';
|
|
double alpha[] = {-2.0, 1.0};
|
|
double beta[] = {1.0, -1.0};
|
|
|
|
double norm = check_zgemm(transa, transb, m, n, k, alpha, lda, ldb, beta, ldc);
|
|
|
|
ASSERT_DBL_NEAR_TOL(0.0, norm, DOUBLE_EPS);
|
|
}
|
|
#endif |