Stokhos Package Browser (Single Doxygen Collection)
Version of the Day
Loading...
Searching...
No Matches
src
sacado
kokkos
vector
Stokhos_MP_Vector_MaskTraits_sfs_relops_tmpl.hpp
Go to the documentation of this file.
1
// @HEADER
2
// ***********************************************************************
3
//
4
// Stokhos Package
5
// Copyright (2009) Sandia Corporation
6
//
7
// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
8
// license for use of this work by or on behalf of the U.S. Government.
9
//
10
// Redistribution and use in source and binary forms, with or without
11
// modification, are permitted provided that the following conditions are
12
// met:
13
//
14
// 1. Redistributions of source code must retain the above copyright
15
// notice, this list of conditions and the following disclaimer.
16
//
17
// 2. Redistributions in binary form must reproduce the above copyright
18
// notice, this list of conditions and the following disclaimer in the
19
// documentation and/or other materials provided with the distribution.
20
//
21
// 3. Neither the name of the Corporation nor the names of the
22
// contributors may be used to endorse or promote products derived from
23
// this software without specific prior written permission.
24
//
25
// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
26
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
29
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
30
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
31
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
32
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33
// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
34
// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
35
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36
//
37
// Questions? Contact Eric T. Phipps (etphipp@sandia.gov).
38
//
39
// ***********************************************************************
40
// @HEADER
41
42
namespace
Sacado
{
43
namespace
MP {
44
45
template
<
typename
O,
typename
T,
int
N,
typename
D>
46
KOKKOS_INLINE_FUNCTION
47
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> >
48
operator
OPNAME
(
const
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>& a,
49
const
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>& b)
50
{
51
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> > mask;
52
#ifdef STOKHOS_HAVE_PRAGMA_IVDEP
53
#pragma ivdep
54
#endif
55
#ifdef STOKHOS_HAVE_PRAGMA_VECTOR_ALIGNED
56
#pragma vector aligned
57
#endif
58
#ifdef STOKHOS_HAVE_PRAGMA_UNROLL
59
#pragma unroll
60
#endif
61
for
(O i=0; i<a.size(); ++i)
62
mask.
set
(i, a.fastAccessCoeff(i)
OPNAME
b.fastAccessCoeff(i));
63
return
mask;
64
}
65
66
template
<
typename
O,
typename
T,
int
N,
typename
D>
67
KOKKOS_INLINE_FUNCTION
68
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> >
69
operator
OPNAME
(
const
volatile
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>& a,
70
const
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>& b)
71
{
72
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> > mask;
73
#ifdef STOKHOS_HAVE_PRAGMA_IVDEP
74
#pragma ivdep
75
#endif
76
#ifdef STOKHOS_HAVE_PRAGMA_VECTOR_ALIGNED
77
#pragma vector aligned
78
#endif
79
#ifdef STOKHOS_HAVE_PRAGMA_UNROLL
80
#pragma unroll
81
#endif
82
for
(O i=0; i<a.size(); ++i)
83
mask.
set
(i, a.fastAccessCoeff(i)
OPNAME
b.fastAccessCoeff(i));
84
return
mask;
85
}
86
87
template
<
typename
O,
typename
T,
int
N,
typename
D>
88
KOKKOS_INLINE_FUNCTION
89
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> >
90
operator
OPNAME
(
const
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>& a,
91
const
volatile
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>& b)
92
{
93
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> > mask;
94
#ifdef STOKHOS_HAVE_PRAGMA_IVDEP
95
#pragma ivdep
96
#endif
97
#ifdef STOKHOS_HAVE_PRAGMA_VECTOR_ALIGNED
98
#pragma vector aligned
99
#endif
100
#ifdef STOKHOS_HAVE_PRAGMA_UNROLL
101
#pragma unroll
102
#endif
103
for
(O i=0; i<a.size(); ++i)
104
mask.
set
(i, a.fastAccessCoeff(i)
OPNAME
b.fastAccessCoeff(i));
105
return
mask;
106
}
107
108
template
<
typename
O,
typename
T,
int
N,
typename
D>
109
KOKKOS_INLINE_FUNCTION
110
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> >
111
operator
OPNAME
(
const
volatile
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>& a,
112
const
volatile
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>& b)
113
{
114
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> > mask;
115
#ifdef STOKHOS_HAVE_PRAGMA_IVDEP
116
#pragma ivdep
117
#endif
118
#ifdef STOKHOS_HAVE_PRAGMA_VECTOR_ALIGNED
119
#pragma vector aligned
120
#endif
121
#ifdef STOKHOS_HAVE_PRAGMA_UNROLL
122
#pragma unroll
123
#endif
124
for
(O i=0; i<a.size(); ++i)
125
mask.
set
(i, a.fastAccessCoeff(i)
OPNAME
b.fastAccessCoeff(i));
126
return
mask;
127
}
128
129
template
<
typename
O,
typename
T,
int
N,
typename
D>
130
KOKKOS_INLINE_FUNCTION
131
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> >
132
operator
OPNAME
(
const
typename
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>::value_type& a,
133
const
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>& b)
134
{
135
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> > mask;
136
#ifdef STOKHOS_HAVE_PRAGMA_IVDEP
137
#pragma ivdep
138
#endif
139
#ifdef STOKHOS_HAVE_PRAGMA_VECTOR_ALIGNED
140
#pragma vector aligned
141
#endif
142
#ifdef STOKHOS_HAVE_PRAGMA_UNROLL
143
#pragma unroll
144
#endif
145
for
(O i=0; i<b.size(); ++i)
146
mask.
set
(i, a
OPNAME
b.fastAccessCoeff(i));
147
return
mask;
148
}
149
150
template
<
typename
O,
typename
T,
int
N,
typename
D>
151
KOKKOS_INLINE_FUNCTION
152
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> >
153
operator
OPNAME
(
const
typename
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>::value_type& a,
154
const
volatile
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>& b)
155
{
156
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> > mask;
157
#ifdef STOKHOS_HAVE_PRAGMA_IVDEP
158
#pragma ivdep
159
#endif
160
#ifdef STOKHOS_HAVE_PRAGMA_VECTOR_ALIGNED
161
#pragma vector aligned
162
#endif
163
#ifdef STOKHOS_HAVE_PRAGMA_UNROLL
164
#pragma unroll
165
#endif
166
for
(O i=0; i<b.size(); ++i)
167
mask.
set
(i, a
OPNAME
b.fastAccessCoeff(i));
168
return
mask;
169
}
170
171
template
<
typename
O,
typename
T,
int
N,
typename
D>
172
KOKKOS_INLINE_FUNCTION
173
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> >
174
operator
OPNAME
(
const
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>& a,
175
const
typename
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>::value_type& b)
176
{
177
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> > mask;
178
#ifdef STOKHOS_HAVE_PRAGMA_IVDEP
179
#pragma ivdep
180
#endif
181
#ifdef STOKHOS_HAVE_PRAGMA_VECTOR_ALIGNED
182
#pragma vector aligned
183
#endif
184
#ifdef STOKHOS_HAVE_PRAGMA_UNROLL
185
#pragma unroll
186
#endif
187
for
(O i=0; i<a.size(); ++i)
188
mask.
set
(i, a.fastAccessCoeff(i)
OPNAME
b);
189
return
mask;
190
}
191
192
template
<
typename
O,
typename
T,
int
N,
typename
D>
193
KOKKOS_INLINE_FUNCTION
194
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> >
195
operator
OPNAME
(
const
volatile
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>& a,
196
const
typename
Vector
<
Stokhos::StaticFixedStorage<O,T,N,D>
>::value_type& b)
197
{
198
Mask< Vector< Stokhos::StaticFixedStorage<O,T,N,D>
> > mask;
199
#ifdef STOKHOS_HAVE_PRAGMA_IVDEP
200
#pragma ivdep
201
#endif
202
#ifdef STOKHOS_HAVE_PRAGMA_VECTOR_ALIGNED
203
#pragma vector aligned
204
#endif
205
#ifdef STOKHOS_HAVE_PRAGMA_UNROLL
206
#pragma unroll
207
#endif
208
for
(O i=0; i<a.size(); ++i)
209
mask.
set
(i, a.fastAccessCoeff(i)
OPNAME
b);
210
return
mask;
211
}
212
}
213
}
OPNAME
#define OPNAME
Definition:
Sacado_MP_Vector_SFS_ops.hpp:177
Mask
Definition:
Stokhos_MP_Vector_MaskTraits.hpp:508
Mask::set
KOKKOS_INLINE_FUNCTION void set(int i, bool b)
Definition:
Stokhos_MP_Vector_MaskTraits.hpp:674
Sacado::MP::Vector
Definition:
Belos_SolverManager_MP_Vector.hpp:48
Stokhos::StaticFixedStorage
Statically allocated storage class.
Definition:
Stokhos_StaticFixedStorage.hpp:69
Sacado
Definition:
Kokkos_View_UQ_PCE_Fwd.hpp:62
Generated by
1.9.6