Added support for pre-broadcast when packing B.

Details:
- Added support for being able to duplicate (broadcast) elements in
  memory when packing matrix B (ie: the left-hand operand) in level-3
  operations. This turns out advantageous for some architectures that
  can afford the cost of the extra bandwidth and somehow benefit from
  the pre-broadcast elements (and thus being able to avoid using
  broadcast-style load instructions on micro-rows of B in the gemm
  microkernel).
- Support optionally disabling right-side hemm and symm. If this occurs,
  hemm_r is implemented in terms of hemm_l (and symm_r in terms of
  symm_l). This is needed when broadcasting during packing because the
  alternative--supporting the broadcast of B while also allowing matrix
  B to be Hermitian/symmetric--would be an absolute mess.
- Support alignment factors for packed blocks of A, B, and C separately
  (as well as for general-purpose buffers). In addition, we support
  byte offsets from those alignment values (which is different from
  aligning by align+offset bytes to begin with). The default alignment
  values are BLIS_PAGE_SIZE in all four cases, with the offset values
  defaulting to zero.
- Pass pack_t schema into bli_?packm_cxk() so that it can be then passed
  into the packm kernel, where it will be needed by packm kernels that
  perform broadcasts of B, since the idea is that we *only* want to
  broadcast when packing micropanels of B and not A.
- Added definition for variadic bli_cntx_set_l3_vir_ukrs(), which can be
  used to set custom virtual level-3 microkernels in the cntx_t, which
  would typically be done in the bli_cntx_init_*() function defined in
  the subconfiguration of interest.
- Added a "broadcast B" kernel function for use with NP/NR = 12/6,
  defined in in ref_kernels/1m/bli_packm_cxk_bb_ref.c.
- Added a gemm, gemmtrsm, and trsm "broadcast B" reference kernels
  defined in ref_kernels/3/bb. (These kernels have been tested with
  double real with NP/NR = 12/6.)
- Added #ifndef ... #endif guards around several macro constants defined
  in frame/include/bli_kernel_macro_defs.h.
- Defined a few "broadcast B" static functions in
  frame/include/level0/bb for use by "broadcast B"-style packm reference
  kernels. For now, only the real domain kernels are tested and fully
  defined.
- Output the alignment and offset values for packed blocks of A and B
  in the testsuite's "BLIS configuration info" section.
- Comment updates to various files.
- Bumped so_version to 3.0.0.
This commit is contained in:
Field G. Van Zee
2019-09-17 17:42:10 -05:00
parent fd9bf497cd
commit 31c8657f1d
36 changed files with 2237 additions and 122 deletions

View File

@@ -0,0 +1,74 @@
/*
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(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.
*/
#ifndef BLIS_BCASTBBS_MXN_H
#define BLIS_BCASTBBS_MXN_H
// bcastbbs_mxn
#undef GENTFUNC
#define GENTFUNC( ctype, ch, opname ) \
\
static void PASTEMAC(ch,opname) \
( \
const dim_t m, \
const dim_t n, \
ctype* restrict y, const inc_t incy, const inc_t ldy \
) \
{ \
/* Assume that the duplication factor is the column stride of y. */ \
const dim_t d = ldy; \
const dim_t ds_y = 1; \
\
for ( dim_t i = 0; i < m; ++i ) \
{ \
ctype* restrict yi = y + i*incy; \
\
for ( dim_t j = 0; j < n; ++j ) \
{ \
ctype* restrict yij = yi + j*ldy; \
\
for ( dim_t p = 1; p < d; ++p ) \
{ \
ctype* restrict yijd = yij + p*ds_y; \
\
PASTEMAC(ch,copys)( *yij, *yijd ); \
} \
} \
} \
}
INSERT_GENTFUNC_BASIC0( bcastbbs_mxn )
#endif

View File

