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
psormlqrh.c File Reference
#include "common.h"
Include dependency graph for psormlqrh.c:

Go to the source code of this file.

Macros

#define A(m, n)   BLKADDR(A, float, (m), (n))
#define B(m, n)   BLKADDR(B, float, (m), (n))
#define T(m, n)   BLKADDR(T, float, (m), (n))
#define T2(m, n)   BLKADDR(T, float, (m), (n)+A.nt)

Functions

void plasma_psormlqrh_quark (PLASMA_enum side, PLASMA_enum trans, PLASMA_desc A, PLASMA_desc B, PLASMA_desc T, int BS, 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:
Hatem Ltaief
Jakub Kurzak
Dulceneia Becker
Date:
2010-11-15 s Tue Nov 22 14:35:41 2011

Definition in file psormlqrh.c.

Macro Definition Documentation

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

Definition at line 19 of file psormlqrh.c.

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

Definition at line 20 of file psormlqrh.c.

#define T (   m,
  n 
)    BLKADDR(T, float, (m), (n))

Definition at line 21 of file psormlqrh.c.

#define T2 (   m,
  n 
)    BLKADDR(T, float, (m), (n)+A.nt)

Definition at line 22 of file psormlqrh.c.

Function Documentation

void plasma_psormlqrh_quark ( PLASMA_enum  side,
PLASMA_enum  trans,
PLASMA_desc  A,
PLASMA_desc  B,
PLASMA_desc  T,
int  BS,
PLASMA_sequence *  sequence,
PLASMA_request *  request 
)

Parallel application of Q using tile V - LQ factorization (reduction Householder) - dynamic scheduling

Definition at line 27 of file psormlqrh.c.

References A, B, BLKLDD, plasma_desc_t::m, plasma_desc_t::mb, min, plasma_desc_t::mt, plasma_desc_t::n, plasma_desc_t::nb, plasma_desc_t::nt, plasma_context_self(), PLASMA_IB, PLASMA_SUCCESS, PlasmaLeft, PlasmaNoTrans, plasma_context_struct::quark, QUARK_CORE_sormlq(), QUARK_CORE_stsmlq(), QUARK_CORE_sttmlq(), plasma_sequence_t::quark_sequence, QUARK_Task_Flag_Set(), Quark_Task_Flags_Initializer, plasma_sequence_t::status, T, T2, and TASK_SEQUENCE.

