mirror of
https://github.com/amd/blis.git
synced 2026-04-29 12:01:12 +00:00
5fec95b99f
Details: - Implemented support for gemm where A, B, and C may have different storage datatypes, as well as a computational precision (and implied computation domain) that may be different from the storage precision of either A or B. This results in 128 different combinations, all which are implemented within this commit. (For now, the mixed-datatype functionality is only supported via the object API.) If desired, the mixed-datatype support may be disabled at configure-time. - Added a memory-intensive optimization to certain mixed-datatype cases that requires a single m-by-n matrix be allocated (temporarily) per call to gemm. This optimization aims to avoid the overhead involved in repeatedly updating C with general stride, or updating C after a typecast from the computation precision. This memory optimization may be disabled at configure-time (provided that the mixed-datatype support is enabled in the first place). - Added support for testing mixed-datatype combinations to testsuite. The user may test gemm with mixed domains, precisions, both, or neither. - Added a standalone test driver directory for building and running mixed-datatype performance experiments. - Defined a new variation of castm, castnzm, which operates like castm except that imaginary values are not touched when casting a real operand to a complex operand. (By contrast, in these situations castm sets the imaginary components of the destination matrix to zero.) - Defined bli_obj_imag_is_zero() and substituted calls in lieu of all usages of bli_obj_imag_equals() that tested against BLIS_ZERO, and also simplified the implementation of bli_obj_imag_equals(). - Fixed bad behavior from bli_obj_is_real() and bli_obj_is_complex() when given BLIS_CONSTANT objects. - Disabled dt_on_output field in auxinfo_t structure as well as all accessor functions. Also commented out all usage of accessor functions within macrokernels. (Typecasting in the microkernel is still feasible, though probably unrealistic for now given the additional complexity required.) - Use void function pointer type (instead of void*) for storing function pointers in bli_l0_fpa.c. - Added documentation for using gemm with mixed datatypes in docs/MixedDatatypes.md and example code in examples/oapi/11gemm_md.c. - Defined level-1d operation xpbyd and level-1m operation xpbym. - Added xpbym test module to testsuite. - Updated frame/include/bli_x86_asm_macros.h with additional macros (courtsey of Devin Matthews).
382 lines
9.8 KiB
C
382 lines
9.8 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.
|
|
|
|
*/
|
|
|
|
// Guard the function definitions so that they are only compiled when
|
|
// #included from files that define the object API macros.
|
|
#ifdef BLIS_ENABLE_OAPI
|
|
|
|
//
|
|
// Define object-based interfaces.
|
|
//
|
|
|
|
#undef GENFRONT
|
|
#define GENFRONT( opname ) \
|
|
\
|
|
void PASTEMAC(opname,EX_SUF) \
|
|
( \
|
|
obj_t* x, \
|
|
obj_t* y \
|
|
BLIS_OAPI_EX_PARAMS \
|
|
) \
|
|
{ \
|
|
bli_init_once(); \
|
|
\
|
|
BLIS_OAPI_EX_DECLS \
|
|
\
|
|
num_t dt = bli_obj_dt( x ); \
|
|
\
|
|
doff_t diagoffx = bli_obj_diag_offset( x ); \
|
|
diag_t diagx = bli_obj_diag( x ); \
|
|
trans_t transx = bli_obj_conjtrans_status( x ); \
|
|
dim_t m = bli_obj_length( y ); \
|
|
dim_t n = bli_obj_width( y ); \
|
|
void* buf_x = bli_obj_buffer_at_off( x ); \
|
|
inc_t rs_x = bli_obj_row_stride( x ); \
|
|
inc_t cs_x = bli_obj_col_stride( x ); \
|
|
void* buf_y = bli_obj_buffer_at_off( y ); \
|
|
inc_t rs_y = bli_obj_row_stride( y ); \
|
|
inc_t cs_y = bli_obj_col_stride( y ); \
|
|
\
|
|
if ( bli_error_checking_is_enabled() ) \
|
|
PASTEMAC(opname,_check)( x, y ); \
|
|
\
|
|
/* Query a type-specific function pointer, except one that uses
|
|
void* instead of typed pointers. */ \
|
|
PASTECH2(opname,BLIS_TAPI_EX_SUF,_vft) f = \
|
|
PASTEMAC2(opname,BLIS_TAPI_EX_SUF,_qfp)( dt ); \
|
|
\
|
|
f \
|
|
( \
|
|
diagoffx, \
|
|
diagx, \
|
|
transx, \
|
|
m, \
|
|
n, \
|
|
buf_x, rs_x, cs_x, \
|
|
buf_y, rs_y, cs_y, \
|
|
cntx, \
|
|
rntm \
|
|
); \
|
|
}
|
|
|
|
GENFRONT( addd )
|
|
GENFRONT( copyd )
|
|
GENFRONT( subd )
|
|
|
|
|
|
#undef GENFRONT
|
|
#define GENFRONT( opname ) \
|
|
\
|
|
void PASTEMAC(opname,EX_SUF) \
|
|
( \
|
|
obj_t* alpha, \
|
|
obj_t* x, \
|
|
obj_t* y \
|
|
BLIS_OAPI_EX_PARAMS \
|
|
) \
|
|
{ \
|
|
bli_init_once(); \
|
|
\
|
|
BLIS_OAPI_EX_DECLS \
|
|
\
|
|
num_t dt = bli_obj_dt( x ); \
|
|
\
|
|
doff_t diagoffx = bli_obj_diag_offset( x ); \
|
|
diag_t diagx = bli_obj_diag( x ); \
|
|
trans_t transx = bli_obj_conjtrans_status( x ); \
|
|
dim_t m = bli_obj_length( y ); \
|
|
dim_t n = bli_obj_width( y ); \
|
|
void* buf_x = bli_obj_buffer_at_off( x ); \
|
|
inc_t rs_x = bli_obj_row_stride( x ); \
|
|
inc_t cs_x = bli_obj_col_stride( x ); \
|
|
void* buf_y = bli_obj_buffer_at_off( y ); \
|
|
inc_t rs_y = bli_obj_row_stride( y ); \
|
|
inc_t cs_y = bli_obj_col_stride( y ); \
|
|
\
|
|
void* buf_alpha; \
|
|
\
|
|
obj_t alpha_local; \
|
|
\
|
|
if ( bli_error_checking_is_enabled() ) \
|
|
PASTEMAC(opname,_check)( alpha, x, y ); \
|
|
\
|
|
/* Create local copy-casts of scalars (and apply internal conjugation
|
|
as needed). */ \
|
|
bli_obj_scalar_init_detached_copy_of( dt, BLIS_NO_CONJUGATE, \
|
|
alpha, &alpha_local ); \
|
|
buf_alpha = bli_obj_buffer_for_1x1( dt, &alpha_local ); \
|
|
\
|
|
/* Query a type-specific function pointer, except one that uses
|
|
void* instead of typed pointers. */ \
|
|
PASTECH2(opname,BLIS_TAPI_EX_SUF,_vft) f = \
|
|
PASTEMAC2(opname,BLIS_TAPI_EX_SUF,_qfp)( dt ); \
|
|
\
|
|
f \
|
|
( \
|
|
diagoffx, \
|
|
diagx, \
|
|
transx, \
|
|
m, \
|
|
n, \
|
|
buf_alpha, \
|
|
buf_x, rs_x, cs_x, \
|
|
buf_y, rs_y, cs_y, \
|
|
cntx, \
|
|
rntm \
|
|
); \
|
|
}
|
|
|
|
GENFRONT( axpyd )
|
|
GENFRONT( scal2d )
|
|
|
|
|
|
#undef GENFRONT
|
|
#define GENFRONT( opname ) \
|
|
\
|
|
void PASTEMAC(opname,EX_SUF) \
|
|
( \
|
|
obj_t* x \
|
|
BLIS_OAPI_EX_PARAMS \
|
|
) \
|
|
{ \
|
|
bli_init_once(); \
|
|
\
|
|
BLIS_OAPI_EX_DECLS \
|
|
\
|
|
num_t dt = bli_obj_dt( x ); \
|
|
\
|
|
doff_t diagoffx = bli_obj_diag_offset( x ); \
|
|
dim_t m = bli_obj_length( x ); \
|
|
dim_t n = bli_obj_width( x ); \
|
|
void* buf_x = bli_obj_buffer_at_off( x ); \
|
|
inc_t rs_x = bli_obj_row_stride( x ); \
|
|
inc_t cs_x = bli_obj_col_stride( x ); \
|
|
\
|
|
if ( bli_error_checking_is_enabled() ) \
|
|
PASTEMAC(opname,_check)( x ); \
|
|
\
|
|
/* Query a type-specific function pointer, except one that uses
|
|
void* instead of typed pointers. */ \
|
|
PASTECH2(opname,BLIS_TAPI_EX_SUF,_vft) f = \
|
|
PASTEMAC2(opname,BLIS_TAPI_EX_SUF,_qfp)( dt ); \
|
|
\
|
|
f \
|
|
( \
|
|
diagoffx, \
|
|
m, \
|
|
n, \
|
|
buf_x, rs_x, cs_x, \
|
|
cntx, \
|
|
rntm \
|
|
); \
|
|
}
|
|
|
|
GENFRONT( invertd )
|
|
|
|
|
|
#undef GENFRONT
|
|
#define GENFRONT( opname ) \
|
|
\
|
|
void PASTEMAC(opname,EX_SUF) \
|
|
( \
|
|
obj_t* alpha, \
|
|
obj_t* x \
|
|
BLIS_OAPI_EX_PARAMS \
|
|
) \
|
|
{ \
|
|
bli_init_once(); \
|
|
\
|
|
BLIS_OAPI_EX_DECLS \
|
|
\
|
|
num_t dt = bli_obj_dt( x ); \
|
|
\
|
|
/* conj_t conjalpha = bli_obj_conj_status( alpha ); */ \
|
|
doff_t diagoffx = bli_obj_diag_offset( x ); \
|
|
dim_t m = bli_obj_length( x ); \
|
|
dim_t n = bli_obj_width( x ); \
|
|
void* buf_x = bli_obj_buffer_at_off( x ); \
|
|
inc_t rs_x = bli_obj_row_stride( x ); \
|
|
inc_t cs_x = bli_obj_col_stride( x ); \
|
|
\
|
|
void* buf_alpha; \
|
|
\
|
|
obj_t alpha_local; \
|
|
\
|
|
if ( bli_error_checking_is_enabled() ) \
|
|
PASTEMAC(opname,_check)( alpha, x ); \
|
|
\
|
|
/* Create local copy-casts of scalars (and apply internal conjugation
|
|
as needed). */ \
|
|
bli_obj_scalar_init_detached_copy_of( dt, BLIS_NO_CONJUGATE, \
|
|
alpha, &alpha_local ); \
|
|
buf_alpha = bli_obj_buffer_for_1x1( dt, &alpha_local ); \
|
|
\
|
|
/* Query a type-specific function pointer, except one that uses
|
|
void* instead of typed pointers. */ \
|
|
PASTECH2(opname,BLIS_TAPI_EX_SUF,_vft) f = \
|
|
PASTEMAC2(opname,BLIS_TAPI_EX_SUF,_qfp)( dt ); \
|
|
\
|
|
f \
|
|
( \
|
|
BLIS_NO_CONJUGATE, /* internal conjugation applied during copy-cast. */ \
|
|
diagoffx, \
|
|
m, \
|
|
n, \
|
|
buf_alpha, \
|
|
buf_x, rs_x, cs_x, \
|
|
cntx, \
|
|
rntm \
|
|
); \
|
|
}
|
|
|
|
GENFRONT( scald )
|
|
GENFRONT( setd )
|
|
|
|
|
|
#undef GENFRONT
|
|
#define GENFRONT( opname ) \
|
|
\
|
|
void PASTEMAC(opname,EX_SUF) \
|
|
( \
|
|
obj_t* alpha, \
|
|
obj_t* x \
|
|
BLIS_OAPI_EX_PARAMS \
|
|
) \
|
|
{ \
|
|
bli_init_once(); \
|
|
\
|
|
BLIS_OAPI_EX_DECLS \
|
|
\
|
|
num_t dt = bli_obj_dt( x ); \
|
|
\
|
|
doff_t diagoffx = bli_obj_diag_offset( x ); \
|
|
dim_t m = bli_obj_length( x ); \
|
|
dim_t n = bli_obj_width( x ); \
|
|
void* buf_x = bli_obj_buffer_at_off( x ); \
|
|
inc_t rs_x = bli_obj_row_stride( x ); \
|
|
inc_t cs_x = bli_obj_col_stride( x ); \
|
|
\
|
|
void* buf_alpha = bli_obj_buffer_for_1x1( dt, alpha ); \
|
|
\
|
|
if ( bli_error_checking_is_enabled() ) \
|
|
PASTEMAC(opname,_check)( alpha, x ); \
|
|
\
|
|
/* Query a type-specific function pointer, except one that uses
|
|
void* instead of typed pointers. */ \
|
|
PASTECH2(opname,BLIS_TAPI_EX_SUF,_vft) f = \
|
|
PASTEMAC2(opname,BLIS_TAPI_EX_SUF,_qfp)( dt ); \
|
|
\
|
|
f \
|
|
( \
|
|
diagoffx, \
|
|
m, \
|
|
n, \
|
|
buf_alpha, \
|
|
buf_x, rs_x, cs_x, \
|
|
cntx, \
|
|
rntm \
|
|
); \
|
|
}
|
|
|
|
GENFRONT( setid )
|
|
|
|
|
|
#undef GENFRONT
|
|
#define GENFRONT( opname ) \
|
|
\
|
|
void PASTEMAC(opname,EX_SUF) \
|
|
( \
|
|
obj_t* x, \
|
|
obj_t* beta, \
|
|
obj_t* y \
|
|
BLIS_OAPI_EX_PARAMS \
|
|
) \
|
|
{ \
|
|
bli_init_once(); \
|
|
\
|
|
BLIS_OAPI_EX_DECLS \
|
|
\
|
|
num_t dt = bli_obj_dt( x ); \
|
|
\
|
|
doff_t diagoffx = bli_obj_diag_offset( x ); \
|
|
diag_t diagx = bli_obj_diag( x ); \
|
|
trans_t transx = bli_obj_conjtrans_status( x ); \
|
|
dim_t m = bli_obj_length( y ); \
|
|
dim_t n = bli_obj_width( y ); \
|
|
void* buf_x = bli_obj_buffer_at_off( x ); \
|
|
inc_t rs_x = bli_obj_row_stride( x ); \
|
|
inc_t cs_x = bli_obj_col_stride( x ); \
|
|
void* buf_y = bli_obj_buffer_at_off( y ); \
|
|
inc_t rs_y = bli_obj_row_stride( y ); \
|
|
inc_t cs_y = bli_obj_col_stride( y ); \
|
|
\
|
|
void* buf_beta; \
|
|
\
|
|
obj_t beta_local; \
|
|
\
|
|
if ( bli_error_checking_is_enabled() ) \
|
|
PASTEMAC(opname,_check)( x, beta, y ); \
|
|
\
|
|
/* Create local copy-casts of scalars (and apply internal conjugation
|
|
as needed). */ \
|
|
bli_obj_scalar_init_detached_copy_of( dt, BLIS_NO_CONJUGATE, \
|
|
beta, &beta_local ); \
|
|
buf_beta = bli_obj_buffer_for_1x1( dt, &beta_local ); \
|
|
\
|
|
/* Query a type-specific function pointer, except one that uses
|
|
void* instead of typed pointers. */ \
|
|
PASTECH2(opname,BLIS_TAPI_EX_SUF,_vft) f = \
|
|
PASTEMAC2(opname,BLIS_TAPI_EX_SUF,_qfp)( dt ); \
|
|
\
|
|
f \
|
|
( \
|
|
diagoffx, \
|
|
diagx, \
|
|
transx, \
|
|
m, \
|
|
n, \
|
|
buf_x, rs_x, cs_x, \
|
|
buf_beta, \
|
|
buf_y, rs_y, cs_y, \
|
|
cntx, \
|
|
rntm \
|
|
); \
|
|
}
|
|
|
|
GENFRONT( xpbyd )
|
|
|
|
|
|
#endif
|
|
|