@@ -0,0 +1,204 @@
/*
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(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.
*/
#ifndef BLIS_SCAL2BBS_MXN_H
#define BLIS_SCAL2BBS_MXN_H
// scal2bbs_mxn
#undef GENTFUNCRO
#define GENTFUNCRO( ctype, ch, opname ) \
\
static void PASTEMAC(ch,opname) \
( \
const conj_t conjx, \
const dim_t m, \
const dim_t n, \
ctype* restrict alpha, \
ctype* restrict x, const inc_t incx, const inc_t ldx, \
ctype* restrict y, const inc_t incy, const inc_t ldy \
) \
{ \
/* Assume that the duplication factor is the row stride of y. */ \
const dim_t d = incy; \
const dim_t ds_y = 1; \
\
if ( bli_is_conj( conjx ) ) \
{ \
for ( dim_t j = 0; j < n; ++j ) \
{ \
ctype* restrict xj = x + j*ldx; \
ctype* restrict yj = y + j*ldy; \
\
for ( dim_t i = 0; i < m; ++i ) \
{ \
ctype* restrict xij = xj + i*incx; \
ctype* restrict yij = yj + i*incy; \
\
PASTEMAC(ch,scal2js)( *alpha, *xij, *yij ); \
\
for ( dim_t p = 1; p < d; ++p ) \
{ \
ctype* restrict yijd = yij + p*ds_y; \
\
PASTEMAC(ch,copys)( *yij, *yijd ); \
} \
} \
} \
} \
else /* if ( bli_is_noconj( conjx ) ) */ \
{ \
for ( dim_t j = 0; j < n; ++j ) \
{ \
ctype* restrict xj = x + j*ldx; \
ctype* restrict yj = y + j*ldy; \
\
for ( dim_t i = 0; i < m; ++i ) \
{ \
ctype* restrict xij = xj + i*incx; \
ctype* restrict yij = yj + i*incy; \
\
PASTEMAC(ch,scal2s)( *alpha, *xij, *yij ); \
\
for ( dim_t p = 1; p < d; ++p ) \
{ \
ctype* restrict yijd = yij + p*ds_y; \
\
PASTEMAC(ch,copys)( *yij, *yijd ); \
} \
} \
} \
} \
}
INSERT_GENTFUNCRO_BASIC0( scal2bbs_mxn )
#undef GENTFUNCCO
#define GENTFUNCCO( ctype, ctype_r, ch, chr, opname ) \
\
static void PASTEMAC(ch,opname) \
( \
const conj_t conjx, \
const dim_t m, \
const dim_t n, \
ctype* restrict alpha, \
ctype* restrict x, const inc_t incx, const inc_t ldx, \
ctype* restrict y, const inc_t incy, const inc_t ldy \
) \
{ \
/* Assume that the duplication factor is the row stride of y. */ \
const dim_t d = incy; \
const dim_t ds_y = 1; \
\
const inc_t incx2 = 2 * incx; \
const inc_t ldx2 = 2 * ldx; \
\
const inc_t incy2 = 2 * incy; \
const inc_t ldy2 = 2 * ldy; \
\
ctype_r* restrict alpha_r = ( ctype_r* )alpha; \
ctype_r* restrict alpha_i = ( ctype_r* )alpha + 1; \
ctype_r* restrict chi_r = ( ctype_r* )x; \
ctype_r* restrict chi_i = ( ctype_r* )x + 1; \
ctype_r* restrict psi_r = ( ctype_r* )y; \
ctype_r* restrict psi_i = ( ctype_r* )y + 1*d; \
\
if ( bli_is_conj( conjx ) ) \
{ \
for ( dim_t j = 0; j < n; ++j ) \
{ \
ctype_r* restrict chij_r = chi_r + j*ldx2; \
ctype_r* restrict chij_i = chi_i + j*ldx2; \
ctype_r* restrict psij_r = psi_r + j*ldy2; \
ctype_r* restrict psij_i = psi_i + j*ldy2; \
\
for ( dim_t i = 0; i < m; ++i ) \
{ \
ctype_r* restrict chiij_r = chij_r + i*incx2; \
ctype_r* restrict chiij_i = chij_i + i*incx2; \
ctype_r* restrict psiij_r = psij_r + i*incy2; \
ctype_r* restrict psiij_i = psij_i + i*incy2; \
\
PASTEMAC(ch,scal2jris)( *alpha_r, *alpha_i, \
*chiij_r, *chiij_i, \
*psiij_r, *psiij_i ); \
\
for ( dim_t p = 1; i < d; ++p ) \
{ \
ctype_r* restrict psiijd_r = psiij_r + p*ds_y; \
ctype_r* restrict psiijd_i = psiij_i + p*ds_y; \
\
PASTEMAC(ch,copyris)( *psiij_r, *psiij_i, \
*psiijd_r, *psiijd_i ); \
} \
} \
} \
} \
else /* if ( bli_is_noconj( conjx ) ) */ \
{ \
for ( dim_t j = 0; j < n; ++j ) \
{ \
ctype_r* restrict chij_r = chi_r + j*ldx2; \
ctype_r* restrict chij_i = chi_i + j*ldx2; \
ctype_r* restrict psij_r = psi_r + j*ldy2; \
ctype_r* restrict psij_i = psi_i + j*ldy2; \
\
for ( dim_t i = 0; i < m; ++i ) \
{ \
ctype_r* restrict chiij_r = chij_r + i*incx2; \
ctype_r* restrict chiij_i = chij_i + i*incx2; \
ctype_r* restrict psiij_r = psij_r + i*incy2; \
ctype_r* restrict psiij_i = psij_i + i*incy2; \
\
PASTEMAC(ch,scal2ris)( *alpha_r, *alpha_i, \
*chiij_r, *chiij_i, \
*psiij_r, *psiij_i ); \
\
for ( dim_t p = 1; i < d; ++p ) \
{ \
ctype_r* restrict psiijd_r = psiij_r + p*ds_y; \
ctype_r* restrict psiijd_i = psiij_i + p*ds_y; \
\
PASTEMAC(ch,copyris)( *psiij_r, *psiij_i, \
*psiijd_r, *psiijd_i ); \
} \
} \
} \
} \
}
INSERT_GENTFUNCCO_BASIC0( scal2bbs_mxn )
#endif