{
plasma_context_t *plasma;
Quark_Task_Flags task_flags = Quark_Task_Flags_Initializer;
int k, m, n;
int K, N, RD, lastRD;
int ldaN, ldak;
int ldbN, ldbm, ldbNRD;
int tempNn, tempkm, tempnn, tempmm, tempNRDn, tempkmin;
int ib;
plasma = plasma_context_self();
if (sequence->status != PLASMA_SUCCESS)
return;
QUARK_Task_Flag_Set(&task_flags, TASK_SEQUENCE, (intptr_t)sequence->quark_sequence);
ib = PLASMA_IB;
K = min(A.mt, A.nt);
if (side == PlasmaLeft ) {
if (trans == PlasmaNoTrans) {
/*
* PlasmaLeft / PlasmaNoTrans
*/
for (k = 0; k < K; k++) {
tempkm = k == A.mt-1 ? A.m-k*A.mb : A.mb;
ldak = BLKLDD(A, k);
for (N = k; N < A.nt; N += BS) {
tempNn = N == A.nt-1 ? A.n-N*A.nb : A.nb;
tempkmin = min(tempkm,tempNn);
ldaN = BLKLDD(A, N);
ldbN = BLKLDD(B, N);
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_sormlq(
plasma->quark, &task_flags,
side, trans,
tempNn, tempnn,
tempkmin, ib, T.nb,
A(k, N), ldak,
T(k, N), T.mb,
B(N, n), ldbN);
}
for (m = N+1; m < min(N+BS, A.nt); m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
ldbm = BLKLDD(B, m);
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_stsmlq(
plasma->quark, &task_flags,
side, trans,
B.nb, tempnn, tempmm, tempnn,
tempkm, ib, T.nb,
B(N, n), ldbN,
B(m, n), ldbm,
A(k, m), ldak,
T(k, m), T.mb);
}
}
}
for (RD = BS; RD < A.nt-k; RD *= 2) {
for (N = k; N+RD < A.nt; N += 2*RD) {
tempNRDn = N+RD == A.nt-1 ? A.n-(N+RD)*A.nb : A.nb;
ldbN = BLKLDD(B, N );
ldbNRD = BLKLDD(B, N+RD);
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_sttmlq(
plasma->quark, &task_flags,
side, trans,
B.mb, tempnn, tempNRDn, tempnn,
tempkm, ib, T.nb,
B (N, n), ldbN,
B (N+RD, n), ldbNRD,
A (k, N+RD), ldak,
T2(k, N+RD), T.mb);
}
}
}
}
} else {
/*
* PlasmaLeft / PlasmaTrans
*/
for (k = K-1; k >= 0; k--) {
tempkm = k == A.mt-1 ? A.m-k*A.mb : A.mb;
ldak = BLKLDD(A, k);
lastRD = 0;
for (RD = BS; RD < A.nt-k; RD *= 2)
lastRD = RD;
for (RD = lastRD; RD >= BS; RD /= 2) {
for (N = k; N+RD < A.nt; N += 2*RD) {
tempNRDn = N+RD == A.nt-1 ? A.n-(N+RD)*A.nb : A.nb;
ldbN = BLKLDD(B, N );
ldbNRD = BLKLDD(B, N+RD);
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_sttmlq(
plasma->quark, &task_flags,
side, trans,
B.nb, tempnn, tempNRDn, tempnn,
tempkm, ib, T.nb,
B (N, n), ldbN,
B (N+RD, n), ldbNRD,
A (k, N+RD), ldak,
T2(k, N+RD), T.mb);
}
}
}
for (N = k; N < A.nt; N += BS) {
tempNn = N == A.nt-1 ? A.n-N*A.nb : A.nb;
tempkmin = min(tempkm,tempNn);
ldaN = BLKLDD(A, N);
ldbN = BLKLDD(B, N);
for (m = min(N+BS, A.nt)-1; m > N; m--) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
ldbm = BLKLDD(B, m);
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_stsmlq(
plasma->quark, &task_flags,
side, trans,
B.mb, tempnn, tempmm, tempnn,
tempkm, ib, T.nb,
B(N, n), ldbN,
B(m, n), ldbm,
A(k, m), ldak,
T(k, m), T.mb);
}
}
for (n = 0; n < B.nt; n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
QUARK_CORE_sormlq(
plasma->quark, &task_flags,
side, trans,
tempNn, tempnn,
tempkmin, ib, T.nb,
A(k, N), ldak,
T(k, N), T.mb,
B(N, n), ldbN);
}
}
}
}
} else {
if (trans == PlasmaNoTrans) {
/*
* PlasmaRight / PlasmaNoTrans
*/
for (k = K-1; k >= 0; k--) {
tempkm = k == A.mt-1 ? A.m-k*A.mb : A.mb;
ldak = BLKLDD(A, k);
lastRD = 0;
for (RD = BS; RD < A.nt-k; RD *= 2)
lastRD = RD;
for (RD = lastRD; RD >= BS; RD /= 2) {
for (N = k; N+RD < A.nt; N += 2*RD) {
tempNRDn = N+RD == A.nt-1 ? A.n-(N+RD)*A.nb : A.nb;
for (m = 0; m < B.mt; m++) {
ldbm = BLKLDD(B, m);
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
QUARK_CORE_sttmlq(
plasma->quark, &task_flags,
side, trans,
tempmm, B.nb, tempmm, tempNRDn,
tempkm, ib, T.nb,
B (m, N ), ldbm,
B (m, N+RD), ldbm,
A (k, N+RD), ldak,
T2(k, N+RD), T.mb);
}
}
}
for (N = k; N < A.nt; N += BS) {
tempNn = N == A.nt-1 ? A.n-N*A.nb : A.nb;
tempkmin = min(tempkm,tempNn);
for (n = min(N+BS, A.nt)-1; n > N; n--) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
for (m = 0; m < B.mt; m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
ldbm = BLKLDD(B, m);
QUARK_CORE_stsmlq(
plasma->quark, &task_flags,
side, trans,
tempmm, B.nb, tempmm, tempnn,
tempkm, ib, T.nb,
B(m, N), ldbm,
B(m, n), ldbm,
A(k, n), ldak,
T(k, n), T.mb);
}
}
for (m = 0; m < B.mt; m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
ldbm = BLKLDD(B, m);
QUARK_CORE_sormlq(
plasma->quark, &task_flags,
side, trans,
tempmm, tempNn,
tempkmin, ib, T.nb,
A(k, N), ldak,
T(k, N), T.mb,
B(m, N), ldbm);
}
}
}
} else {
/*
* PlasmaRight / PlasmaTrans
*/
for (k = 0; k < K; k++) {
tempkm = k == A.mt-1 ? A.m-k*A.mb : A.mb;
ldak = BLKLDD(A, k);
for (N = k; N < A.nt; N += BS) {
tempNn = N == A.nt-1 ? A.n-N*A.nb : A.nb;
tempkmin = min(tempkm,tempNn);
ldaN = BLKLDD(A, N);
for (m = 0; m < B.mt; m++) {
ldbm = BLKLDD(B, m);
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
QUARK_CORE_sormlq(
plasma->quark, &task_flags,
side, trans,
tempmm, tempNn,
tempkmin, ib, T.nb,
A(k, N), ldaN,
T(k, N), T.mb,
B(m, N), ldbm);
}
for (n = N+1; n < min(N+BS, A.nt); n++) {
tempnn = n == B.nt-1 ? B.n-n*B.nb : B.nb;
for (m = 0; m < B.mt; m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
ldbm = BLKLDD(B, m);
QUARK_CORE_stsmlq(
plasma->quark, &task_flags,
side, trans,
tempmm, tempNn, tempmm, tempnn,
tempkm, ib, T.nb,
B(m, N), ldbm,
B(m, n), ldbm,
A(k, n), ldak,
T(k, n), T.mb);
}
}
}
for (RD = BS; RD < A.nt-k; RD *= 2) {
for (N = k; N+RD < A.nt; N += 2*RD) {
tempNRDn = N+RD == A.nt-1 ? A.n-(N+RD)*A.nb : A.nb;
for (m = 0; m < B.mt; m++) {
tempmm = m == B.mt-1 ? B.m-m*B.mb : B.mb;
ldbm = BLKLDD(B, m);
QUARK_CORE_sttmlq(
plasma->quark, &task_flags,
side, trans,
tempmm, B.nb, tempmm, tempNRDn,
tempkm, ib, T.nb,
B (m, N ), ldbm,
B (m, N+RD), ldbm,
A (k, N+RD), ldak,
T2(k, N+RD), T.mb);
}
}
}
}
}
}
}

Here is the call graph for this function: