floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #3548 from martin-frbg/rela-gemmt 

Enable the ?GEMMT functions in ReLAPACK
This commit is contained in:
Martin Kroeker 2022-02-26 21:48:39 +01:00 committed by GitHub
parent ab304cca69 4058f32492
commit 9c3e0bf319
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
relapack/config.h
View File

@ -115,7 +115,7 @@
#define INCLUDE_CTGSYL INCLUDE_XTGSYL #define INCLUDE_CTGSYL INCLUDE_XTGSYL
#define INCLUDE_ZTGSYL INCLUDE_XTGSYL #define INCLUDE_ZTGSYL INCLUDE_XTGSYL
#define INCLUDE_XGEMMT 0 #define INCLUDE_XGEMMT 1
#define INCLUDE_SGEMMT INCLUDE_XGEMMT #define INCLUDE_SGEMMT INCLUDE_XGEMMT
#define INCLUDE_DGEMMT INCLUDE_XGEMMT #define INCLUDE_DGEMMT INCLUDE_XGEMMT
#define INCLUDE_CGEMMT INCLUDE_XGEMMT #define INCLUDE_CGEMMT INCLUDE_XGEMMT

12
relapack/src/lapack_wrappers.c
View File

@ -566,7 +566,8 @@ void LAPACK(sgemmt)(
const float *B, const blasint *ldB, const float *B, const blasint *ldB,
const float *beta, float *C, const blasint *ldC const float *beta, float *C, const blasint *ldC
) { ) {
RELAPACK_sgemmt(uplo, n, A, ldA, info); blasint info;
RELAPACK_sgemmt(uplo, transA, transB, n, k, alpha, A, ldA, B, ldB, beta, C, info);
} }
#endif #endif
@ -578,7 +579,8 @@ void LAPACK(dgemmt)(
const double *B, const blasint *ldB, const double *B, const blasint *ldB,
const double *beta, double *C, const blasint *ldC const double *beta, double *C, const blasint *ldC
) { ) {
RELAPACK_dgemmt(uplo, n, A, ldA, info); blasint info;
RELAPACK_dgemmt(uplo, transA, transB, n, k, alpha, A, ldA, B, ldB, beta, C, info);
} }
#endif #endif
@ -590,7 +592,8 @@ void LAPACK(cgemmt)(
const float *B, const blasint *ldB, const float *B, const blasint *ldB,
const float *beta, float *C, const blasint *ldC const float *beta, float *C, const blasint *ldC
) { ) {
RELAPACK_cgemmt(uplo, n, A, ldA, info); blasint info;
RELAPACK_cgemmt(uplo, transA, transB, n, k, alpha, A, ldA, B, ldB, beta, C, info);
} }
#endif #endif
@ -602,6 +605,7 @@ void LAPACK(zgemmt)(
const double *B, const blasint *ldB, const double *B, const blasint *ldB,
const double *beta, double *C, const blasint *ldC const double *beta, double *C, const blasint *ldC
) { ) {
RELAPACK_zgemmt(uplo, n, A, ldA, info); blasint info;
RELAPACK_zgemmt(uplo, transA, transB, n, k, alpha, A, ldA, B, ldB, beta, C, info);
} }
#endif #endif