Stokhos Package Browser (Single Doxygen Collection)  Version of the Day
Kokkos_View_UQ_PCE_Utils.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 KOKKOS_VIEW_UQ_PCE_UTILS_HPP
43 #define KOKKOS_VIEW_UQ_PCE_UTILS_HPP
44 
45 #include "Kokkos_View_Utils.hpp"
46 #include "Sacado_UQ_PCE.hpp"
47 
48 namespace Kokkos {
49 
50 // Type name for a local, unmanaged view with possibly a different static size
51 template <typename ViewType,
52  unsigned LocalSize,
53  unsigned Rank = ViewType::Rank,
54  bool isStatic = ViewType::is_static>
55 struct LocalUQPCEView {};
56 
57 template <typename ViewType,
58  unsigned LocalSize>
59 struct LocalUQPCEView< ViewType, LocalSize, 1, true > {
61  typedef typename ViewType::array_layout array_layout;
63  typedef typename ViewType::device_type device_type;
65  typedef typename storage_type::template apply_N<LocalSize> StorageApply;
66  typedef typename StorageApply::type local_storage_type;
68 
69  typedef Kokkos::View< local_value_type*,
72  Kokkos::MemoryUnmanaged > type;
73 };
74 
75 template <typename ViewType,
76  unsigned LocalSize>
77 struct LocalUQPCEView<ViewType, LocalSize, 1, false> {
79  typedef typename ViewType::array_layout array_layout;
81  typedef typename ViewType::device_type device_type;
82 
83  typedef Kokkos::View< vector_type*,
86  Kokkos::MemoryUnmanaged > type;
87 };
88 
89 namespace Impl {
90 
91 template< class OldStorageType , class Device >
92 struct RebindStokhosStorageDevice< Sacado::UQ::PCE< OldStorageType > , Device >
93 {
94  typedef typename
95  OldStorageType::template apply<
98  Device >
100 
101  typedef typename NewStorageApply::type NewStorageType ;
102  typedef typename Sacado::UQ::PCE< OldStorageType >::template apply< NewStorageType > NewVectorApply ;
103 
104  typedef typename NewVectorApply::type type ;
105 };
106 
107 template< class OldStorageType , class Device >
108 struct RebindStokhosStorageDevice< const Sacado::UQ::PCE< OldStorageType > , Device >
109 {
110  typedef typename
111  OldStorageType::template apply<
113  typename OldStorageType::value_type ,
114  Device >
116 
117  typedef typename NewStorageApply::type NewStorageType ;
118  typedef typename Sacado::UQ::PCE< OldStorageType >::template apply< NewStorageType > NewVectorApply ;
119 
120  typedef const typename NewVectorApply::type type ;
121 };
122 
123 } // namespace Impl
124 
125 // Whether a given type is a view with scalar type Sacado::UQ::PCE
126 template <typename view_type>
127 struct is_view_uq_pce { static const bool value = false; };
128 
129 // Typename of the Cijk tensor in a view
130 template <typename view_type, typename Enabled = void>
131 struct CijkType {};
132 
133 } // namespace Kokkos
134 
135 #endif // KOKKOS_UQ_PCE_UTILS_HPP
Stokhos::StandardStorage< int, double > storage_type
Kokkos::DefaultExecutionSpace execution_space
Sacado::UQ::PCE< OldStorageType >::template apply< NewStorageType > NewVectorApply
Kokkos::View< vector_type *, array_layout, device_type, Kokkos::MemoryUnmanaged > type
storage_type::template apply_N< LocalSize > StorageApply
Kokkos::View< local_value_type *, array_layout, device_type, Kokkos::MemoryUnmanaged > type
OldStorageType::template apply< typename OldStorageType::ordinal_type, typename OldStorageType::value_type, Device > NewStorageApply
OldStorageType::template apply< typename OldStorageType::ordinal_type, typename OldStorageType::value_type, Device > NewStorageApply
Sacado::UQ::PCE< OldStorageType >::template apply< NewStorageType > NewVectorApply