mirror of
https://github.com/amd/blis.git
synced 2026-05-11 17:50:00 +00:00
bb6df2814f
Details: - Defined a new level-1d operation called 'shiftd', including object and typed APIs. This operation adds a scalar value to every element along an arbitrary diagonal of a matrix. Currently, shiftd is implemented in terms of the addv kernel. (The scalar is passed in as the x vector with an increment of zero.) - Replaced ad-hoc usage of setd and addd (after creating a temporary matrix object) with use of shiftd, which is much more concise, in various test driver files in the testsuite. Similar changes were made to the standalone test drivers and the example code. - Added documentation entries in BLISObjectAPI.md and BLISTypedAPI.md for bli_shiftd() and bli_?shiftd(), respectively. - Added observed object properties to level-1d documentation in BLISObjectAPI.md.
304 lines
7.4 KiB
C
304 lines
7.4 KiB
C
/*
|
|
|
|
BLIS
|
|
An object-based framework for developing high-performance BLAS-like
|
|
libraries.
|
|
|
|
Copyright (C) 2014, The University of Texas at Austin
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions are
|
|
met:
|
|
- Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
- Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in the
|
|
documentation and/or other materials provided with the distribution.
|
|
- Neither the name of The University of Texas at Austin nor the names
|
|
of its contributors may be used to endorse or promote products
|
|
derived from this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
#include <unistd.h>
|
|
#include "blis.h"
|
|
|
|
|
|
//#define PRINT
|
|
|
|
int main( int argc, char** argv )
|
|
{
|
|
obj_t a, c;
|
|
obj_t c_save;
|
|
obj_t alpha;
|
|
dim_t m, n;
|
|
dim_t p;
|
|
dim_t p_begin, p_end, p_inc;
|
|
int m_input, n_input;
|
|
num_t dt;
|
|
int r, n_repeats;
|
|
side_t side;
|
|
uplo_t uploa;
|
|
trans_t transa;
|
|
diag_t diaga;
|
|
f77_char f77_side;
|
|
f77_char f77_uploa;
|
|
f77_char f77_transa;
|
|
f77_char f77_diaga;
|
|
|
|
double dtime;
|
|
double dtime_save;
|
|
double gflops;
|
|
|
|
//bli_init();
|
|
|
|
//bli_error_checking_level_set( BLIS_NO_ERROR_CHECKING );
|
|
|
|
n_repeats = 3;
|
|
|
|
#ifndef PRINT
|
|
p_begin = 200;
|
|
p_end = 2000;
|
|
p_inc = 200;
|
|
|
|
m_input = -1;
|
|
n_input = -1;
|
|
#else
|
|
p_begin = 16;
|
|
p_end = 16;
|
|
p_inc = 1;
|
|
|
|
m_input = 4;
|
|
n_input = 4;
|
|
#endif
|
|
|
|
#if 1
|
|
//dt = BLIS_FLOAT;
|
|
dt = BLIS_DOUBLE;
|
|
#else
|
|
//dt = BLIS_SCOMPLEX;
|
|
dt = BLIS_DCOMPLEX;
|
|
#endif
|
|
|
|
side = BLIS_LEFT;
|
|
//side = BLIS_RIGHT;
|
|
|
|
uploa = BLIS_LOWER;
|
|
//uploa = BLIS_UPPER;
|
|
|
|
transa = BLIS_NO_TRANSPOSE;
|
|
|
|
diaga = BLIS_NONUNIT_DIAG;
|
|
|
|
bli_param_map_blis_to_netlib_side( side, &f77_side );
|
|
bli_param_map_blis_to_netlib_uplo( uploa, &f77_uploa );
|
|
bli_param_map_blis_to_netlib_trans( transa, &f77_transa );
|
|
bli_param_map_blis_to_netlib_diag( diaga, &f77_diaga );
|
|
|
|
// Begin with initializing the last entry to zero so that
|
|
// matlab allocates space for the entire array once up-front.
|
|
for ( p = p_begin; p + p_inc <= p_end; p += p_inc ) ;
|
|
#ifdef BLIS
|
|
printf( "data_trsm_blis" );
|
|
#else
|
|
printf( "data_trsm_%s", BLAS );
|
|
#endif
|
|
printf( "( %2lu, 1:3 ) = [ %4lu %4lu %7.2f ];\n",
|
|
( unsigned long )(p - p_begin + 1)/p_inc + 1,
|
|
( unsigned long )0,
|
|
( unsigned long )0, 0.0 );
|
|
|
|
for ( p = p_begin; p <= p_end; p += p_inc )
|
|
{
|
|
if ( m_input < 0 ) m = p * ( dim_t )abs(m_input);
|
|
else m = ( dim_t ) m_input;
|
|
if ( n_input < 0 ) n = p * ( dim_t )abs(n_input);
|
|
else n = ( dim_t ) n_input;
|
|
|
|
bli_obj_create( dt, 1, 1, 0, 0, &alpha );
|
|
|
|
if ( bli_is_left( side ) )
|
|
bli_obj_create( dt, m, m, 0, 0, &a );
|
|
else
|
|
bli_obj_create( dt, n, n, 0, 0, &a );
|
|
bli_obj_create( dt, m, n, 0, 0, &c );
|
|
bli_obj_create( dt, m, n, 0, 0, &c_save );
|
|
|
|
bli_randm( &a );
|
|
bli_randm( &c );
|
|
|
|
bli_obj_set_struc( BLIS_TRIANGULAR, &a );
|
|
bli_obj_set_uplo( uploa, &a );
|
|
bli_obj_set_conjtrans( transa, &a );
|
|
bli_obj_set_diag( diaga, &a );
|
|
|
|
// Randomize A and zero the unstored triangle to ensure the
|
|
// implementation reads only from the stored region.
|
|
bli_randm( &a );
|
|
bli_mktrim( &a );
|
|
|
|
// Load the diagonal of A to make it more likely to be invertible.
|
|
bli_shiftd( &BLIS_TWO, &a );
|
|
|
|
bli_setsc( (2.0/1.0), 1.0, &alpha );
|
|
|
|
|
|
bli_copym( &c, &c_save );
|
|
|
|
dtime_save = DBL_MAX;
|
|
|
|
for ( r = 0; r < n_repeats; ++r )
|
|
{
|
|
bli_copym( &c_save, &c );
|
|
|
|
|
|
dtime = bli_clock();
|
|
|
|
|
|
#ifdef PRINT
|
|
bli_invertd( &a );
|
|
bli_printm( "a", &a, "%4.1f", "" );
|
|
bli_invertd( &a );
|
|
bli_printm( "c", &c, "%4.1f", "" );
|
|
#endif
|
|
|
|
#ifdef BLIS
|
|
|
|
bli_trsm( side,
|
|
&alpha,
|
|
&a,
|
|
&c );
|
|
#else
|
|
|
|
if ( bli_is_float( dt ) )
|
|
{
|
|
f77_int mm = bli_obj_length( &c );
|
|
f77_int nn = bli_obj_width( &c );
|
|
f77_int lda = bli_obj_col_stride( &a );
|
|
f77_int ldc = bli_obj_col_stride( &c );
|
|
float* alphap = bli_obj_buffer( &alpha );
|
|
float* ap = bli_obj_buffer( &a );
|
|
float* cp = bli_obj_buffer( &c );
|
|
|
|
strsm_( &f77_side,
|
|
&f77_uploa,
|
|
&f77_transa,
|
|
&f77_diaga,
|
|
&mm,
|
|
&nn,
|
|
alphap,
|
|
ap, &lda,
|
|
cp, &ldc );
|
|
}
|
|
else if ( bli_is_double( dt ) )
|
|
{
|
|
f77_int mm = bli_obj_length( &c );
|
|
f77_int nn = bli_obj_width( &c );
|
|
f77_int lda = bli_obj_col_stride( &a );
|
|
f77_int ldc = bli_obj_col_stride( &c );
|
|
double* alphap = bli_obj_buffer( &alpha );
|
|
double* ap = bli_obj_buffer( &a );
|
|
double* cp = bli_obj_buffer( &c );
|
|
|
|
dtrsm_( &f77_side,
|
|
&f77_uploa,
|
|
&f77_transa,
|
|
&f77_diaga,
|
|
&mm,
|
|
&nn,
|
|
alphap,
|
|
ap, &lda,
|
|
cp, &ldc );
|
|
}
|
|
else if ( bli_is_scomplex( dt ) )
|
|
{
|
|
f77_int mm = bli_obj_length( &c );
|
|
f77_int nn = bli_obj_width( &c );
|
|
f77_int lda = bli_obj_col_stride( &a );
|
|
f77_int ldc = bli_obj_col_stride( &c );
|
|
scomplex* alphap = bli_obj_buffer( &alpha );
|
|
scomplex* ap = bli_obj_buffer( &a );
|
|
scomplex* cp = bli_obj_buffer( &c );
|
|
|
|
ctrsm_( &f77_side,
|
|
&f77_uploa,
|
|
&f77_transa,
|
|
&f77_diaga,
|
|
&mm,
|
|
&nn,
|
|
alphap,
|
|
ap, &lda,
|
|
cp, &ldc );
|
|
}
|
|
else if ( bli_is_dcomplex( dt ) )
|
|
{
|
|
f77_int mm = bli_obj_length( &c );
|
|
f77_int nn = bli_obj_width( &c );
|
|
f77_int lda = bli_obj_col_stride( &a );
|
|
f77_int ldc = bli_obj_col_stride( &c );
|
|
dcomplex* alphap = bli_obj_buffer( &alpha );
|
|
dcomplex* ap = bli_obj_buffer( &a );
|
|
dcomplex* cp = bli_obj_buffer( &c );
|
|
|
|
ztrsm_( &f77_side,
|
|
&f77_uploa,
|
|
&f77_transa,
|
|
&f77_diaga,
|
|
&mm,
|
|
&nn,
|
|
alphap,
|
|
ap, &lda,
|
|
cp, &ldc );
|
|
}
|
|
#endif
|
|
|
|
#ifdef PRINT
|
|
bli_printm( "c after", &c, "%9.5f", "" );
|
|
exit(1);
|
|
#endif
|
|
|
|
dtime_save = bli_clock_min_diff( dtime_save, dtime );
|
|
}
|
|
|
|
if ( bli_is_left( side ) )
|
|
gflops = ( 1.0 * m * m * n ) / ( dtime_save * 1.0e9 );
|
|
else
|
|
gflops = ( 1.0 * m * n * n ) / ( dtime_save * 1.0e9 );
|
|
|
|
if ( bli_is_complex( dt ) ) gflops *= 4.0;
|
|
|
|
#ifdef BLIS
|
|
printf( "data_trsm_blis" );
|
|
#else
|
|
printf( "data_trsm_%s", BLAS );
|
|
#endif
|
|
printf( "( %2lu, 1:3 ) = [ %4lu %4lu %7.2f ];\n",
|
|
( unsigned long )(p - p_begin + 1)/p_inc + 1,
|
|
( unsigned long )m,
|
|
( unsigned long )n, gflops );
|
|
|
|
bli_obj_free( &alpha );
|
|
|
|
bli_obj_free( &a );
|
|
bli_obj_free( &c );
|
|
bli_obj_free( &c_save );
|
|
}
|
|
|
|
//bli_finalize();
|
|
|
|
return 0;
|
|
}
|
|
|