1 /*
2 * Copyright (c) 2010, Michael Lehn
3 *
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1) Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2) Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in
14 * the documentation and/or other materials provided with the
15 * distribution.
16 * 3) Neither the name of the FLENS development group nor the names of
17 * its contributors may be used to endorse or promote products derived
18 * from this software without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
24 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
25 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
26 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
27 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
28 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
29 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
30 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31 */
32
33 #ifndef CXXBLAS_LEVEL2_SPMV_TCC
34 #define CXXBLAS_LEVEL2_SPMV_TCC 1
35
36 #include <complex>
37 #include <cxxblas/level1/level1.h>
38
39 namespace cxxblas {
40
41 template <typename IndexType, typename ALPHA, typename MA, typename VX,
42 typename BETA, typename VY>
43 void
44 spmv_generic(StorageOrder order, StorageUpLo upLo,
45 IndexType n,
46 const ALPHA &alpha,
47 const MA *A,
48 const VX *x, IndexType incX,
49 const BETA &beta,
50 VY *y, IndexType incY)
51 {
52 if (order==ColMajor) {
53 upLo = (upLo==Upper) ? Lower : Upper;
54 }
55 scal_generic(n, beta, y, incY);
56 if (upLo==Upper) {
57 for (IndexType i=0, iY=0, iX=0; i<n; ++i, iX+=incX, iY+=incY) {
58 VY _y = VY(0);
59 dot_generic(n-i, A+i*(2*n-i+1)/2, IndexType(1),
60 x+iX, incX, _y);
61 y[iY] += alpha*_y;
62 axpy_generic(n-i-1, alpha*x[iX], A+i*(2*n-i+1)/2+1, IndexType(1),
63 y+iY+incY, incY);
64 }
65 } else {
66 for (IndexType i=0, iY=0, iX=0; i<n; ++i, iX+=incX, iY+=incY) {
67 VY _y = VY(0);
68 dot_generic(i, A+i*(i+1)/2, IndexType(1), x, incX, _y);
69 y[iY] += alpha*_y;
70 axpy_generic(i+1, alpha*x[iX], A+i*(i+1)/2, IndexType(1),
71 y, incY);
72 }
73 }
74 }
75
76 //------------------------------------------------------------------------------
77
78 template <typename IndexType, typename ALPHA, typename MA, typename VX,
79 typename BETA, typename VY>
80 void
81 spmv(StorageOrder order, StorageUpLo upLo,
82 IndexType n,
83 const ALPHA &alpha,
84 const MA *A,
85 const VX *x, IndexType incX,
86 const BETA &beta,
87 VY *y, IndexType incY)
88 {
89 CXXBLAS_DEBUG_OUT("spmv_generic");
90
91 if (incX<0) {
92 x -= incX*(n-1);
93 }
94 if (incY<0) {
95 y -= incY*(n-1);
96 }
97 spmv_generic(order, upLo, n, alpha, A, x, incX, beta, y, incY);
98 }
99
100 } // namespace cxxblas
101
102 #endif // CXXBLAS_LEVEL2_SPMV_TCC
2 * Copyright (c) 2010, Michael Lehn
3 *
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1) Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2) Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in
14 * the documentation and/or other materials provided with the
15 * distribution.
16 * 3) Neither the name of the FLENS development group nor the names of
17 * its contributors may be used to endorse or promote products derived
18 * from this software without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
24 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
25 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
26 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
27 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
28 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
29 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
30 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31 */
32
33 #ifndef CXXBLAS_LEVEL2_SPMV_TCC
34 #define CXXBLAS_LEVEL2_SPMV_TCC 1
35
36 #include <complex>
37 #include <cxxblas/level1/level1.h>
38
39 namespace cxxblas {
40
41 template <typename IndexType, typename ALPHA, typename MA, typename VX,
42 typename BETA, typename VY>
43 void
44 spmv_generic(StorageOrder order, StorageUpLo upLo,
45 IndexType n,
46 const ALPHA &alpha,
47 const MA *A,
48 const VX *x, IndexType incX,
49 const BETA &beta,
50 VY *y, IndexType incY)
51 {
52 if (order==ColMajor) {
53 upLo = (upLo==Upper) ? Lower : Upper;
54 }
55 scal_generic(n, beta, y, incY);
56 if (upLo==Upper) {
57 for (IndexType i=0, iY=0, iX=0; i<n; ++i, iX+=incX, iY+=incY) {
58 VY _y = VY(0);
59 dot_generic(n-i, A+i*(2*n-i+1)/2, IndexType(1),
60 x+iX, incX, _y);
61 y[iY] += alpha*_y;
62 axpy_generic(n-i-1, alpha*x[iX], A+i*(2*n-i+1)/2+1, IndexType(1),
63 y+iY+incY, incY);
64 }
65 } else {
66 for (IndexType i=0, iY=0, iX=0; i<n; ++i, iX+=incX, iY+=incY) {
67 VY _y = VY(0);
68 dot_generic(i, A+i*(i+1)/2, IndexType(1), x, incX, _y);
69 y[iY] += alpha*_y;
70 axpy_generic(i+1, alpha*x[iX], A+i*(i+1)/2, IndexType(1),
71 y, incY);
72 }
73 }
74 }
75
76 //------------------------------------------------------------------------------
77
78 template <typename IndexType, typename ALPHA, typename MA, typename VX,
79 typename BETA, typename VY>
80 void
81 spmv(StorageOrder order, StorageUpLo upLo,
82 IndexType n,
83 const ALPHA &alpha,
84 const MA *A,
85 const VX *x, IndexType incX,
86 const BETA &beta,
87 VY *y, IndexType incY)
88 {
89 CXXBLAS_DEBUG_OUT("spmv_generic");
90
91 if (incX<0) {
92 x -= incX*(n-1);
93 }
94 if (incY<0) {
95 y -= incY*(n-1);
96 }
97 spmv_generic(order, upLo, n, alpha, A, x, incX, beta, y, incY);
98 }
99
100 } // namespace cxxblas
101
102 #endif // CXXBLAS_LEVEL2_SPMV_TCC