43 #ifndef PANZER_SUM_IMPL_HPP
44 #define PANZER_SUM_IMPL_HPP
53 template<
typename EvalT,
typename Traits>
58 std::string sum_name = p.
get<std::string>(
"Sum Name");
64 TEUCHOS_ASSERT(static_cast<int>(value_names->size()) < MAX_VALUES);
67 auto local_scalars = PHX::View<double *>(
"scalars",value_names->size());
68 auto local_scalars_host = Kokkos::create_mirror_view(local_scalars);
75 for (std::size_t i=0; i < value_names->size(); ++i)
76 local_scalars_host(i) = scalars_v[i];
79 for (std::size_t i=0; i < value_names->size(); ++i)
80 local_scalars_host(i) = 1.0;
82 Kokkos::deep_copy(local_scalars,local_scalars_host);
84 scalars = local_scalars;
88 this->addEvaluatedField(sum);
90 for (std::size_t i=0; i < value_names->size(); ++i) {
92 this->addDependentField(values[i]);
102 std::string n =
"Sum Evaluator";
107 template<
typename EvalT,
typename Traits>
114 cell_data_size = sum.size() / sum.fieldTag().dataLayout().extent(0);
119 template<
typename EvalT,
typename TRAITS>
120 template<
unsigned int RANK>
121 KOKKOS_INLINE_FUNCTION
123 auto num_vals = scalars.extent(0);
128 for (std::size_t iv = 0; iv < num_vals; ++iv)
129 sum(i) += scalars(iv)*(values[iv](i));
133 const size_t dim_1 = sum.extent(1);
134 for (std::size_t j = 0; j < dim_1; ++j)
135 for (std::size_t iv = 0; iv < num_vals; ++iv)
136 sum(i,j) += scalars(iv)*(values[iv](i,j));
140 const size_t dim_1 = sum.extent(1),dim_2 = sum.extent(2);
141 for (std::size_t j = 0; j < dim_1; ++j)
142 for (std::size_t k = 0; k < dim_2; ++k)
143 for (std::size_t iv = 0; iv < num_vals; ++iv)
144 sum(i,j,k) += scalars(iv)*(values[iv](i,j,k));
148 const size_t dim_1 = sum.extent(1),dim_2 = sum.extent(2),dim_3 = sum.extent(3);
149 for (std::size_t j = 0; j < dim_1; ++j)
150 for (std::size_t k = 0; k < dim_2; ++k)
151 for (std::size_t l = 0; l < dim_3; ++l)
152 for (std::size_t iv = 0; iv < num_vals; ++iv)
153 sum(i,j,k,l) += scalars(iv)*(values[iv](i,j,k,l));
157 const size_t dim_1 = sum.extent(1),dim_2 = sum.extent(2),dim_3 = sum.extent(3),dim_4 = sum.extent(4);
158 for (std::size_t j = 0; j < dim_1; ++j)
159 for (std::size_t k = 0; k < dim_2; ++k)
160 for (std::size_t l = 0; l < dim_3; ++l)
161 for (std::size_t m = 0; m < dim_4; ++m)
162 for (std::size_t iv = 0; iv < num_vals; ++iv)
163 sum(i,j,k,l,m) += scalars(iv)*(values[iv](i,j,k,l,m));
167 const size_t dim_1 = sum.extent(1),dim_2 = sum.extent(2),dim_3 = sum.extent(3),dim_4 = sum.extent(4),dim_5 = sum.extent(5);
168 for (std::size_t j = 0; j < dim_1; ++j)
169 for (std::size_t k = 0; k < dim_2; ++k)
170 for (std::size_t l = 0; l < dim_3; ++l)
171 for (std::size_t m = 0; m < dim_4; ++m)
172 for (std::size_t n = 0; n < dim_5; ++n)
173 for (std::size_t iv = 0; iv < num_vals; ++iv)
174 sum(i,j,k,l,m,n) += scalars(iv)*(values[iv](i,j,k,l,m,n));
179 template<
typename EvalT,
typename Traits>
188 size_t rank = sum.rank();
189 const size_t length = sum.extent(0);
192 Kokkos::parallel_for(Kokkos::RangePolicy<
PanzerSumTag<1> >(0, length), *
this);
196 Kokkos::parallel_for(Kokkos::RangePolicy<
PanzerSumTag<2> >(0, length), *
this);
200 Kokkos::parallel_for(Kokkos::RangePolicy<
PanzerSumTag<3> >(0, length), *
this);
204 Kokkos::parallel_for(Kokkos::RangePolicy<
PanzerSumTag<4> >(0, length), *
this);
208 Kokkos::parallel_for(Kokkos::RangePolicy<
PanzerSumTag<5> >(0, length), *
this);
212 Kokkos::parallel_for(Kokkos::RangePolicy<
PanzerSumTag<6> >(0, length), *
this);
227 template<
typename EvalT,
typename TRAITS,
typename Tag0>
231 std::string sum_name = p.
get<std::string>(
"Sum Name");
242 this->addEvaluatedField(sum);
244 values.resize(value_names->size());
245 for (std::size_t i=0; i < value_names->size(); ++i) {
247 this->addDependentField(values[i]);
250 std::string n =
"SumStatic Rank 1 Evaluator";
256 template<
typename EvalT,
typename TRAITS,
typename Tag0>
261 for (std::size_t i = 0; i < sum.extent(0); ++i)
262 for (std::size_t d = 0; d < values.size(); ++d)
263 sum(i) += (values[d])(i);
269 template<
typename EvalT,
typename TRAITS,
typename Tag0,
typename Tag1>
273 std::string sum_name = p.
get<std::string>(
"Sum Name");
288 PHX::View<double*> scalars_nc = PHX::View<double*>(
"scalars",scalar_values->size());
290 for(std::size_t i=0;i<scalar_values->size();i++)
291 scalars_nc(i) = (*scalar_values)[i];
293 scalars = scalars_nc;
304 this->addEvaluatedField(sum);
306 values.resize(value_names->size());
307 for (std::size_t i=0; i < value_names->size(); ++i) {
309 this->addDependentField(values[i]);
311 numValues = value_names->size();
314 std::string n =
"SumStatic Rank 2 Evaluator";
320 template<
typename EvalT,
typename TRAITS,
typename Tag0,
typename Tag1>
323 const std::vector<double> & scalar_values,
329 if(scalars.size()==0) {
335 PHX::View<double*> scalars_nc = PHX::View<double*>(
"scalars",scalar_values.size());
337 for(std::size_t i=0;i<scalar_values.size();i++)
338 scalars_nc(i) = scalar_values[i];
340 scalars = scalars_nc;
347 this->addEvaluatedField(sum);
349 values.resize(inputs.size());
350 for (std::size_t i=0; i < inputs.size(); ++i) {
351 values[i] = inputs[i];
352 this->addDependentField(values[i]);
355 numValues = inputs.size();
357 std::string n =
"SumStatic Rank 2 Evaluator";
363 template<
typename EvalT,
typename TRAITS,
typename Tag0,
typename Tag1>
368 for (std::size_t i=0; i < values.size(); ++i)
369 value_views[i] = values[i].get_static_view();
374 template<
typename EvalT,
typename TRAITS,
typename Tag0,
typename Tag1>
382 Kokkos::parallel_for(Kokkos::RangePolicy<PHX::Device,ScalarsTag>(0,sum.extent(0)), *
this);
384 Kokkos::parallel_for(Kokkos::RangePolicy<PHX::Device,NoScalarsTag>(0,sum.extent(0)), *
this);
389 template<
typename EvalT,
typename TRAITS,
typename Tag0,
typename Tag1>
390 KOKKOS_INLINE_FUNCTION
394 for (
int i=0;i<numValues;i++) {
395 for (
int j = 0; j < sum.extent_int(1); ++j)
396 sum(c,j) += scalars(i)*value_views[i](c,j);
402 template<
typename EvalT,
typename TRAITS,
typename Tag0,
typename Tag1>
403 KOKKOS_INLINE_FUNCTION
407 for (
int i=0;i<numValues;i++) {
408 for (
int j = 0; j < sum.extent_int(1); ++j)
409 sum(c,j) += value_views[i](c,j);
466 template<
typename EvalT,
typename TRAITS,
typename Tag0,
typename Tag1,
typename Tag2>
469 const std::vector<std::string> & value_names,
473 p.
set<std::string>(
"Sum Name",sum_name);
Teuchos::RCP< PHX::Evaluator< TRAITS > > buildStaticSumEvaluator(const std::string &sum_name, const std::vector< std::string > &value_names, const Teuchos::RCP< PHX::DataLayout > &data_layout)
Sum(const Teuchos::ParameterList &p)
T & get(const std::string &name, T def_value)
ParameterList & set(std::string const &name, T const &value, std::string const &docString="", RCP< const ParameterEntryValidator > const &validator=null)
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
void postRegistrationSetup(typename Traits::SetupData d, PHX::FieldManager< Traits > &fm)
KOKKOS_INLINE_FUNCTION void operator()(PanzerSumTag< RANK >, const int &i) const
SumStatic(const Teuchos::ParameterList &p)
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
virtual void postRegistrationSetup(typename Traits::SetupData d, PHX::FieldManager< Traits > &vm)=0
void evaluateFields(typename Traits::EvalData d)
typename EvalT::ScalarT ScalarT
bool isType(const std::string &name) const
void evaluateFields(typename TRAITS::EvalData d)
#define TEUCHOS_ASSERT(assertion_test)