PLASMA  2.4.5
PLASMA - Parallel Linear Algebra for Scalable Multi-core Architectures
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups
Macros | Functions
pdtrsm.c File Reference
#include "common.h"
Include dependency graph for pdtrsm.c:

Go to the source code of this file.

Macros

#define A(m, n)   BLKADDR(A, double, m, n)
#define B(m, n)   BLKADDR(B, double, m, n)

Functions

void plasma_pdtrsm (plasma_context_t *plasma)
void plasma_pdtrsm_quark (PLASMA_enum side, PLASMA_enum uplo, PLASMA_enum trans, PLASMA_enum diag, double alpha, PLASMA_desc A, PLASMA_desc B, PLASMA_sequence *sequence, PLASMA_request *request)

Detailed Description

PLASMA auxiliary routines PLASMA is a software package provided by Univ. of Tennessee, Univ. of California Berkeley and Univ. of Colorado Denver

Version:
2.4.5
Author:
Jakub Kurzak
Hatem Ltaief
Mathieu Faverge
Date:
2010-11-15 d Tue Nov 22 14:35:39 2011

Definition in file pdtrsm.c.

Macro Definition Documentation

#define A (   m,
  n 
)    BLKADDR(A, double, m, n)

Definition at line 19 of file pdtrsm.c.

#define B (   m,
  n 
)    BLKADDR(B, double, m, n)

Definition at line 20 of file pdtrsm.c.

Function Documentation

void plasma_pdtrsm ( plasma_context_t *  plasma)

Parallel tile triangular solve - static scheduling

Definition at line 24 of file pdtrsm.c.

References A, B, BLKLDD, CORE_dgemm(), CORE_dtrsm(), diag, plasma_desc_t::m, plasma_desc_t::mb, plasma_desc_t::mt, plasma_desc_t::n, plasma_desc_t::nb, plasma_desc_t::nt, PLASMA_RANK, PLASMA_SIZE, PLASMA_SUCCESS, plasma_unpack_args_9, PlasmaLeft, PlasmaLower, PlasmaNoTrans, PlasmaUpper, side, ss_cond_set, ss_cond_wait, ss_finalize, ss_init, plasma_sequence_t::status, trans, and uplo.

