[dense-linear-algebra-libraries/linalg.git] / src / ti / linalg / blis / frame / 1d / scald / bli_scald_unb_var1.c
diff --git a/src/ti/linalg/blis/frame/1d/scald/bli_scald_unb_var1.c b/src/ti/linalg/blis/frame/1d/scald/bli_scald_unb_var1.c
--- /dev/null
@@ -0,0 +1,152 @@
+/*
+
+ 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 "blis.h"
+
+#define FUNCPTR_T scald_fp
+
+typedef void (*FUNCPTR_T)(
+ conj_t conjbeta,
+ doff_t diagoffx,
+ dim_t m,
+ dim_t n,
+ void* beta,
+ void* x, inc_t rs_x, inc_t cs_x
+ );
+
+// 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,scald_unb_var1);
+#else
+#ifdef BLIS_ENABLE_MIXED_DOMAIN_SUPPORT
+static FUNCPTR_T GENARRAY2_EXT(ftypes,scald_unb_var1);
+#else
+static FUNCPTR_T GENARRAY2_MIN(ftypes,scald_unb_var1);
+#endif
+#endif
+
+
+void bli_scald_unb_var1( obj_t* beta,
+ obj_t* x )
+{
+ num_t dt_x = bli_obj_datatype( *x );
+
+ conj_t conjbeta = bli_obj_conj_status( *beta );
+
+ 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_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( conjbeta,
+ diagoffx,
+ m,
+ n,
+ buf_beta,
+ buf_x, rs_x, cs_x );
+}
+
+
+#undef GENTFUNC2
+#define GENTFUNC2( ctype_b, ctype_x, chb, chx, varname, kername ) \
+\
+void PASTEMAC2(chb,chx,varname)( \
+ conj_t conjbeta, \
+ doff_t diagoffx, \
+ dim_t m, \
+ dim_t n, \
+ void* beta, \
+ void* x, inc_t rs_x, inc_t cs_x \
+ ) \
+{ \
+ ctype_b* beta_cast = beta; \
+ ctype_x* x_cast = x; \
+ ctype_x* x1; \
+ dim_t n_elem; \
+ dim_t offx; \
+ inc_t incx; \
+\
+ if ( bli_zero_dim2( m, n ) ) return; \
+\
+ /* If beta is unit, the entire operation is a no-op. */ \
+ if ( PASTEMAC(chb,eq1)( *beta_cast ) ) return; \
+\
+ if ( bli_is_outside_diag( diagoffx, BLIS_NO_TRANSPOSE, m, n ) ) return; \
+\
+ /* Determine the distance to the diagonal, the number of diagonal
+ elements, and the diagonal increment. */ \
+ bli_set_dims_incs_1d( diagoffx, \
+ m, n, rs_x, cs_x, \
+ offx, n_elem, incx ); \
+\
+ x1 = x_cast + offx; \
+\
+ PASTEMAC2(chb,chx,kername)( conjbeta, \
+ n_elem, \
+ beta_cast, \
+ x1, incx ); \
+}
+
+// Define the basic set of functions unconditionally, and then also some
+// mixed datatype functions if requested.
+INSERT_GENTFUNC2_BASIC( scald_unb_var1, SCALV_KERNEL )
+
+#ifdef BLIS_ENABLE_MIXED_DOMAIN_SUPPORT
+INSERT_GENTFUNC2_MIX_D( scald_unb_var1, SCALV_KERNEL )
+#endif
+
+#ifdef BLIS_ENABLE_MIXED_PRECISION_SUPPORT
+INSERT_GENTFUNC2_MIX_P( scald_unb_var1, SCALV_KERNEL )
+#endif
+