Browse Source

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

Enable the ?GEMMT functions in ReLAPACK
tags/v0.3.21
Martin Kroeker GitHub 3 years ago
parent
commit
9c3e0bf319
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 5 deletions
  1. +1
    -1
      relapack/config.h
  2. +8
    -4
      relapack/src/lapack_wrappers.c

+ 1
- 1
relapack/config.h View File

@@ -115,7 +115,7 @@
#define INCLUDE_CTGSYL INCLUDE_XTGSYL
#define INCLUDE_ZTGSYL INCLUDE_XTGSYL

#define INCLUDE_XGEMMT 0
#define INCLUDE_XGEMMT 1
#define INCLUDE_SGEMMT INCLUDE_XGEMMT
#define INCLUDE_DGEMMT INCLUDE_XGEMMT
#define INCLUDE_CGEMMT INCLUDE_XGEMMT


+ 8
- 4
relapack/src/lapack_wrappers.c View File

@@ -566,7 +566,8 @@ void LAPACK(sgemmt)(
const float *B, const blasint *ldB,
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

@@ -578,7 +579,8 @@ void LAPACK(dgemmt)(
const double *B, const blasint *ldB,
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

@@ -590,7 +592,8 @@ void LAPACK(cgemmt)(
const float *B, const blasint *ldB,
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

@@ -602,6 +605,7 @@ void LAPACK(zgemmt)(
const double *B, const blasint *ldB,
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

Loading…
Cancel
Save