Intrepid2
Intrepid2_ArrayToolsDefCloneScale.hpp
Go to the documentation of this file.
1 // @HEADER
2 // *****************************************************************************
3 // Intrepid2 Package
4 //
5 // Copyright 2007 NTESS and the Intrepid2 contributors.
6 // SPDX-License-Identifier: BSD-3-Clause
7 // *****************************************************************************
8 // @HEADER
9 
16 #ifndef __INTREPID2_ARRAYTOOLS_DEF_CLONESCALE_HPP__
17 #define __INTREPID2_ARRAYTOOLS_DEF_CLONESCALE_HPP__
18 
19 namespace Intrepid2 {
20 
21  namespace FunctorArrayTools {
22 
26  template<typename OutputViewType,
27  typename inputViewType,
28  ordinal_type valRank>
29  struct F_clone {
30  OutputViewType _output;
31  const inputViewType _input;
32 
33  KOKKOS_INLINE_FUNCTION
34  F_clone(OutputViewType output_,
35  inputViewType input_)
36  : _output(output_),
37  _input(input_) {}
38 
39  // clone data
40  KOKKOS_INLINE_FUNCTION
41  void
42  operator()(const ordinal_type cl,
43  const ordinal_type pt) const {
44  switch (valRank) {
45  case 0: {
46  _output.access(cl, pt) = _input.access(pt);
47  break;
48  }
49  case 1: {
50  const ordinal_type iend = _output.extent(2);
51  for (ordinal_type i=0;i<iend;++i)
52  _output.access(cl, pt, i) = _input.access(pt, i);
53  break;
54  }
55  case 2: {
56  const ordinal_type
57  iend = _output.extent(2),
58  jend = _output.extent(3);
59  for (ordinal_type i=0;i<iend;++i)
60  for (ordinal_type j=0;j<jend;++j)
61  _output.access(cl, pt, i, j) = _input.access(pt, i, j);
62  break;
63  }
64  }
65  }
66 
67  // clone fields
68  KOKKOS_INLINE_FUNCTION
69  void
70  operator()(const ordinal_type cl,
71  const ordinal_type bf,
72  const ordinal_type pt) const {
73  switch (valRank) {
74  case 0: {
75  _output.access(cl, bf, pt) = _input.access(bf, pt);
76  break;
77  }
78  case 1: {
79  const ordinal_type iend = _output.extent(3);
80  for (ordinal_type i=0;i<iend;++i)
81  _output.access(cl, bf, pt, i) = _input.access(bf, pt, i);
82  break;
83  }
84  case 2: {
85  const ordinal_type
86  iend = _output.extent(3),
87  jend = _output.extent(4);
88  for (ordinal_type i=0;i<iend;++i)
89  for (ordinal_type j=0;j<jend;++j)
90  _output.access(cl, bf, pt, i, j) = _input.access(bf, pt, i, j);
91  break;
92  }
93  }
94  }
95  };
96  }
97 
98  template<typename DeviceType>
99  template<typename outputValueType, class ...outputProperties,
100  typename inputValueType, class ...inputProperties>
101  void ArrayTools<DeviceType>::
102  cloneFields( Kokkos::DynRankView<outputValueType,outputProperties...> output,
103  const Kokkos::DynRankView<inputValueType, inputProperties...> input ) {
104 #ifdef HAVE_INTREPID2_DEBUG
105  {
106  INTREPID2_TEST_FOR_EXCEPTION( ( input.rank() < 2 || input.rank() > 4 ), std::invalid_argument,
107  ">>> ERROR (ArrayTools::clone): Input fields container must have rank 2, 3, or 4.");
108  INTREPID2_TEST_FOR_EXCEPTION( ( output.rank() != (input.rank()+1) ), std::invalid_argument,
109  ">>> ERROR (ArrayTools::clone): The rank of the input fields container must be one less than the rank of the output fields container.");
110  for (size_type i=0;i< input.rank();++i) {
111  INTREPID2_TEST_FOR_EXCEPTION( (input.extent(i) != output.extent(i+1)), std::invalid_argument,
112  ">>> ERROR (ArrayTools::clone): Dimensions of input and output fields containers do not match.");
113  }
114  }
115 #endif
116 
117  typedef Kokkos::DynRankView<outputValueType,outputProperties...> OutputViewType;
118  typedef Kokkos::DynRankView<inputValueType, inputProperties...> inputViewType;
119 
120  using range_policy_type = Kokkos::MDRangePolicy
121  < ExecSpaceType, Kokkos::Rank<3>, Kokkos::IndexType<ordinal_type> >;
122 
123  range_policy_type policy( { 0, 0, 0 },
124  { /*C*/ output.extent(0), /*F*/ output.extent(1), /*P*/ output.extent(2) } );
125  const ordinal_type valRank = output.rank() - 3;
126  switch (valRank) {
127  case 0: {
128  typedef FunctorArrayTools::F_clone<OutputViewType,inputViewType,0> FunctorType;
129  Kokkos::parallel_for( policy, FunctorType(output, input) );
130  break;
131  }
132  case 1: {
133  typedef FunctorArrayTools::F_clone<OutputViewType,inputViewType,1> FunctorType;
134  Kokkos::parallel_for( policy, FunctorType(output, input) );
135  break;
136  }
137  case 2: {
138  typedef FunctorArrayTools::F_clone<OutputViewType,inputViewType,2> FunctorType;
139  Kokkos::parallel_for( policy, FunctorType(output, input) );
140  break;
141  }
142  }
143  }
144 
145  template<typename DeviceType>
146  template<typename outputValueType, class ...outputProperties,
147  typename inputValueType, class ...inputProperties>
148  void ArrayTools<DeviceType>::
149  cloneData( Kokkos::DynRankView<outputValueType,outputProperties...> output,
150  const Kokkos::DynRankView<inputValueType, inputProperties...> input ) {
151 #ifdef HAVE_INTREPID2_DEBUG
152  {
153  INTREPID2_TEST_FOR_EXCEPTION( ( input.rank() < 1 || input.rank() > 3 ), std::invalid_argument,
154  ">>> ERROR (ArrayTools::clone): Input fields container must have rank 1, 2, or 3.");
155  INTREPID2_TEST_FOR_EXCEPTION( ( output.rank() != (input.rank()+1) ), std::invalid_argument,
156  ">>> ERROR (ArrayTools::clone): The rank of the input fields container must be one less than the rank of the output fields container.");
157  for (ordinal_type i=0;i<input.rank();++i) {
158  INTREPID2_TEST_FOR_EXCEPTION( (input.extent(i) != output.extent(i+1)), std::invalid_argument,
159  ">>> ERROR (ArrayTools::clone): Dimensions of input and output fields containers do not match.");
160  }
161  }
162 #endif
163 
164  typedef Kokkos::DynRankView<outputValueType,outputProperties...> OutputViewType;
165  typedef Kokkos::DynRankView<inputValueType, inputProperties...> inputViewType;
166 
167  using range_policy_type = Kokkos::MDRangePolicy
168  < ExecSpaceType, Kokkos::Rank<2>, Kokkos::IndexType<ordinal_type> >;
169 
170  range_policy_type policy( { 0, 0 },
171  { /*C*/ output.extent(0), /*P*/ output.extent(1) } );
172  const ordinal_type valRank = output.rank() - 2;
173  switch (valRank) {
174  case 0: {
175  typedef FunctorArrayTools::F_clone<OutputViewType,inputViewType,0> FunctorType;
176  Kokkos::parallel_for( policy, FunctorType(output, input) );
177  break;
178  }
179  case 1: {
180  typedef FunctorArrayTools::F_clone<OutputViewType,inputViewType,1> FunctorType;
181  Kokkos::parallel_for( policy, FunctorType(output, input) );
182  break;
183  }
184  case 2: {
185  typedef FunctorArrayTools::F_clone<OutputViewType,inputViewType,2> FunctorType;
186  Kokkos::parallel_for( policy, FunctorType(output, input) );
187  break;
188  }
189  }
190  }
191 
192 } // end namespace Intrepid2
193 
194 #endif
Intrepid2::FunctorArrayTools::F_clone
Functor for clone see Intrepid2::ArrayTools for more.
Definition: Intrepid2_ArrayToolsDefCloneScale.hpp:29
Intrepid2::ArrayTools::cloneData
static void cloneData(Kokkos::DynRankView< outputDataValueType, outputDataProperties...> outputData, const Kokkos::DynRankView< inputDataValueType, inputDataProperties...> inputData)
Replicates a rank-2, 3, or 4 container with dimensions (F,P), (F,P,D1) or (F,P,D1,D2), representing the values of a scalar, vector or a tensor field, into an output value container of size (C,F,P), (C,F,P,D1) or (C,F,P,D1,D2).
Intrepid2::ArrayTools::cloneFields
static void cloneFields(Kokkos::DynRankView< outputFieldValueType, outputFieldProperties...> outputFields, const Kokkos::DynRankView< inputFieldValueType, inputFieldProperties...> inputFields)
Replicates a rank-2, 3, or 4 container with dimensions (F,P), (F,P,D1) or (F,P,D1,D2), representing the values of a scalar, vector or a tensor field, into an output value container of size (C,F,P), (C,F,P,D1) or (C,F,P,D1,D2).
Generated by   doxygen 1.8.5