Stokhos Package Browser (Single Doxygen Collection) Version of the Day
Loading...
Searching...
No Matches
Stokhos_Ifpack2_UQ_PCE.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#ifndef STOKHOS_IFPACK2_UQ_PCE_HPP
43#define STOKHOS_IFPACK2_UQ_PCE_HPP
44
45// This header file should be included whenever compiling any Ifpack2
46// code with Stokhos scalar types
47
48// MP includes and specializations
50
51// Specialization of LocalReciprocalThreshold functor in
52// Ifpack2_Details_Chebyshev_def.hpp
53namespace Ifpack2 {
54namespace Details {
55
56template <typename XV, class SizeType>
58
59// Mean-based implementation of ReciprocalThreshold
60template <typename S, typename ... P, class SizeType>
62 Kokkos::View< Sacado::UQ::PCE<S>*,P... >,
63 SizeType >
64{
65 typedef Kokkos::View< Sacado::UQ::PCE<S>*,P... > XVector;
66 typedef typename XVector::array_type array_type;
67
68 typedef typename array_type::execution_space execution_space;
69 typedef SizeType size_type;
70 typedef typename array_type::non_const_value_type value_type;
71 typedef Kokkos::Details::ArithTraits<value_type> KAT;
72 typedef typename KAT::mag_type mag_type;
73
74 const XVector m_x;
78
80 const XVector& x,
81 const typename XVector::non_const_value_type& min_val) :
82 m_x(x),
83 m_min_val(min_val.fastAccessCoeff(0)),
84 m_min_val_mag(KAT::abs(m_min_val)),
85 m_n_pce(Kokkos::dimension_scalar(x)) {}
86 //--------------------------------------------------------------------------
87
88 KOKKOS_INLINE_FUNCTION
89 void operator()( const size_type i) const
90 {
91 value_type z = m_x(i).fastAccessCoeff(0);
92 if (KAT::abs(z) < m_min_val_mag)
93 z = m_min_val;
94 else
95 z = KAT::one() / z;
96 m_x(i) = z;
97 }
98};
99
100template<class XV, class SizeType>
102
103template<class S, class ...P, class SizeType>
105 Kokkos::View< Sacado::UQ::PCE<S>*,P... >, SizeType > {
106 typedef Kokkos::View< Sacado::UQ::PCE<S>*,P... > XV;
107
108 static void
109 compute (const XV& X,
110 const typename XV::non_const_value_type& minVal)
111 {
112 if (!Sacado::is_constant(minVal)) {
114 "LocalReciprocalThreshold not implemented for non-constant minVal");
115 }
116
117 if (Kokkos::dimension_scalar(X) == 1) {
118 typedef typename Kokkos::FlatArrayType<XV>::type Flat_XV;
119 Flat_XV flat_X = X;
121 minVal.coeff(0) );
122 }
123 else {
125 Kokkos::parallel_for( X.extent(0), op );
126 }
127 }
128};
129
130}
131}
132
133#endif // STOKHOS_IFPACK2_UQ_PCE_HPP
expr1 expr1 expr1 expr2 expr1 expr1 c expr2 expr1 c fastAccessCoeff(j) - expr2.val(j)
KOKKOS_INLINE_FUNCTION void raise_error(const char *msg)
KOKKOS_INLINE_FUNCTION constexpr std::enable_if< is_view_uq_pce< View< T, P... > >::value, unsigned >::type dimension_scalar(const View< T, P... > &view)
KOKKOS_INLINE_FUNCTION bool is_constant(const T &x)
static void compute(const XV &X, const typename XV::non_const_value_type &minVal)
V_ReciprocalThresholdSelfFunctor(const XVector &x, const typename XVector::non_const_value_type &min_val)