lapacke_stprfb.c File Reference

#include "lapacke_utils.h"

Go to the source code of this file.

Functions
lapack_int	LAPACKE_stprfb (int matrix_layout, char side, char trans, char direct, char storev, lapack_int m, lapack_int n, lapack_int k, lapack_int l, const float *v, lapack_int ldv, const float *t, lapack_int ldt, float *a, lapack_int lda, float *b, lapack_int ldb)

Function Documentation

LAPACKE_stprfb()

lapack_int LAPACKE_stprfb	(	int	matrix_layout,
		char	side,
		char	trans,
		char	direct,
		char	storev,
		lapack_int	m,
		lapack_int	n,
		lapack_int	k,
		lapack_int	l,
		const float *	v,
		lapack_int	ldv,
		const float *	t,
		lapack_int	ldt,
		float *	a,
		lapack_int	lda,
		float *	b,
		lapack_int	ldb )

Definition at line 35 of file lapacke_stprfb.c.

{
    lapack_int ncols_v, nrows_v, ncols_a, nrows_a;
    lapack_int info = 0;
    lapack_int ldwork;
    lapack_int work_size;
    float* work = NULL;
    if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) {
        LAPACKE_xerbla( "LAPACKE_stprfb", -1 );
        return -1;
    }
#ifndef LAPACK_DISABLE_NAN_CHECK
    if( LAPACKE_get_nancheck() ) {
        /* Optionally check input matrices for NaNs
         * V is m-by-k (left,  columnwise)
         *   or n-by-k (right, columnwise)
         *   or k-by-m (left,  rowwise)
         *   or k-by-n (right, rowwise)
         * T is k-by-k
         * A is k-by-n (left)
         *   or m-by-k (right)
         * B is m-by-n
         */
        if( LAPACKE_lsame( storev, 'C' ) ) {
            ncols_v = k;
            nrows_v = LAPACKE_lsame( side, 'L' ) ? m :
                      LAPACKE_lsame( side, 'R' ) ? n : 0;
        } else if( LAPACKE_lsame( storev, 'R' ) ) {
            ncols_v = LAPACKE_lsame( side, 'L' ) ? m :
                      LAPACKE_lsame( side, 'R' ) ? n : 0;
            nrows_v = k;
        } else {
            ncols_v = 0;
            nrows_v = 0;
        }
        nrows_a = LAPACKE_lsame( side, 'L' ) ? k :
                  LAPACKE_lsame( side, 'R' ) ? m : 0;
        ncols_a = LAPACKE_lsame( side, 'L' ) ? n :
                  LAPACKE_lsame( side, 'R' ) ? k : 0;
        if( LAPACKE_sge_nancheck( matrix_layout, ncols_a, nrows_a, a, lda ) ) {
            return -14;
        }
        if( LAPACKE_sge_nancheck( matrix_layout, m, n, b, ldb ) ) {
            return -16;
        }
        if( LAPACKE_sge_nancheck( matrix_layout, k, k, t, ldt ) ) {
            return -12;
        }
        if( LAPACKE_sge_nancheck( matrix_layout, nrows_v, ncols_v, v, ldv ) ) {
            return -10;
        }
    }
#endif
    if (side=='l' ||  side=='L') {
        ldwork = k;
        work_size = MAX(1,ldwork) * MAX(1,n);
    }
    else {
        ldwork = m;
        work_size = MAX(1,ldwork) * MAX(1,k);
    }
    /* Allocate memory for working array(s) */
    work = (float*)
        LAPACKE_malloc( sizeof(float) * work_size );
    if( work == NULL ) {
        info = LAPACK_WORK_MEMORY_ERROR;
        goto exit_level_0;
    }
    /* Call middle-level interface */
    info = LAPACKE_stprfb_work( matrix_layout, side, trans, direct, storev, m, n,
                                k, l, v, ldv, t, ldt, a, lda, b, ldb, work,
                                ldwork );
    /* Release memory and exit */
    LAPACKE_free( work );
exit_level_0:
    if( info == LAPACK_WORK_MEMORY_ERROR ) {
        LAPACKE_xerbla( "LAPACKE_stprfb", info );
    }
    return info;
}