{
PLASMA_enum side;
PLASMA_enum uplo;
PLASMA_enum trans;
PLASMA_enum diag;
double alpha;
PLASMA_desc A;
PLASMA_desc B;
PLASMA_sequence *sequence;
PLASMA_request *request;
int k, m, n;
int next_k;
int next_m;
int next_n;
int lda, ldb;
int tempkm, tempnn, tempmm, tempkn;
double zone = (double) 1.0;
double mzone = (double)-1.0;
double lalpha;
double minvalpha;
plasma_unpack_args_9(side, uplo, trans, diag, alpha, A, B, sequence, request);
minvalpha = mzone / alpha;
if (sequence->status != PLASMA_SUCCESS)
return;
ss_init(B.mt, B.nt, -1);
/*
* PlasmaLeft
*/
if (side == PlasmaLeft) {
k = 0;
m = PLASMA_RANK;
while (m >= B.mt) {
k++;
m = m - B.mt + k;
}
n = 0;
while (k < B.mt && m < B.mt) {
next_n = n;
next_m = m;
next_k = k;
next_n++;
if (next_n >= B.nt) {
next_m += PLASMA_SIZE;
while (next_m >= B.mt && next_k < B.mt) {
next_k++;
next_m = next_m - B.mt + next_k;
}
next_n = 0;
}
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
lalpha = k == 0 ? alpha : zone;
if (m == k) {
ss_cond_wait(m, n, k-1);
/*
* PlasmaLeft / PlasmaLower / PlasmaNoTrans
* PlasmaLeft / PlasmaUpper / Plasma[Conj]Trans
*/
if ((uplo == PlasmaLower && trans == PlasmaNoTrans)
|| (uplo == PlasmaUpper && trans != PlasmaNoTrans)) {
tempkm = k == B.mt-1 ? B.m-k*B.mb : B.mb;
lda = BLKLDD(A, k);
ldb = BLKLDD(B, k);
CORE_dtrsm(
side, uplo, trans, diag,
tempkm, tempnn,
lalpha, A(k, k), lda,
B(k, n), ldb);
}
/*
* PlasmaLeft / PlasmaLower / Plasma[Cojn]Trans
* PlasmaLeft / PlasmaUpper / PlasmaNoTrans
*/
else {
tempkm = k == 0 ? B.m-(B.mt-1)*B.mb : B.mb;
lda = BLKLDD(A, B.mt-1-k);
ldb = BLKLDD(B, B.mt-1-k);
CORE_dtrsm(
side, uplo, trans, diag,
tempkm, tempnn,
lalpha, A(B.mt-1-k, B.mt-1-k), lda,
B(B.mt-1-k, n ), ldb);
}
ss_cond_set(k, n, k);
}
else {
ss_cond_wait(k, n, k);
ss_cond_wait(m, n, k-1);
/*
* PlasmaRight / PlasmaLower / PlasmaNoTrans
*/
if (uplo == PlasmaLower) {
if (trans == PlasmaNoTrans) {
lda = BLKLDD(A, m);
ldb = BLKLDD(B, m);
CORE_dgemm(
PlasmaNoTrans, PlasmaNoTrans,
tempmm, tempnn, B.mb,
mzone, A(m, k), lda,
B(k, n), B.mb,
lalpha, B(m, n), ldb);
}
/*
* PlasmaRight / PlasmaLower / Plasma[Conj]Trans
*/
else {
tempkm = k == 0 ? A.m-(A.mt-1)*A.mb : A.mb;
lda = BLKLDD(A, B.mt-1-k);
ldb = BLKLDD(B, B.mt-1-k);
CORE_dgemm(
trans, PlasmaNoTrans,
B.mb, tempnn, tempkm,
mzone, A(A.mt-1-k, A.mt-1-m), lda,
B(B.mt-1-k, n ), ldb,
lalpha, B(B.mt-1-m, n ), B.mb);
}
}
else {
/*
* PlasmaRight / PlasmaUpper / PlasmaNoTrans
*/
if (trans == PlasmaNoTrans) {
tempkm = k == 0 ? A.m-(A.mt-1)*A.mb : A.mb;
ldb = BLKLDD(B, B.mt-1-k);
CORE_dgemm(
PlasmaNoTrans, PlasmaNoTrans,
B.mb, tempnn, tempkm,
mzone, A(A.mt-1-m, A.mt-1-k), A.mb,
B(B.mt-1-k, n ), ldb,
lalpha, B(B.mt-1-m, n ), B.mb);
}
/*
* PlasmaRight / PlasmaUpper / Plasma[Conj]Trans
*/
else {
ldb = BLKLDD(B, m);
CORE_dgemm(
trans, PlasmaNoTrans,
tempmm, tempnn, B.mb,
mzone, A(k, m), A.mb,
B(k, n), B.mb,
lalpha, B(m, n), ldb);
}
}
ss_cond_set(m, n, k);
}
n = next_n;
m = next_m;
k = next_k;
}
}
/*
* PlasmaRight
*/
else {
k = 0;
n = PLASMA_RANK;
while (n >= B.nt) {
k++;
n = n - B.nt + k;
}
m = 0;
while (k < B.nt && n < B.nt) {
next_n = n;
next_m = m;
next_k = k;
next_m++;
if (next_m >= B.mt) {
next_n += PLASMA_SIZE;
while (next_n >= B.nt && next_k < B.nt) {
next_k++;
next_n = next_n - B.nt + next_k;
}
next_m = 0;
}
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
lalpha = k == 0 ? alpha : zone;
if (n == k) {
ss_cond_wait(m, n, k-1);
/*
* PlasmaRight / PlasmaLower / PlasmaNoTrans
*/
if (uplo == PlasmaLower) {
if (trans == PlasmaNoTrans) {
tempkn = k == 0 ? B.n-(B.nt-1)*B.nb : B.nb;
lda = BLKLDD(A, B.nt-1-k);
ldb = BLKLDD(B, m);
CORE_dtrsm(
side, uplo, trans, diag,
tempmm, tempkn,
lalpha, A(B.nt-1-k, B.nt-1-k), lda,
B(m, B.nt-1-k), ldb);
}
/*
* PlasmaRight / PlasmaLower / Plasma[Conj]Trans
*/
else {
tempkn = k == B.nt-1 ? B.n-k*B.nb : B.nb;
lda = BLKLDD(A, k);
ldb = BLKLDD(B, m);
CORE_dtrsm(
side, uplo, trans, diag,
tempmm, tempkn,
alpha, A(k, k), lda,
B(m, k), ldb);
}
}
else {
/*
* PlasmaRight / PlasmaUpper / PlasmaNoTrans
*/
if (trans == PlasmaNoTrans) {
tempkn = k == B.nt-1 ? B.n-k*B.nb : B.nb;
lda = BLKLDD(A, k);
ldb = BLKLDD(B, m);
CORE_dtrsm(
side, uplo, trans, diag,
tempmm, tempkn,
lalpha, A(k, k), lda,
B(m, k), ldb);
}
/*
* PlasmaRight / PlasmaUpper / Plasma[Conj]Trans
*/
else {
tempkn = k == 0 ? B.n-(B.nt-1)*B.nb : B.nb;
lda = BLKLDD(A, B.nt-1-k);
ldb = BLKLDD(B, m);
CORE_dtrsm(
side, uplo, trans, diag,
tempmm, tempkn,
alpha, A(B.nt-1-k, B.nt-1-k), lda,
B(m, B.nt-1-k), ldb);
}
}
ss_cond_set(m, k, k);
}
else {
ss_cond_wait(m, k, k);
ss_cond_wait(m, n, k-1);
/*
* PlasmaRight / PlasmaLower / PlasmaNoTrans
*/
if (uplo == PlasmaLower) {
if (trans == PlasmaNoTrans) {
tempkn = k == 0 ? B.n-(B.nt-1)*B.nb : B.nb;
lda = BLKLDD(A, B.nt-1-k);
ldb = BLKLDD(B, m);
CORE_dgemm(
PlasmaNoTrans, PlasmaNoTrans,
tempmm, B.mb, tempkn,
mzone, B(m, B.nt-1-k), ldb,
A(B.nt-1-k, B.nt-1-n), lda,
lalpha, B(m, B.nt-1-n), ldb);
}
/*
* PlasmaRight / PlasmaLower / Plasma[Conj]Trans
*/
else {
lda = BLKLDD(A, n);
ldb = BLKLDD(B, m);
CORE_dgemm(
PlasmaNoTrans, trans,
tempmm, tempnn, B.mb,
minvalpha, B(m, k), ldb,
A(n, k), lda,
zone, B(m, n), ldb);
}
}
else {
/*
* PlasmaRight / PlasmaUpper / PlasmaNoTrans
*/
if (trans == PlasmaNoTrans) {
lda = BLKLDD(A, k);
ldb = BLKLDD(B, m);
CORE_dgemm(
PlasmaNoTrans, PlasmaNoTrans,
tempmm, tempnn, B.mb,
mzone, B(m, k), ldb,
A(k, n), lda,
lalpha, B(m, n), ldb);
}
/*
* PlasmaRight / PlasmaUpper / Plasma[Conj]Trans
*/
else {
tempkn = k == 0 ? B.n-(B.nt-1)*B.nb : B.nb;
ldb = BLKLDD(B, m);
CORE_dgemm(
PlasmaNoTrans, trans,
tempmm, B.nb, tempkn,
minvalpha, B(m, B.nt-1-k), ldb,
A(B.nt-1-n, B.nt-1-k), A.mb,
zone, B(m, B.nt-1-n), ldb);
}
}
ss_cond_set(m, n, k);
}
n = next_n;
m = next_m;
k = next_k;
}
}
ss_finalize();
}

