mirror of
https://github.com/amd/blis.git
synced 2026-04-20 15:48:50 +00:00
fde5f1fdec
Details:
- Standard names for reference kernels (levels-1v, -1f and 3) are now
macro constants. Examples:
BLIS_SAXPYV_KERNEL_REF
BLIS_DDOTXF_KERNEL_REF
BLIS_ZGEMM_UKERNEL_REF
- Developers no longer have to name all datatype instances of a kernel
with a common base name; [sdcz] datatype flavors of each kernel or
micro-kernel (level-1v, -1f, or 3) may now be named independently.
This means you can now, if you wish, encode the datatype-specific
register blocksizes in the name of the micro-kernel functions.
- Any datatype instances of any kernel (1v, 1f, or 3) that is left
undefined in bli_kernel.h will default to the corresponding reference
implementation. For example, if BLIS_DGEMM_UKERNEL is left undefined,
it will be defined to be BLIS_DGEMM_UKERNEL_REF.
- Developers no longer need to name level-1v/-1f kernels with multiple
datatype chars to match the number of types the kernel WOULD take in
a mixed type environment, as in bli_dddaxpyv_opt(). Now, one char is
sufficient, as in bli_daxpyv_opt().
- There is no longer a need to define an obj_t wrapper to go along with
your level-1v/-1f kernels. The framework now prvides a _kernel()
function which serves as the obj_t wrapper for whatever kernels are
specified (or defaulted to) via bli_kernel.h
- Developers no longer need to prototype their kernels, and thus no
longer need to include any prototyping headers from within
bli_kernel.h. The framework now generates kernel prototypes, with the
proper type signature, based on the kernel names defined (or defaulted
to) via bli_kernel.h.
- If the complex datatype x (of [cz]) implementation of the gemm micro-
kernel is left undefined by bli_kernel.h, but its same-precision real
domain equivalent IS defined, BLIS will use a 4m-based implementation
for the datatype x implementations of all level-3 operations, using
only the real gemm micro-kernel.
114 lines
3.8 KiB
C
114 lines
3.8 KiB
C
/*
|
|
|
|
BLIS
|
|
An object-based framework for developing high-performance BLAS-like
|
|
libraries.
|
|
|
|
Copyright (C) 2014, The University of Texas
|
|
|
|
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 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 "blis.h"
|
|
|
|
#define FUNCPTR_T setv_fp
|
|
|
|
typedef void (*FUNCPTR_T)(
|
|
dim_t n,
|
|
void* beta,
|
|
void* x, inc_t incx
|
|
);
|
|
|
|
// If some mixed datatype functions will not be compiled, we initialize
|
|
// the corresponding elements of the function array to NULL.
|
|
#ifdef BLIS_ENABLE_MIXED_PRECISION_SUPPORT
|
|
static FUNCPTR_T GENARRAY2_ALL(ftypes,setv_kernel_void);
|
|
#else
|
|
#ifdef BLIS_ENABLE_MIXED_DOMAIN_SUPPORT
|
|
static FUNCPTR_T GENARRAY2_EXT(ftypes,setv_kernel_void);
|
|
#else
|
|
static FUNCPTR_T GENARRAY2_MIN(ftypes,setv_kernel_void);
|
|
#endif
|
|
#endif
|
|
|
|
|
|
void bli_setv_kernel( obj_t* beta,
|
|
obj_t* x )
|
|
{
|
|
num_t dt_x = bli_obj_datatype( *x );
|
|
|
|
dim_t n = bli_obj_vector_dim( *x );
|
|
|
|
void* buf_x = bli_obj_buffer_at_off( *x );
|
|
inc_t inc_x = bli_obj_vector_inc( *x );
|
|
|
|
void* buf_beta;
|
|
num_t dt_beta;
|
|
|
|
FUNCPTR_T f;
|
|
|
|
// If beta is a scalar constant, use dt_x to extract the address of the
|
|
// corresponding constant value; otherwise, use the datatype encoded
|
|
// within the beta object and extract the buffer at the beta offset.
|
|
bli_set_scalar_dt_buffer( beta, dt_x, dt_beta, buf_beta );
|
|
|
|
// Index into the type combination array to extract the correct
|
|
// function pointer.
|
|
f = ftypes[dt_beta][dt_x];
|
|
|
|
// Invoke the function.
|
|
f( n,
|
|
buf_beta,
|
|
buf_x, inc_x );
|
|
}
|
|
|
|
|
|
#undef GENTFUNC2
|
|
#define GENTFUNC2( ctype_b, ctype_x, chb, chx, varname, kername ) \
|
|
\
|
|
void PASTEMAC2(chb,chx,varname)( \
|
|
dim_t n, \
|
|
void* beta, \
|
|
void* x, inc_t incx \
|
|
) \
|
|
{ \
|
|
PASTEMAC2(chb,chx,kername)( n, \
|
|
beta, \
|
|
x, incx ); \
|
|
}
|
|
|
|
// Define the basic set of functions unconditionally, and then also some
|
|
// mixed datatype functions if requested.
|
|
INSERT_GENTFUNC2_BASIC( setv_kernel_void, SETV_KERNEL )
|
|
|
|
#ifdef BLIS_ENABLE_MIXED_DOMAIN_SUPPORT
|
|
INSERT_GENTFUNC2_MIX_D( setv_kernel_void, SETV_KERNEL )
|
|
#endif
|
|
|
|
#ifdef BLIS_ENABLE_MIXED_PRECISION_SUPPORT
|
|
INSERT_GENTFUNC2_MIX_P( setv_kernel_void, SETV_KERNEL )
|
|
#endif
|