libflame revision_anchor
Functions
bli_setmr.c File Reference

(r)

Functions

void bli_ssetmr (char uplo, int m, int n, float *sigma, float *a, int a_rs, int a_cs)
void bli_dsetmr (char uplo, int m, int n, double *sigma, double *a, int a_rs, int a_cs)
void bli_csetmr (char uplo, int m, int n, scomplex *sigma, scomplex *a, int a_rs, int a_cs)
void bli_zsetmr (char uplo, int m, int n, dcomplex *sigma, dcomplex *a, int a_rs, int a_cs)

Function Documentation

void bli_csetmr ( char  uplo,
int  m,
int  n,
scomplex sigma,
scomplex a,
int  a_rs,
int  a_cs 
)

References bli_csetv(), bli_is_row_storage(), bli_is_upper(), and bli_zero_dim2().

Referenced by FLA_Triangularize().

{
    scomplex* a_begin;
    int       lda, inca;
    int       n_iter;
    int       n_elem_max;
    int       n_elem;
    int       j;

    // Return early if possible.
    if ( bli_zero_dim2( m, n ) ) return;

    // Initialize with optimal values for column-major storage.
    n_iter     = n;
    n_elem_max = m;
    lda        = a_cs;
    inca       = a_rs;

    // An optimization: if A is row-major, then let's access the matrix by
    // rows instead of by columns to increase spatial locality.
    if ( bli_is_row_storage( a_rs, a_cs ) )
    {
        bli_swap_ints( n_iter, n_elem_max );
        bli_swap_ints( lda, inca );
        bli_toggle_uplo( uplo );
    }
    
    if ( bli_is_upper( uplo ) )
    {
        for ( j = 0; j < n_iter; j++ )
        {
            n_elem  = bli_min( j, n_elem_max );
            a_begin = a + j*lda;

            bli_csetv( n_elem,
                       sigma,
                       a_begin, inca );
        }
    }
    else // if ( bli_is_lower( uplo ) )
    {
        for ( j = 0; j < n_iter; j++ )
        {
            n_elem  = bli_max( 0, n_elem_max - j - 1 );
            a_begin = a + j*lda + (j + 1)*inca;

            bli_csetv( n_elem,
                       sigma,
                       a_begin, inca );
        }
    }
}
void bli_dsetmr ( char  uplo,
int  m,
int  n,
double *  sigma,
double *  a,
int  a_rs,
int  a_cs 
)

References bli_dsetv(), bli_is_row_storage(), bli_is_upper(), and bli_zero_dim2().

Referenced by FLA_Triangularize().

{
    double*   a_begin;
    int       lda, inca;
    int       n_iter;
    int       n_elem_max;
    int       n_elem;
    int       j;

    // Return early if possible.
    if ( bli_zero_dim2( m, n ) ) return;

    // Initialize with optimal values for column-major storage.
    n_iter     = n;
    n_elem_max = m;
    lda        = a_cs;
    inca       = a_rs;

    // An optimization: if A is row-major, then let's access the matrix by
    // rows instead of by columns to increase spatial locality.
    if ( bli_is_row_storage( a_rs, a_cs ) )
    {
        bli_swap_ints( n_iter, n_elem_max );
        bli_swap_ints( lda, inca );
        bli_toggle_uplo( uplo );
    }
    
    if ( bli_is_upper( uplo ) )
    {
        for ( j = 0; j < n_iter; j++ )
        {
            n_elem  = bli_min( j, n_elem_max );
            a_begin = a + j*lda;

            bli_dsetv( n_elem,
                       sigma,
                       a_begin, inca );
        }
    }
    else // if ( bli_is_lower( uplo ) )
    {
        for ( j = 0; j < n_iter; j++ )
        {
            n_elem  = bli_max( 0, n_elem_max - j - 1 );
            a_begin = a + j*lda + (j + 1)*inca;

            bli_dsetv( n_elem,
                       sigma,
                       a_begin, inca );
        }
    }
}
void bli_ssetmr ( char  uplo,
int  m,
int  n,
float *  sigma,
float *  a,
int  a_rs,
int  a_cs 
)

References bli_is_row_storage(), bli_is_upper(), bli_ssetv(), and bli_zero_dim2().

Referenced by FLA_Triangularize().

{
    float*    a_begin;
    int       lda, inca;
    int       n_iter;
    int       n_elem_max;
    int       n_elem;
    int       j;

    // Return early if possible.
    if ( bli_zero_dim2( m, n ) ) return;

    // Initialize with optimal values for column-major storage.
    n_iter     = n;
    n_elem_max = m;
    lda        = a_cs;
    inca       = a_rs;

    // An optimization: if A is row-major, then let's access the matrix by
    // rows instead of by columns to increase spatial locality.
    if ( bli_is_row_storage( a_rs, a_cs ) )
    {
        bli_swap_ints( n_iter, n_elem_max );
        bli_swap_ints( lda, inca );
        bli_toggle_uplo( uplo );
    }
    
    if ( bli_is_upper( uplo ) )
    {
        for ( j = 0; j < n_iter; j++ )
        {
            n_elem  = bli_min( j, n_elem_max );
            a_begin = a + j*lda;

            bli_ssetv( n_elem,
                       sigma,
                       a_begin, inca );
        }
    }
    else // if ( bli_is_lower( uplo ) )
    {
        for ( j = 0; j < n_iter; j++ )
        {
            n_elem  = bli_max( 0, n_elem_max - j - 1 );
            a_begin = a + j*lda + (j + 1)*inca;

            bli_ssetv( n_elem,
                       sigma,
                       a_begin, inca );
        }
    }
}
void bli_zsetmr ( char  uplo,
int  m,
int  n,
dcomplex sigma,
dcomplex a,
int  a_rs,
int  a_cs 
)

References bli_is_row_storage(), bli_is_upper(), bli_zero_dim2(), and bli_zsetv().

Referenced by FLA_Triangularize().

{
    dcomplex* a_begin;
    int       lda, inca;
    int       n_iter;
    int       n_elem_max;
    int       n_elem;
    int       j;

    // Return early if possible.
    if ( bli_zero_dim2( m, n ) ) return;

    // Initialize with optimal values for column-major storage.
    n_iter     = n;
    n_elem_max = m;
    lda        = a_cs;
    inca       = a_rs;

    // An optimization: if A is row-major, then let's access the matrix by
    // rows instead of by columns to increase spatial locality.
    if ( bli_is_row_storage( a_rs, a_cs ) )
    {
        bli_swap_ints( n_iter, n_elem_max );
        bli_swap_ints( lda, inca );
        bli_toggle_uplo( uplo );
    }
    
    if ( bli_is_upper( uplo ) )
    {
        for ( j = 0; j < n_iter; j++ )
        {
            n_elem  = bli_min( j, n_elem_max );
            a_begin = a + j*lda;

            bli_zsetv( n_elem,
                       sigma,
                       a_begin, inca );
        }
    }
    else // if ( bli_is_lower( uplo ) )
    {
        for ( j = 0; j < n_iter; j++ )
        {
            n_elem  = bli_max( 0, n_elem_max - j - 1 );
            a_begin = a + j*lda + (j + 1)*inca;

            bli_zsetv( n_elem,
                       sigma,
                       a_begin, inca );
        }
    }
}