summaryrefslogtreecommitdiff
path: root/interface/spmv.c
diff options
context:
space:
mode:
authorXianyi Zhang <traits.zhang@gmail.com>2011-01-24 14:54:24 +0000
committerXianyi Zhang <traits.zhang@gmail.com>2011-01-24 14:54:24 +0000
commit342bbc3871d1b43f548e9d1ae9d380a1d4989cb3 (patch)
tree385fc6d27e282ae1bb06d685833c5325615c8169 /interface/spmv.c
downloadopenblas-342bbc3871d1b43f548e9d1ae9d380a1d4989cb3.tar.gz
openblas-342bbc3871d1b43f548e9d1ae9d380a1d4989cb3.tar.bz2
openblas-342bbc3871d1b43f548e9d1ae9d380a1d4989cb3.zip
Import GotoBLAS2 1.13 BSD version codes.
Diffstat (limited to 'interface/spmv.c')
-rw-r--r--interface/spmv.c207
1 files changed, 207 insertions, 0 deletions
diff --git a/interface/spmv.c b/interface/spmv.c
new file mode 100644
index 000000000..8d8902763
--- /dev/null
+++ b/interface/spmv.c
@@ -0,0 +1,207 @@
+/*********************************************************************/
+/* Copyright 2009, 2010 The University of Texas at Austin. */
+/* All rights reserved. */
+/* */
+/* Redistribution and use in source and binary forms, with or */
+/* without modification, are permitted provided that the following */
+/* conditions are met: */
+/* */
+/* 1. Redistributions of source code must retain the above */
+/* copyright notice, this list of conditions and the following */
+/* disclaimer. */
+/* */
+/* 2. 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. */
+/* */
+/* THIS SOFTWARE IS PROVIDED BY THE UNIVERSITY OF TEXAS AT */
+/* AUSTIN ``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 UNIVERSITY OF TEXAS AT */
+/* AUSTIN 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. */
+/* */
+/* The views and conclusions contained in the software and */
+/* documentation are those of the authors and should not be */
+/* interpreted as representing official policies, either expressed */
+/* or implied, of The University of Texas at Austin. */
+/*********************************************************************/
+
+#include <stdio.h>
+#include <ctype.h>
+#include "common.h"
+#ifdef FUNCTION_PROFILE
+#include "functable.h"
+#endif
+
+#ifdef XDOUBLE
+#define ERROR_NAME "QSPMV "
+#elif defined(DOUBLE)
+#define ERROR_NAME "DSPMV "
+#else
+#define ERROR_NAME "SSPMV "
+#endif
+
+static int (*spmv[])(BLASLONG, FLOAT, FLOAT *, FLOAT *, BLASLONG, FLOAT *, BLASLONG, void *) = {
+#ifdef XDOUBLE
+ qspmv_U, qspmv_L,
+#elif defined(DOUBLE)
+ dspmv_U, dspmv_L,
+#else
+ sspmv_U, sspmv_L,
+#endif
+};
+
+#ifdef SMP
+static int (*spmv_thread[])(BLASLONG, FLOAT, FLOAT *, FLOAT *, BLASLONG, FLOAT *, BLASLONG, FLOAT *, int) = {
+#ifdef XDOUBLE
+ qspmv_thread_U, qspmv_thread_L,
+#elif defined(DOUBLE)
+ dspmv_thread_U, dspmv_thread_L,
+#else
+ sspmv_thread_U, sspmv_thread_L,
+#endif
+};
+#endif
+
+#ifndef CBLAS
+
+void NAME(char *UPLO, blasint *N, FLOAT *ALPHA, FLOAT *a,
+ FLOAT *x, blasint *INCX, FLOAT *BETA, FLOAT *y, blasint *INCY){
+
+ char uplo_arg = *UPLO;
+ blasint n = *N;
+ FLOAT alpha = *ALPHA;
+ blasint incx = *INCX;
+ FLOAT beta = *BETA;
+ blasint incy = *INCY;
+
+ blasint info;
+ int uplo;
+ FLOAT *buffer;
+#ifdef SMP
+ int nthreads;
+#endif
+
+ PRINT_DEBUG_NAME;
+
+ TOUPPER(uplo_arg);
+ uplo = -1;
+
+ if (uplo_arg == 'U') uplo = 0;
+ if (uplo_arg == 'L') uplo = 1;
+
+ info = 0;
+
+ if (incy == 0) info = 9;
+ if (incx == 0) info = 6;
+ if (n < 0) info = 2;
+ if (uplo < 0) info = 1;
+
+ if (info != 0) {
+ BLASFUNC(xerbla)(ERROR_NAME, &info, sizeof(ERROR_NAME));
+ return;
+ }
+
+#else
+
+void CNAME(enum CBLAS_ORDER order,
+ enum CBLAS_UPLO Uplo,
+ blasint n,
+ FLOAT alpha,
+ FLOAT *a,
+ FLOAT *x, blasint incx,
+ FLOAT beta,
+ FLOAT *y, blasint incy){
+
+ FLOAT *buffer;
+ int uplo;
+ blasint info;
+#ifdef SMP
+ int nthreads;
+#endif
+
+ PRINT_DEBUG_CNAME;
+
+ uplo = -1;
+ info = 0;
+
+ if (order == CblasColMajor) {
+ if (Uplo == CblasUpper) uplo = 0;
+ if (Uplo == CblasLower) uplo = 1;
+
+ info = -1;
+
+ if (incy == 0) info = 9;
+ if (incx == 0) info = 6;
+ if (n < 0) info = 2;
+ if (uplo < 0) info = 1;
+ }
+
+ if (order == CblasRowMajor) {
+ if (Uplo == CblasUpper) uplo = 1;
+ if (Uplo == CblasLower) uplo = 0;
+
+ info = -1;
+
+ if (incy == 0) info = 9;
+ if (incx == 0) info = 6;
+ if (n < 0) info = 2;
+ if (uplo < 0) info = 1;
+ }
+
+ if (info >= 0) {
+ BLASFUNC(xerbla)(ERROR_NAME, &info, sizeof(ERROR_NAME));
+ return;
+ }
+
+#endif
+
+ if (n == 0) return;
+
+ if (beta != ONE) SCAL_K(n, 0, 0, beta, y, abs(incy), NULL, 0, NULL, 0);
+
+ if (alpha == ZERO) return;
+
+ IDEBUG_START;
+
+ FUNCTION_PROFILE_START();
+
+ if (incx < 0 ) x -= (n - 1) * incx;
+ if (incy < 0 ) y -= (n - 1) * incy;
+
+ buffer = (FLOAT *)blas_memory_alloc(1);
+
+#ifdef SMP
+ nthreads = num_cpu_avail(2);
+
+ if (nthreads == 1) {
+#endif
+
+ (spmv[uplo])(n, alpha, a, x, incx, y, incy, buffer);
+
+#ifdef SMP
+ } else {
+
+ (spmv_thread[uplo])(n, alpha, a, x, incx, y, incy, buffer, nthreads);
+
+ }
+#endif
+
+ blas_memory_free(buffer);
+
+ FUNCTION_PROFILE_END(1, n * n / 2 + n, n * n);
+
+ IDEBUG_END;
+
+ return;
+}