Here is the call graph for this function:

Here is the caller graph for this function:

void plasma_pdtrsm_quark ( PLASMA_enum  side,
PLASMA_enum  uplo,
PLASMA_enum  trans,
PLASMA_enum  diag,
double  alpha,
PLASMA_desc  A,
PLASMA_desc  B,
PLASMA_sequence *  sequence,
PLASMA_request *  request 
)

Parallel tile triangular solve - dynamic scheduling

Definition at line 347 of file pdtrsm.c.

References A, B, BLKLDD, plasma_desc_t::m, plasma_desc_t::mb, plasma_desc_t::mt, plasma_desc_t::n, plasma_desc_t::nb, plasma_desc_t::nt, plasma_context_self(), PLASMA_SUCCESS, PlasmaLeft, PlasmaNoTrans, PlasmaUpper, plasma_context_struct::quark, QUARK_CORE_dgemm(), QUARK_CORE_dtrsm(), plasma_sequence_t::quark_sequence, QUARK_Task_Flag_Set(), Quark_Task_Flags_Initializer, plasma_sequence_t::status, and TASK_SEQUENCE.

{
plasma_context_t *plasma;
Quark_Task_Flags task_flags = Quark_Task_Flags_Initializer;
int k, m, n;
int lda, ldan, ldb;
int tempkm, tempkn, tempmm, tempnn;
double zone = (double) 1.0;
double mzone = (double)-1.0;
double minvalpha = (double)-1.0 / alpha;
double lalpha;
plasma = plasma_context_self();
if (sequence->status != PLASMA_SUCCESS)
return;
QUARK_Task_Flag_Set(&task_flags, TASK_SEQUENCE, (intptr_t)sequence->quark_sequence);
/*
* PlasmaLeft / PlasmaUpper / PlasmaNoTrans
*/
if (side == PlasmaLeft) {
if (uplo == PlasmaUpper) {
if (trans == PlasmaNoTrans) {
for (k = 0; k < B.mt; k++) {
tempkm = k == 0 ? B.m-(B.mt-1)*B.mb : B.mb;
lda = BLKLDD(A, B.mt-1-k);
ldb = BLKLDD(B, B.mt-1-k);
lalpha = k == 0 ? alpha : zone;
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_dtrsm(
plasma->quark, &task_flags,
side, uplo, trans, diag,
tempkm, tempnn, A.mb,
lalpha, A(B.mt-1-k, B.mt-1-k), lda, /* lda * tempkm */
B(B.mt-1-k, n), ldb); /* ldb * tempnn */
}
for (m = k+1; m < B.mt; m++) {
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_dgemm(
plasma->quark, &task_flags,
PlasmaNoTrans, PlasmaNoTrans,
B.mb, tempnn, tempkm, A.mb,
mzone, A(B.mt-1-m, B.mt-1-k), A.mb,
B(B.mt-1-k, n ), ldb,
lalpha, B(B.mt-1-m, n ), B.mb);
}
}
}
}
/*
* PlasmaLeft / PlasmaUpper / Plasma[Conj]Trans
*/
else {
for (k = 0; k < B.mt; k++) {
tempkm = k == B.mt-1 ? B.m-k*B.mb : B.mb;
lda = BLKLDD(A, k);
ldb = BLKLDD(B, k);
lalpha = k == 0 ? alpha : zone;
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_dtrsm(
plasma->quark, &task_flags,
side, uplo, trans, diag,
tempkm, tempnn, A.mb,
lalpha, A(k, k), lda,
B(k, n), ldb);
}
for (m = k+1; m < B.mt; m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
ldb = BLKLDD(B, m);
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_dgemm(
plasma->quark, &task_flags,
trans, PlasmaNoTrans,
tempmm, tempnn, B.mb, A.mb,
mzone, A(k, m), A.mb,
B(k, n), B.mb,
lalpha, B(m, n), ldb);
}
}
}
}
}
/*
* PlasmaLeft / PlasmaLower / PlasmaNoTrans
*/
else {
if (trans == PlasmaNoTrans) {
for (k = 0; k < B.mt; k++) {
tempkm = k == B.mt-1 ? B.m-k*B.mb : B.mb;
lda = BLKLDD(A, k);
ldb = BLKLDD(B, k);
lalpha = k == 0 ? alpha : zone;
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_dtrsm(
plasma->quark, &task_flags,
side, uplo, trans, diag,
tempkm, tempnn, A.mb,
lalpha, A(k, k), lda,
B(k, n), ldb);
}
for (m = k+1; m < B.mt; m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
lda = BLKLDD(A, m);
ldb = BLKLDD(B, m);
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_dgemm(
plasma->quark, &task_flags,
PlasmaNoTrans, PlasmaNoTrans,
tempmm, tempnn, B.mb, A.mb,
mzone, A(m, k), lda,
B(k, n), B.mb,
lalpha, B(m, n), ldb);
}
}
}
}
/*
* PlasmaLeft / PlasmaLower / Plasma[Conj]Trans
*/
else {
for (k = 0; k < B.mt; k++) {
tempkm = k == 0 ? B.m-(B.mt-1)*B.mb : B.mb;
lda = BLKLDD(A, B.mt-1-k);
ldb = BLKLDD(B, B.mt-1-k);
lalpha = k == 0 ? alpha : zone;
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_dtrsm(
plasma->quark, &task_flags,
side, uplo, trans, diag,
tempkm, tempnn, A.mb,
lalpha, A(B.mt-1-k, B.mt-1-k), lda,
B(B.mt-1-k, n), ldb);
}
for (m = k+1; m < B.mt; m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_dgemm(
plasma->quark, &task_flags,
trans, PlasmaNoTrans,
B.mb, tempnn, tempkm, A.mb,
mzone, A(B.mt-1-k, B.mt-1-m), lda,
B(B.mt-1-k, n ), ldb,
lalpha, B(B.mt-1-m, n ), B.mb);
}
}
}
}
}
}
/*
* PlasmaRight / PlasmaUpper / PlasmaNoTrans
*/
else {
if (uplo == PlasmaUpper) {
if (trans == PlasmaNoTrans) {
for (k = 0; k < B.nt; k++) {
tempkn = k == B.nt-1 ? B.n-k*B.nb : B.nb;
lda = BLKLDD(A, k);
lalpha = k == 0 ? alpha : zone;
for (m = 0; m < B.mt; m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
ldb = BLKLDD(B, m);
QUARK_CORE_dtrsm(
plasma->quark, &task_flags,
side, uplo, trans, diag,
tempmm, tempkn, A.mb,
lalpha, A(k, k), lda, /* lda * tempkn */
B(m, k), ldb); /* ldb * tempkn */
}
for (m = 0; m < B.mt; m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
ldb = BLKLDD(B, m);
for (n = k+1; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_dgemm(
plasma->quark, &task_flags,
PlasmaNoTrans, PlasmaNoTrans,
tempmm, tempnn, B.mb, A.mb,
mzone, B(m, k), ldb, /* ldb * B.mb */
A(k, n), lda, /* lda * tempnn */
lalpha, B(m, n), ldb); /* ldb * tempnn */
}
}
}
}
/*
* PlasmaRight / PlasmaUpper / Plasma[Conj]Trans
*/
else {
for (k = 0; k < B.nt; k++) {
tempkn = k == 0 ? B.n-(B.nt-1)*B.nb : B.nb;
lda = BLKLDD(A, B.nt-1-k);
for (m = 0; m < B.mt; m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
ldb = BLKLDD(B, m);
QUARK_CORE_dtrsm(
plasma->quark, &task_flags,
side, uplo, trans, diag,
tempmm, tempkn, A.mb,
alpha, A(B.nt-1-k, B.nt-1-k), lda, /* lda * tempkn */
B( m, B.nt-1-k), ldb); /* ldb * tempkn */
for (n = k+1; n < B.nt; n++) {
QUARK_CORE_dgemm(
plasma->quark, &task_flags,
PlasmaNoTrans, trans,
tempmm, B.nb, tempkn, A.mb,
minvalpha, B(m, B.nt-1-k), ldb, /* ldb * tempkn */
A(B.nt-1-n, B.nt-1-k), A.mb, /* A.mb * tempkn (Never last row) */
zone, B(m, B.nt-1-n), ldb); /* ldb * B.nb */
}
}
}
}
}
/*
* PlasmaRight / PlasmaLower / PlasmaNoTrans
*/
else {
if (trans == PlasmaNoTrans) {
for (k = 0; k < B.nt; k++) {
tempkn = k == 0 ? B.n-(B.nt-1)*B.nb : B.nb;
lda = BLKLDD(A, B.nt-1-k);
lalpha = k == 0 ? alpha : zone;
for (m = 0; m < B.mt; m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
ldb = BLKLDD(B, m);
QUARK_CORE_dtrsm(
plasma->quark, &task_flags,
side, uplo, trans, diag,
tempmm, tempkn, A.mb,
lalpha, A(B.nt-1-k, B.nt-1-k), lda, /* lda * tempkn */
B( m, B.nt-1-k), ldb); /* ldb * tempkn */
for (n = k+1; n < B.nt; n++) {
QUARK_CORE_dgemm(
plasma->quark, &task_flags,
PlasmaNoTrans, PlasmaNoTrans,
tempmm, B.nb, tempkn, A.mb,
mzone, B(m, B.nt-1-k), ldb, /* ldb * tempkn */
A(B.nt-1-k, B.nt-1-n), lda, /* lda * B.nb */
lalpha, B(m, B.nt-1-n), ldb); /* ldb * B.nb */
}
}
}
}
/*
* PlasmaRight / PlasmaLower / Plasma[Conj]Trans
*/
else {
for (k = 0; k < B.nt; k++) {
tempkn = k == B.nt-1 ? B.n-k*B.nb : B.nb;
lda = BLKLDD(A, k);
for (m = 0; m < B.mt; m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
ldb = BLKLDD(B, m);
QUARK_CORE_dtrsm(
plasma->quark, &task_flags,
side, uplo, trans, diag,
tempmm, tempkn, A.mb,
alpha, A(k, k), lda, /* lda * tempkn */
B(m, k), ldb); /* ldb * tempkn */
for (n = k+1; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
ldan = BLKLDD(A, n);
QUARK_CORE_dgemm(
plasma->quark, &task_flags,
PlasmaNoTrans, trans,
tempmm, tempnn, B.mb, A.mb,
minvalpha, B(m, k), ldb, /* ldb * tempkn */
A(n, k), ldan, /* ldan * tempkn */
zone, B(m, n), ldb); /* ldb * tempnn */
}
}
}
}
}
}
}

Here is the call graph for this function:

Here is the caller graph for this function: