mirror of
https://github.com/amd/blis.git
synced 2026-04-19 23:28:52 +00:00
7a0ba4194f
Details: - Implemented a new feature called addons, which are similar to sandboxes except that there is no requirement to define gemm or any other particular operation. - Updated configure to accept --enable-addon=<name> or -a <name> syntax for requesting an addon be included within a BLIS build. configure now outputs the list of enabled addons into config.mk. It also outputs the corresponding #include directives for the addons' headers to a new companion to the bli_config.h header file named bli_addon.h. Because addons may wish to make use of existing BLIS types within their own definitions, the addons' headers must be included sometime after that of bli_config.h (which currently is #included before bli_type_defs.h). This is why the #include directives needed to go into a new top-level header file rather than the existing bli_config.h file. - Added a markdown document, docs/Addons.md, to explain addons, how to build with them, and what assumptions their authors should keep in mind as they create them. - Added a gemmlike-like implementation of sandwich gemm called 'gemmd' as an addon in addon/gemmd. The code uses a 'bao_' prefix for local functions, including the user-level object and typed APIs. - Updated .gitignore so that git ignores bli_addon.h files. Change-Id: Ie7efdea366481ce25075cb2459bdbcfd52309717
124 lines
4.1 KiB
C
124 lines
4.1 KiB
C
/*
|
|
|
|
BLIS
|
|
An object-based framework for developing high-performance BLAS-like
|
|
libraries.
|
|
|
|
Copyright (C) 2021, 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(s) of the copyright holder(s) 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.
|
|
|
|
*/
|
|
|
|
#undef GENTPROT
|
|
#define GENTPROT( ctype, ch, opname ) \
|
|
\
|
|
void PASTECH2(bao_,ch,opname) \
|
|
( \
|
|
dim_t k, \
|
|
dim_t n, \
|
|
dim_t nr, \
|
|
cntx_t* restrict cntx, \
|
|
rntm_t* restrict rntm, \
|
|
mem_t* restrict mem, \
|
|
thrinfo_t* restrict thread \
|
|
); \
|
|
|
|
//INSERT_GENTPROT_BASIC0( packm_init_mem_b )
|
|
GENTPROT( float, s, packm_init_mem_b )
|
|
GENTPROT( double, d, packm_init_mem_b )
|
|
GENTPROT( scomplex, c, packm_init_mem_b )
|
|
GENTPROT( dcomplex, z, packm_init_mem_b )
|
|
|
|
|
|
#undef GENTPROT
|
|
#define GENTPROT( ctype, ch, opname ) \
|
|
\
|
|
void PASTECH2(bao_,ch,opname) \
|
|
( \
|
|
rntm_t* restrict rntm, \
|
|
mem_t* restrict mem, \
|
|
thrinfo_t* restrict thread \
|
|
); \
|
|
|
|
//INSERT_GENTPROT_BASIC0( packm_finalize_mem_b )
|
|
GENTPROT( float, s, packm_finalize_mem_b )
|
|
GENTPROT( double, d, packm_finalize_mem_b )
|
|
GENTPROT( scomplex, c, packm_finalize_mem_b )
|
|
GENTPROT( dcomplex, z, packm_finalize_mem_b )
|
|
|
|
|
|
#undef GENTPROT
|
|
#define GENTPROT( ctype, ch, opname ) \
|
|
\
|
|
void PASTECH2(bao_,ch,opname) \
|
|
( \
|
|
pack_t* restrict schema, \
|
|
dim_t k, \
|
|
dim_t n, \
|
|
dim_t nr, \
|
|
dim_t* restrict k_max, \
|
|
dim_t* restrict n_max, \
|
|
ctype** p, inc_t* restrict rs_p, inc_t* restrict cs_p, \
|
|
dim_t* restrict pd_p, inc_t* restrict ps_p, \
|
|
mem_t* restrict mem \
|
|
); \
|
|
|
|
//INSERT_GENTPROT_BASIC0( packm_init_b )
|
|
GENTPROT( float, s, packm_init_b )
|
|
GENTPROT( double, d, packm_init_b )
|
|
GENTPROT( scomplex, c, packm_init_b )
|
|
GENTPROT( dcomplex, z, packm_init_b )
|
|
|
|
|
|
#undef GENTPROT
|
|
#define GENTPROT( ctype, ch, opname ) \
|
|
\
|
|
void PASTECH2(bao_,ch,opname) \
|
|
( \
|
|
conj_t conj, \
|
|
dim_t k_alloc, \
|
|
dim_t n_alloc, \
|
|
dim_t k, \
|
|
dim_t n, \
|
|
dim_t nr, \
|
|
ctype* restrict kappa, \
|
|
ctype* restrict d, inc_t incd, \
|
|
ctype* restrict b, inc_t rs_b, inc_t cs_b, \
|
|
ctype** restrict p, inc_t* restrict rs_p, inc_t* restrict cs_p, \
|
|
inc_t* restrict ps_p, \
|
|
cntx_t* restrict cntx, \
|
|
rntm_t* restrict rntm, \
|
|
mem_t* restrict mem, \
|
|
thrinfo_t* restrict thread \
|
|
); \
|
|
|
|
//INSERT_GENTPROT_BASIC0( packm_b )
|
|
GENTPROT( float, s, packm_b )
|
|
GENTPROT( double, d, packm_b )
|
|
GENTPROT( scomplex, c, packm_b )
|
|
GENTPROT( dcomplex, z, packm_b )
|
|
|