View File

@@ -0,0 +1,74 @@
/*
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(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.
*/
#ifndef BLIS_SET0BBS_MXN_H
#define BLIS_SET0BBS_MXN_H
// set0bbs_mxn
#undef GENTFUNC
#define GENTFUNC( ctype, ch, opname ) \
\
static void PASTEMAC(ch,opname) \
( \
const dim_t m, \
const dim_t n, \
ctype* restrict y, const inc_t incy, const inc_t ldy \
) \
{ \
/* Assume that the duplication factor is the row stride of y. */ \
const dim_t d = incy; \
const dim_t ds_y = 1; \
\
for ( dim_t j = 0; j < n; ++j ) \
{ \
ctype* restrict yj = y + j*ldy; \
\
for ( dim_t i = 0; i < m; ++i ) \
{ \
ctype* restrict yij = yj + i*incy; \
\
for ( dim_t p = 0; p < d; ++p ) \
{ \
ctype* restrict yijd = yij + p*ds_y; \
\
PASTEMAC(ch,set0s)( *yijd ); \
} \
} \
} \
}
INSERT_GENTFUNC_BASIC0( set0bbs_mxn )
#endif

View File

@@ -0,0 +1,89 @@
/*
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(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.
*/
#ifndef BLIS_SCAL2S_MXN_H
#define BLIS_SCAL2S_MXN_H
// scal2s_mxn
#undef GENTFUNC
#define GENTFUNC( ctype, ch, opname ) \
\
static void PASTEMAC(ch,opname) \
( \
const conj_t conjx, \
const dim_t m, \
const dim_t n, \
ctype* restrict alpha, \
ctype* restrict x, const inc_t rs_x, const inc_t cs_x, \
ctype* restrict y, const inc_t rs_y, const inc_t cs_y \
) \
{ \
if ( bli_is_conj( conjx ) ) \
{ \
for ( dim_t j = 0; j < n; ++j ) \
{ \
ctype* restrict xj = x + j*cs_x; \
ctype* restrict yj = y + j*cs_y; \
\
for ( dim_t i = 0; i < m; ++i ) \
{ \
ctype* restrict xij = xj + i*rs_x; \
ctype* restrict yij = yj + i*rs_y; \
\
PASTEMAC(ch,scal2js)( *alpha, *xij, *yij ); \
} \
} \
} \
else /* if ( bli_is_noconj( conjx ) ) */ \
{ \
for ( dim_t j = 0; j < n; ++j ) \
{ \
ctype* restrict xj = x + j*cs_x; \
ctype* restrict yj = y + j*cs_y; \
\
for ( dim_t i = 0; i < m; ++i ) \
{ \
ctype* restrict xij = xj + i*rs_x; \
ctype* restrict yij = yj + i*rs_y; \
\
PASTEMAC(ch,scal2s)( *alpha, *xij, *yij ); \
} \
} \
} \
}
INSERT_GENTFUNC_BASIC0( scal2s_mxn )
#endif