Panzer  Version of the Day
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Panzer_CommonArrayFactories_impl.hpp
Go to the documentation of this file.
1 // @HEADER
2 // *****************************************************************************
3 // Panzer: A partial differential equation assembly
4 // engine for strongly coupled complex multiphysics systems
5 //
6 // Copyright 2011 NTESS and the Panzer contributors.
7 // SPDX-License-Identifier: BSD-3-Clause
8 // *****************************************************************************
9 // @HEADER
10 
11 #ifndef PANZER_COMMON_ARRAY_FACTORIES_IMPL_HPP
12 #define PANZER_COMMON_ARRAY_FACTORIES_IMPL_HPP
13 
14 #include "Teuchos_RCP.hpp"
15 #include "Phalanx_DataLayout_MDALayout.hpp"
16 #include "Phalanx_KokkosViewFactory.hpp"
17 
18 namespace panzer {
19 
20 // Implementation for intrepid container factory
21 template <typename Scalar,typename T0>
22 Kokkos::DynRankView<Scalar,PHX::Device> Intrepid2FieldContainerFactory::
23 buildArray(const std::string & str,int d0) const
24 {
25  static_assert(std::is_same<Scalar,double>::value,"ERROR: CommonArryFactory for DynRankView only supports double scalar type!");
26  return Kokkos::DynRankView<Scalar,PHX::Device>(str,d0);
27 }
28 
29 template <typename Scalar,typename T0,typename T1>
30 Kokkos::DynRankView<Scalar,PHX::Device> Intrepid2FieldContainerFactory::
31 buildArray(const std::string & str,int d0,int d1) const
32 {
33  static_assert(std::is_same<Scalar,double>::value,"ERROR: CommonArryFactory for DynRankView only supports double scalar type!");
34  return Kokkos::DynRankView<Scalar,PHX::Device>(str,d0,d1);
35 }
36 
37 template <typename Scalar,typename T0,typename T1,typename T2>
38 Kokkos::DynRankView<Scalar,PHX::Device> Intrepid2FieldContainerFactory::
39 buildArray(const std::string & str,int d0,int d1,int d2) const
40 {
41  static_assert(std::is_same<Scalar,double>::value,"ERROR: CommonArryFactory for DynRankView only supports double scalar type!");
42  return Kokkos::DynRankView<Scalar,PHX::Device>(str,d0,d1,d2);
43 }
44 
45 template <typename Scalar,typename T0,typename T1,typename T2,typename T3>
46 Kokkos::DynRankView<Scalar,PHX::Device> Intrepid2FieldContainerFactory::
47 buildArray(const std::string & str,int d0,int d1,int d2,int d3) const
48 {
49  static_assert(std::is_same<Scalar,double>::value,"ERROR: CommonArryFactory for DynRankView only supports double scalar type!");
50  return Kokkos::DynRankView<Scalar,PHX::Device>(str,d0,d1,d2,d3);
51 }
52 
53 template <typename Scalar,typename T0,typename T1,typename T2,typename T3,typename T4>
54 Kokkos::DynRankView<Scalar,PHX::Device> Intrepid2FieldContainerFactory::
55 buildArray(const std::string & str,int d0,int d1,int d2,int d3,int d4) const
56 {
57  static_assert(std::is_same<Scalar,double>::value,"ERROR: CommonArryFactory for DynRankView only supports double scalar type!");
58  return Kokkos::DynRankView<Scalar,PHX::Device>(str,d0,d1,d2,d3,d4);
59 }
60 
61 // Implementation for MDField array factory
62 template <typename Scalar,typename T0>
63 PHX::MDField<Scalar> MDFieldArrayFactory::
64 buildArray(const std::string & str,int d0) const
65 {
66  typedef PHX::KokkosViewFactory<Scalar,typename PHX::DevLayout<Scalar>::type,PHX::Device> ViewFactory;
67 
68  PHX::MDField<Scalar> field = PHX::MDField<Scalar>(prefix_+str,Teuchos::rcp(new PHX::MDALayout<T0>(d0)));
69 
70  if(allocArray_)
71  field.setFieldData(ViewFactory::buildView(field.fieldTag(),ddims_));
72 
73  return field;
74 }
75 
76 template <typename Scalar,typename T0,typename T1>
77 PHX::MDField<Scalar> MDFieldArrayFactory::
78 buildArray(const std::string & str,int d0,int d1) const
79 {
80  typedef PHX::KokkosViewFactory<Scalar,typename PHX::DevLayout<Scalar>::type,PHX::Device> ViewFactory;
81 
82  PHX::MDField<Scalar> field = PHX::MDField<Scalar>(prefix_+str,Teuchos::rcp(new PHX::MDALayout<T0,T1 >(d0,d1)));
83 
84  if(allocArray_)
85  field.setFieldData(ViewFactory::buildView(field.fieldTag(),ddims_));
86 
87  return field;
88 }
89 
90 template <typename Scalar,typename T0,typename T1,typename T2>
91 PHX::MDField<Scalar> MDFieldArrayFactory::
92 buildArray(const std::string & str,int d0,int d1,int d2) const
93 {
94  typedef PHX::KokkosViewFactory<Scalar,typename PHX::DevLayout<Scalar>::type,PHX::Device> ViewFactory;
95 
96  PHX::MDField<Scalar> field = PHX::MDField<Scalar>(prefix_+str,Teuchos::rcp(new PHX::MDALayout<T0,T1,T2>(d0,d1,d2)));
97 
98  if(allocArray_)
99  field.setFieldData(ViewFactory::buildView(field.fieldTag(),ddims_));
100 
101  return field;
102 }
103 
104 template <typename Scalar,typename T0,typename T1,typename T2,typename T3>
105 PHX::MDField<Scalar> MDFieldArrayFactory::
106 buildArray(const std::string & str,int d0,int d1,int d2,int d3) const
107 {
108  typedef PHX::KokkosViewFactory<Scalar,typename PHX::DevLayout<Scalar>::type,PHX::Device> ViewFactory;
109 
110  PHX::MDField<Scalar> field = PHX::MDField<Scalar>(prefix_+str,Teuchos::rcp(new PHX::MDALayout<T0,T1,T2,T3>(d0,d1,d2,d3)));
111 
112  if(allocArray_)
113  field.setFieldData(ViewFactory::buildView(field.fieldTag(),ddims_));
114 
115  return field;
116 }
117 
118 template <typename Scalar,typename T0,typename T1,typename T2,typename T3,typename T4>
119 PHX::MDField<Scalar> MDFieldArrayFactory::
120 buildArray(const std::string & str,int d0,int d1,int d2,int d3,int d4) const
121 {
122  typedef PHX::KokkosViewFactory<Scalar,typename PHX::DevLayout<Scalar>::type,PHX::Device> ViewFactory;
123 
124  PHX::MDField<Scalar> field = PHX::MDField<Scalar>(prefix_+str,Teuchos::rcp(new PHX::MDALayout<T0,T1,T2,T3,T4>(d0,d1,d2,d3,d4)));
125 
126  if(allocArray_)
127  field.setFieldData(ViewFactory::buildView(field.fieldTag(),ddims_));
128 
129  return field;
130 }
131 
132 // Implementation for MDField array factory
133 template <typename Scalar,typename T0>
134 PHX::MDField<Scalar,T0> MDFieldArrayFactory::
135 buildStaticArray(const std::string & str,int d0) const
136 {
137  typedef PHX::KokkosViewFactory<Scalar,typename PHX::DevLayout<Scalar>::type,PHX::Device> ViewFactory;
138 
139  PHX::MDField<Scalar,T0> field = PHX::MDField<Scalar,T0>(prefix_+str,Teuchos::rcp(new PHX::MDALayout<T0>(d0)));
140 
141  if(allocArray_)
142  field.setFieldData(ViewFactory::buildView(field.fieldTag(),ddims_));
143 
144  return field;
145 }
146 
147 template <typename Scalar,typename T0,typename T1>
148 PHX::MDField<Scalar,T0,T1> MDFieldArrayFactory::
149 buildStaticArray(const std::string & str,int d0,int d1) const
150 {
151  typedef PHX::KokkosViewFactory<Scalar,typename PHX::DevLayout<Scalar>::type,PHX::Device> ViewFactory;
152 
153  PHX::MDField<Scalar,T0,T1> field = PHX::MDField<Scalar,T0,T1>(prefix_+str,Teuchos::rcp(new PHX::MDALayout<T0,T1>(d0,d1)));
154 
155  if(allocArray_)
156  field.setFieldData(ViewFactory::buildView(field.fieldTag(),ddims_));
157 
158  return field;
159 }
160 
161 template <typename Scalar,typename T0,typename T1,typename T2>
162 PHX::MDField<Scalar,T0,T1,T2> MDFieldArrayFactory::
163 buildStaticArray(const std::string & str,int d0,int d1,int d2) const
164 {
165  typedef PHX::KokkosViewFactory<Scalar,typename PHX::DevLayout<Scalar>::type,PHX::Device> ViewFactory;
166 
167  PHX::MDField<Scalar,T0,T1,T2> field = PHX::MDField<Scalar,T0,T1,T2>(prefix_+str,Teuchos::rcp(new PHX::MDALayout<T0,T1,T2>(d0,d1,d2)));
168 
169  if(allocArray_)
170  field.setFieldData(ViewFactory::buildView(field.fieldTag(),ddims_));
171 
172  return field;
173 }
174 
175 template <typename Scalar,typename T0,typename T1,typename T2,typename T3>
176 PHX::MDField<Scalar,T0,T1,T2,T3> MDFieldArrayFactory::
177 buildStaticArray(const std::string & str,int d0,int d1,int d2,int d3) const
178 {
179  typedef PHX::KokkosViewFactory<Scalar,typename PHX::DevLayout<Scalar>::type,PHX::Device> ViewFactory;
180 
181  PHX::MDField<Scalar,T0,T1,T2,T3> field = PHX::MDField<Scalar,T0,T1,T2,T3>(prefix_+str,Teuchos::rcp(new PHX::MDALayout<T0,T1,T2,T3>(d0,d1,d2,d3)));
182 
183  if(allocArray_)
184  field.setFieldData(ViewFactory::buildView(field.fieldTag(),ddims_));
185 
186  return field;
187 }
188 
189 template <typename Scalar,typename T0,typename T1,typename T2,typename T3,typename T4>
190 PHX::MDField<Scalar,T0,T1,T2,T3,T4> MDFieldArrayFactory::
191 buildStaticArray(const std::string & str,int d0,int d1,int d2,int d3,int d4) const
192 {
193  typedef PHX::KokkosViewFactory<Scalar,typename PHX::DevLayout<Scalar>::type,PHX::Device> ViewFactory;
194 
195  PHX::MDField<Scalar,T0,T1,T2,T3,T4> field = PHX::MDField<Scalar,T0,T1,T2,T3,T4>(prefix_+str,Teuchos::rcp(new PHX::MDALayout<T0,T1,T2,T3,T4>(d0,d1,d2,d3,d4)));
196 
197  if(allocArray_)
198  field.setFieldData(ViewFactory::buildView(field.fieldTag(),ddims_));
199 
200  return field;
201 }
202 
203 } // end namespace panzer
204 
205 #endif
panzer::MDFieldArrayFactory::ddims_
std::vector< PHX::index_size_type > ddims_
Definition: Panzer_CommonArrayFactories.hpp:104
panzer::MDFieldArrayFactory::buildStaticArray
PHX::MDField< Scalar, T0 > buildStaticArray(const std::string &str, int d0) const
Definition: Panzer_CommonArrayFactories_impl.hpp:135
Teuchos::rcp
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
panzer::Intrepid2FieldContainerFactory::buildArray
Kokkos::DynRankView< Scalar, PHX::Device > buildArray(const std::string &str, int d0) const
Definition: Panzer_CommonArrayFactories_impl.hpp:23
field
PHX::MDField< ScalarT, panzer::Cell, panzer::BASIS > field
A field to which we&#39;ll contribute, or in which we&#39;ll store, the result of computing this integral...
Definition: Panzer_Integrator_CurlBasisDotVector_impl.hpp:559
panzer::MDFieldArrayFactory::buildArray
PHX::MDField< Scalar > buildArray(const std::string &str, int d0) const
Definition: Panzer_CommonArrayFactories_impl.hpp:64
Teuchos_RCP.hpp
Generated on Thu Nov 21 2024 09:17:08 for Panzer by   doxygen 1.8.5