Intrepid2
Intrepid2_TestUtils.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
15#ifndef Intrepid2_TestUtils_h
16#define Intrepid2_TestUtils_h
17
18#include "Kokkos_Core.hpp"
19#include "Kokkos_DynRankView.hpp"
20
21#include "Intrepid2_Basis.hpp"
26#include "Intrepid2_Sacado.hpp" // Sacado includes, guarded by the appropriate preprocessor variable
27#include "Intrepid2_Utils.hpp"
28
29#include "Teuchos_UnitTestHarness.hpp"
30
31namespace Intrepid2
32{
35
37#if defined(KOKKOS_ENABLE_CUDA)
38 using DefaultTestDeviceType = Kokkos::Device<Kokkos::Cuda,Kokkos::CudaSpace>;
39#elif defined(KOKKOS_ENABLE_HIP)
40 using DefaultTestDeviceType = Kokkos::Device<Kokkos::HIP,Kokkos::HIPSpace>;
41#else
42 using DefaultTestDeviceType = typename Kokkos::DefaultExecutionSpace::device_type;
43#endif
44
46 template <class Scalar>
47 const typename Teuchos::ScalarTraits<Scalar>::magnitudeType
49 {
50 using ST = Teuchos::ScalarTraits<Scalar>;
51 return ST::magnitude(Teuchos::RelErrSmallNumber<ST::hasMachineParameters,Scalar>::smallNumber());
52 }
53
55 template <class Scalar1, class Scalar2>
56 KOKKOS_INLINE_FUNCTION
57 bool
58 relErrMeetsTol( const Scalar1 &s1, const Scalar2 &s2, const typename Teuchos::ScalarTraits< typename std::common_type<Scalar1,Scalar2>::type >::magnitudeType &smallNumber, const double &tol )
59 {
60 using Scalar = typename std::common_type<Scalar1,Scalar2>::type;
61 const Scalar s1Abs = fabs(s1);
62 const Scalar s2Abs = fabs(s2);
63 const Scalar maxAbs = (s1Abs > s2Abs) ? s1Abs : s2Abs;
64 const Scalar relErr = fabs( s1 - s2 ) / ( smallNumber + maxAbs );
65 return relErr < tol;
66 }
67
68 template <class Scalar1, class Scalar2>
69 KOKKOS_INLINE_FUNCTION
70 bool
71 errMeetsAbsAndRelTol( const Scalar1 &s1, const Scalar2 &s2, const double &relTol, const double &absTol )
72 {
73 return fabs( s1 - s2 ) <= absTol + fabs(s1) * relTol;
74 }
75
76 static const double TEST_TOLERANCE_TIGHT = 1.e2 * std::numeric_limits<double>::epsilon();
77
78 // we use DynRankView for both input points and values
79 template<typename ScalarType, typename DeviceType>
80 using ViewType = Kokkos::DynRankView<ScalarType,DeviceType>;
81
82 template<typename ScalarType, typename DeviceType>
83 using FixedRankViewType = Kokkos::View<ScalarType,DeviceType>;
84
85 template<typename ScalarType>
86 KOKKOS_INLINE_FUNCTION bool valuesAreSmall(const ScalarType &a, const ScalarType &b, const double &epsilon)
87 {
88 using std::abs;
89 return (abs(a) < epsilon) && (abs(b) < epsilon);
90 }
91
92 inline bool approximatelyEqual(double a, double b, double epsilon)
93 {
94 const double larger_magnitude = (std::abs(a) < std::abs(b) ? std::abs(b) : std::abs(a));
95 return std::abs(a - b) <= larger_magnitude * epsilon;
96 }
97
98 inline bool essentiallyEqual(double a, double b, double epsilon)
99 {
100 const double smaller_magnitude = (std::abs(a) > std::abs(b) ? std::abs(b) : std::abs(a));
101 return std::abs(a - b) <= smaller_magnitude * epsilon;
102 }
103
104 // conversion from the ref element [0,1] (used by ESEAS) to the ref element [-1,1] (used by Intrepid2)
105 KOKKOS_INLINE_FUNCTION double fromZeroOne(double x_zero_one)
106 {
107 return x_zero_one * 2.0 - 1.0;
108 }
109
110 // conversion from the ref element [-1,1] (used by Intrepid2) to the ref element [0,1] (used by ESEAS)
111 KOKKOS_INLINE_FUNCTION double toZeroOne(double x_minus_one_one)
112 {
113 return (x_minus_one_one + 1.0) / 2.0;
114 }
115
116 // conversion from the ref element [0,1] (used by ESEAS) to the ref element [-1,1] (used by Intrepid2)
117 KOKKOS_INLINE_FUNCTION double fromZeroOne_dx(double dx_zero_one)
118 {
119 return dx_zero_one / 2.0;
120 }
121
122 // conversion from the ref element [-1,1] (used by Intrepid2) to the ref element [0,1] (used by ESEAS)
123 KOKKOS_INLINE_FUNCTION double toZeroOne_dx(double dx_minus_one_one)
124 {
125 return dx_minus_one_one * 2.0;
126 }
127
128 template<class DeviceViewType>
129 typename DeviceViewType::host_mirror_type getHostCopy(const DeviceViewType &deviceView)
130 {
131 typename DeviceViewType::host_mirror_type hostView = Kokkos::create_mirror(deviceView);
132 Kokkos::deep_copy(hostView, deviceView);
133 return hostView;
134 }
135
136 template<class BasisFamily>
137 inline Teuchos::RCP< Intrepid2::Basis<DefaultTestDeviceType,double,double> > getBasisUsingFamily(shards::CellTopology cellTopo, Intrepid2::EFunctionSpace fs,
138 int polyOrder_x, int polyOrder_y=-1, int polyOrder_z = -1)
139 {
140 using BasisPtr = typename BasisFamily::BasisPtr;
141
142 BasisPtr basis;
143 using namespace Intrepid2;
144
145 if (cellTopo.getBaseKey() == shards::Line<>::key)
146 {
147 basis = getLineBasis<BasisFamily>(fs,polyOrder_x);
148 }
149 else if (cellTopo.getBaseKey() == shards::Quadrilateral<>::key)
150 {
151 INTREPID2_TEST_FOR_EXCEPTION(polyOrder_y < 0, std::invalid_argument, "polyOrder_y must be specified");
152 basis = getQuadrilateralBasis<BasisFamily>(fs,polyOrder_x,polyOrder_y);
153 }
154 else if (cellTopo.getBaseKey() == shards::Triangle<>::key)
155 {
156 basis = getTriangleBasis<BasisFamily>(fs,polyOrder_x);
157 }
158 else if (cellTopo.getBaseKey() == shards::Hexahedron<>::key)
159 {
160 INTREPID2_TEST_FOR_EXCEPTION(polyOrder_y < 0, std::invalid_argument, "polyOrder_y must be specified");
161 INTREPID2_TEST_FOR_EXCEPTION(polyOrder_z < 0, std::invalid_argument, "polyOrder_z must be specified");
162 basis = getHexahedronBasis<BasisFamily>(fs,polyOrder_x,polyOrder_y,polyOrder_z);
163 }
164 else if (cellTopo.getBaseKey() == shards::Tetrahedron<>::key)
165 {
166 basis = getTetrahedronBasis<BasisFamily>(fs, polyOrder_x);
167 }
168 else if (cellTopo.getBaseKey() == shards::Wedge<>::key)
169 {
170 INTREPID2_TEST_FOR_EXCEPTION(polyOrder_y < 0, std::invalid_argument, "polyOrder_y must be specified");
171 basis = getWedgeBasis<BasisFamily>(fs,polyOrder_x,polyOrder_y);
172 }
173 else if (cellTopo.getBaseKey() == shards::Pyramid<>::key)
174 {
175 basis = getPyramidBasis<BasisFamily>(fs,polyOrder_x);
176 }
177 else
178 {
179 INTREPID2_TEST_FOR_EXCEPTION(true, std::invalid_argument, "Unsupported cell topology");
180 }
181 return basis;
182 }
183
184 template<bool defineVertexFunctions>
185 inline Teuchos::RCP< Intrepid2::Basis<DefaultTestDeviceType,double,double> > getHierarchicalBasis(shards::CellTopology cellTopo, Intrepid2::EFunctionSpace fs,
186 int polyOrder_x, int polyOrder_y=-1, int polyOrder_z = -1)
187 {
188 using DeviceType = DefaultTestDeviceType;
189 using Scalar = double;
190 using namespace Intrepid2;
191
197
199
200 return getBasisUsingFamily<BasisFamily>(cellTopo, fs, polyOrder_x, polyOrder_y, polyOrder_z);
201 }
202
203 template<typename ValueType, typename DeviceType, class ... DimArgs>
204 inline ViewType<ValueType,DeviceType> getView(const std::string &label, DimArgs... dims)
205 {
206 const bool allocateFadStorage = !(std::is_standard_layout<ValueType>::value && std::is_trivial<ValueType>::value);
207 if (!allocateFadStorage)
208 {
209 return ViewType<ValueType,DeviceType>(label,dims...);
210 }
211 else
212 {
213 return ViewType<ValueType,DeviceType>(label,dims...,MAX_FAD_DERIVATIVES_FOR_TESTS+1);
214 }
215 }
216
217 // this method is to allow us to switch tests over incrementally; should collapse with ViewType once everything has been switched
218 template<typename ValueType, class ... DimArgs>
219 inline FixedRankViewType< typename RankExpander<ValueType, sizeof...(DimArgs) >::value_type, DefaultTestDeviceType > getFixedRankView(const std::string &label, DimArgs... dims)
220 {
221 const bool allocateFadStorage = !(std::is_standard_layout<ValueType>::value && std::is_trivial<ValueType>::value);
222 using value_type = typename RankExpander<ValueType, sizeof...(dims) >::value_type;
223 if (!allocateFadStorage)
224 {
225 return FixedRankViewType<value_type,DefaultTestDeviceType>(label,dims...);
226 }
227 else
228 {
229 return FixedRankViewType<value_type,DefaultTestDeviceType>(label,dims...,MAX_FAD_DERIVATIVES_FOR_TESTS+1);
230 }
231 }
232
239 template <typename PointValueType, typename DeviceType>
240 inline ViewType<PointValueType,DeviceType> getInputPointsView(shards::CellTopology &cellTopo, int numPoints_1D)
241 {
242 if (cellTopo.getBaseKey() == shards::Wedge<>::key)
243 {
244 shards::CellTopology lineTopo = shards::CellTopology(shards::getCellTopologyData<shards::Line<> >() );
245 shards::CellTopology triTopo = shards::CellTopology(shards::getCellTopologyData<shards::Triangle<> >() );
246
247 const ordinal_type order = numPoints_1D - 1;
248 ordinal_type numPoints_tri = PointTools::getLatticeSize(triTopo, order);
249 ordinal_type numPoints_line = PointTools::getLatticeSize(lineTopo, order);
250 ordinal_type numPoints = numPoints_tri * numPoints_line;
251 ordinal_type spaceDim = cellTopo.getDimension();
252
253 ViewType<PointValueType,DeviceType> inputPointsTri = getView<PointValueType,DeviceType>("input points",numPoints_tri, 2);
254 ViewType<PointValueType,DeviceType> inputPointsLine = getView<PointValueType,DeviceType>("input points",numPoints_line,1);
255 PointTools::getLattice(inputPointsTri, triTopo, order, 0, POINTTYPE_EQUISPACED );
256 PointTools::getLattice(inputPointsLine, lineTopo, order, 0, POINTTYPE_EQUISPACED );
257
258 ViewType<PointValueType,DeviceType> inputPoints = getView<PointValueType,DeviceType>("input points",numPoints,spaceDim);
259
260 using ExecutionSpace = typename ViewType<PointValueType,DeviceType>::execution_space;
261
262 Kokkos::RangePolicy < ExecutionSpace > policy(0,numPoints_tri);
263 Kokkos::parallel_for( policy,
264 KOKKOS_LAMBDA (const ordinal_type &triPointOrdinal )
265 {
266 ordinal_type pointOrdinal = triPointOrdinal * numPoints_line;
267 for (ordinal_type linePointOrdinal=0; linePointOrdinal<numPoints_line; linePointOrdinal++)
268 {
269 inputPoints(pointOrdinal,0) = inputPointsTri( triPointOrdinal,0);
270 inputPoints(pointOrdinal,1) = inputPointsTri( triPointOrdinal,1);
271 inputPoints(pointOrdinal,2) = inputPointsLine(linePointOrdinal,0);
272 pointOrdinal++;
273 }
274 }
275 );
276
277 return inputPoints;
278 }
279 else
280 {
281 const ordinal_type order = numPoints_1D - 1;
282 ordinal_type numPoints = PointTools::getLatticeSize(cellTopo, order);
283 ordinal_type spaceDim = cellTopo.getDimension();
284
285 ViewType<PointValueType,DeviceType> inputPoints = getView<PointValueType,DeviceType>("input points",numPoints,spaceDim);
286 PointTools::getLattice(inputPoints, cellTopo, order, 0, POINTTYPE_EQUISPACED );
287
288 return inputPoints;
289 }
290 }
291
292 template<typename OutputValueType, typename DeviceType>
293 inline ViewType<OutputValueType,DeviceType> getOutputView(Intrepid2::EFunctionSpace fs, Intrepid2::EOperator op, int basisCardinality, int numPoints, int spaceDim)
294 {
295 switch (fs) {
296 case Intrepid2::FUNCTION_SPACE_HGRAD:
297 switch (op) {
298 case Intrepid2::OPERATOR_VALUE:
299 return getView<OutputValueType,DeviceType>("H^1 value output",basisCardinality,numPoints);
300 case Intrepid2::OPERATOR_GRAD:
301 return getView<OutputValueType,DeviceType>("H^1 derivative output",basisCardinality,numPoints,spaceDim);
302 case Intrepid2::OPERATOR_D1:
303 case Intrepid2::OPERATOR_D2:
304 case Intrepid2::OPERATOR_D3:
305 case Intrepid2::OPERATOR_D4:
306 case Intrepid2::OPERATOR_D5:
307 case Intrepid2::OPERATOR_D6:
308 case Intrepid2::OPERATOR_D7:
309 case Intrepid2::OPERATOR_D8:
310 case Intrepid2::OPERATOR_D9:
311 case Intrepid2::OPERATOR_D10:
312 {
313 const auto dkcard = getDkCardinality(op, spaceDim);
314 return getView<OutputValueType,DeviceType>("H^1 derivative output",basisCardinality,numPoints,dkcard);
315 }
316 default:
317 INTREPID2_TEST_FOR_EXCEPTION(true, std::invalid_argument, "Unsupported op/fs combination");
318 }
319 case Intrepid2::FUNCTION_SPACE_HCURL:
320 switch (op) {
321 case Intrepid2::OPERATOR_VALUE:
322 return getView<OutputValueType,DeviceType>("H(curl) value output",basisCardinality,numPoints,spaceDim);
323 case Intrepid2::OPERATOR_CURL:
324 if (spaceDim == 2)
325 return getView<OutputValueType,DeviceType>("H(curl) derivative output",basisCardinality,numPoints);
326 else if (spaceDim == 3)
327 return getView<OutputValueType,DeviceType>("H(curl) derivative output",basisCardinality,numPoints,spaceDim);
328 [[fallthrough]];
329 default:
330 INTREPID2_TEST_FOR_EXCEPTION(true, std::invalid_argument, "Unsupported op/fs combination");
331 }
332 case Intrepid2::FUNCTION_SPACE_HDIV:
333 switch (op) {
334 case Intrepid2::OPERATOR_VALUE:
335 return getView<OutputValueType,DeviceType>("H(div) value output",basisCardinality,numPoints,spaceDim);
336 case Intrepid2::OPERATOR_DIV:
337 return getView<OutputValueType,DeviceType>("H(div) derivative output",basisCardinality,numPoints);
338 default:
339 INTREPID2_TEST_FOR_EXCEPTION(true, std::invalid_argument, "Unsupported op/fs combination");
340 }
341
342 case Intrepid2::FUNCTION_SPACE_HVOL:
343 switch (op) {
344 case Intrepid2::OPERATOR_VALUE:
345 return getView<OutputValueType,DeviceType>("H(vol) value output",basisCardinality,numPoints);
346 case Intrepid2::OPERATOR_D1:
347 case Intrepid2::OPERATOR_D2:
348 case Intrepid2::OPERATOR_D3:
349 case Intrepid2::OPERATOR_D4:
350 case Intrepid2::OPERATOR_D5:
351 case Intrepid2::OPERATOR_D6:
352 case Intrepid2::OPERATOR_D7:
353 case Intrepid2::OPERATOR_D8:
354 case Intrepid2::OPERATOR_D9:
355 case Intrepid2::OPERATOR_D10:
356 {
357 const auto dkcard = getDkCardinality(op, spaceDim);
358 return getView<OutputValueType,DeviceType>("H(vol) derivative output",basisCardinality,numPoints,dkcard);
359 }
360 default:
361 INTREPID2_TEST_FOR_EXCEPTION(true, std::invalid_argument, "Unsupported op/fs combination");
362 }
363 default:
364 INTREPID2_TEST_FOR_EXCEPTION(true, std::invalid_argument, "Unsupported op/fs combination");
365 }
366 }
367
368 // ! This returns a vector whose entries are vector<int>s containing 1-3 polynomial orders from 1 up to and including those specified
369 // ! Intended for testing bases that support anisotropic polynomial degree, such as the hierarchical bases
370 inline std::vector< std::vector<int> > getBasisTestCasesUpToDegree(int spaceDim, int minDegree, int polyOrder_x, int polyOrder_y=-1, int polyOrder_z=-1)
371 {
372 std::vector<int> degrees(spaceDim);
373 degrees[0] = polyOrder_x;
374 if (spaceDim > 1) degrees[1] = polyOrder_y;
375 if (spaceDim > 2) degrees[2] = polyOrder_z;
376
377 int numCases = degrees[0];
378 for (unsigned d=1; d<degrees.size(); d++)
379 {
380 INTREPID2_TEST_FOR_EXCEPTION(degrees[d] < minDegree, std::invalid_argument, "Unsupported degree/minDegree combination");
381 numCases = numCases * (degrees[d] + 1 - minDegree);
382 }
383 std::vector< std::vector<int> > subBasisDegreeTestCases(numCases);
384 for (int caseOrdinal=0; caseOrdinal<numCases; caseOrdinal++)
385 {
386 std::vector<int> subBasisDegrees(degrees.size());
387 int caseRemainder = caseOrdinal;
388 for (int d=degrees.size()-1; d>=0; d--)
389 {
390 int subBasisDegree = caseRemainder % (degrees[d] + 1 - minDegree);
391 caseRemainder = caseRemainder / (degrees[d] + 1 - minDegree);
392 subBasisDegrees[d] = subBasisDegree + minDegree;
393 }
394 subBasisDegreeTestCases[caseOrdinal] = subBasisDegrees;
395 }
396 return subBasisDegreeTestCases;
397 }
398
400 template<class Functor, class Scalar, int rank>
401 typename ViewType<Scalar,DefaultTestDeviceType>::host_mirror_type copyFunctorToHostView(const Functor &deviceFunctor)
402 {
403 INTREPID2_TEST_FOR_EXCEPTION(rank != getFunctorRank(deviceFunctor), std::invalid_argument, "functor rank must match the template argument");
404
405 using DeviceType = DefaultTestDeviceType;
406 ViewType<Scalar,DeviceType> view;
407 const std::string label = "functor copy";
408 const auto &f = deviceFunctor;
409 switch (rank)
410 {
411 case 0:
412 view = getView<Scalar,DeviceType>(label);
413 break;
414 case 1:
415 view = getView<Scalar,DeviceType>(label, f.extent_int(0));
416 break;
417 case 2:
418 view = getView<Scalar,DeviceType>(label, f.extent_int(0), f.extent_int(1));
419 break;
420 case 3:
421 view = getView<Scalar,DeviceType>(label, f.extent_int(0), f.extent_int(1), f.extent_int(2));
422 break;
423 case 4:
424 view = getView<Scalar,DeviceType>(label, f.extent_int(0), f.extent_int(1), f.extent_int(2), f.extent_int(3));
425 break;
426 case 5:
427 view = getView<Scalar,DeviceType>(label, f.extent_int(0), f.extent_int(1), f.extent_int(2), f.extent_int(3), f.extent_int(4));
428 break;
429 case 6:
430 view = getView<Scalar,DeviceType>(label, f.extent_int(0), f.extent_int(1), f.extent_int(2), f.extent_int(3), f.extent_int(4), f.extent_int(5));
431 break;
432 case 7:
433 view = getView<Scalar,DeviceType>(label, f.extent_int(0), f.extent_int(1), f.extent_int(2), f.extent_int(3), f.extent_int(4), f.extent_int(5), f.extent_int(6));
434 break;
435 default:
436 INTREPID2_TEST_FOR_EXCEPTION(true, std::invalid_argument, "Unsupported functor rank");
437 }
438
439 int entryCount = view.size();
440
441 using ExecutionSpace = typename ViewType<Scalar,DeviceType>::execution_space;
442
443 using ViewIteratorScalar = Intrepid2::ViewIterator<ViewType<Scalar,DeviceType>, Scalar>;
444 using FunctorIteratorScalar = FunctorIterator<Functor, Scalar, rank>;
445
446 Kokkos::RangePolicy < ExecutionSpace > policy(0,entryCount);
447 Kokkos::parallel_for( policy,
448 KOKKOS_LAMBDA (const int &enumerationIndex )
449 {
450 ViewIteratorScalar vi(view);
451 FunctorIteratorScalar fi(f);
452
453 vi.setEnumerationIndex(enumerationIndex);
454 fi.setEnumerationIndex(enumerationIndex);
455
456 vi.set(fi.get());
457 }
458 );
459
460 auto hostView = Kokkos::create_mirror_view(view);
461 Kokkos::deep_copy(hostView, view);
462 return hostView;
463 }
464
465 template<class FunctorType, typename Scalar, int rank>
466 void printFunctor(const FunctorType &functor, std::ostream &out, const std::string &functorName = "")
467 {
468 auto functorHostCopy = copyFunctorToHostView<FunctorType, Scalar, rank>(functor);
469
470 std::string name = (functorName == "") ? "Functor" : functorName;
471
472 out << "\n******** " << name << " (rank " << rank << ") ********\n";
473 out << "dimensions: (";
474 for (int r=0; r<rank; r++)
475 {
476 out << functor.extent_int(r);
477 if (r<rank-1) out << ",";
478 }
479 out << ")\n";
480
481 ViewIterator<decltype(functorHostCopy),Scalar> vi(functorHostCopy);
482
483 bool moreEntries = true;
484 while (moreEntries)
485 {
486 Scalar value = vi.get();
487
488 auto location = vi.getLocation();
489 out << functorName << "(";
490 for (ordinal_type i=0; i<rank; i++)
491 {
492 out << location[i];
493 if (i<rank-1)
494 {
495 out << ",";
496 }
497 }
498 out << ") " << value << std::endl;
499
500 moreEntries = (vi.increment() != -1);
501 }
502 out << "\n";
503 }
504
505 template<class FunctorType>
506 void printFunctor1(const FunctorType &functor, std::ostream &out, const std::string &functorName = "")
507 {
508 using Scalar = typename std::remove_reference<decltype(functor(0))>::type;
509 printFunctor<FunctorType, Scalar, 1>(functor, out, functorName);
510 }
511
512 template<class FunctorType>
513 void printFunctor2(const FunctorType &functor, std::ostream &out, const std::string &functorName = "")
514 {
515 using Scalar = typename std::remove_const<typename std::remove_reference<decltype(functor(0,0))>::type>::type;
516 printFunctor<FunctorType, Scalar, 2>(functor, out, functorName);
517 }
518
519 template<class FunctorType>
520 void printFunctor3(const FunctorType &functor, std::ostream &out, const std::string &functorName = "")
521 {
522 using Scalar = typename std::remove_const<typename std::remove_reference<decltype(functor(0,0,0))>::type>::type;
523 printFunctor<FunctorType, Scalar, 3>(functor, out, functorName);
524 }
525
526 template<class FunctorType>
527 void printFunctor4(const FunctorType &functor, std::ostream &out, const std::string &functorName = "")
528 {
529 using Scalar = typename std::remove_const<typename std::remove_reference<decltype(functor(0,0,0,0))>::type>::type;
530 printFunctor<FunctorType, Scalar, 4>(functor, out, functorName);
531 }
532
533 template<class FunctorType>
534 void printFunctor5(const FunctorType &functor, std::ostream &out, const std::string &functorName = "")
535 {
536 using Scalar = typename std::remove_const<typename std::remove_reference<decltype(functor(0,0,0,0,0))>::type>::type;
537 printFunctor<FunctorType, Scalar, 5>(functor, out, functorName);
538 }
539
540 template<class FunctorType>
541 void printFunctor6(const FunctorType &functor, std::ostream &out, const std::string &functorName = "")
542 {
543 using Scalar = typename std::remove_const<typename std::remove_reference<decltype(functor(0,0,0,0,0,0))>::type>::type;
544 printFunctor<FunctorType, Scalar, 6>(functor, out, functorName);
545 }
546
547 template<class FunctorType>
548 void printFunctor7(const FunctorType &functor, std::ostream &out, const std::string &functorName = "")
549 {
550 using Scalar = typename std::remove_const<typename std::remove_reference<decltype(functor(0,0,0,0,0,0,0))>::type>::type;
551 printFunctor<FunctorType, Scalar, 7>(functor, out, functorName);
552 }
553
554 template<class View>
555 void printView(const View &view, std::ostream &out, const std::string &viewName = "")
556 {
557 using Scalar = typename View::value_type;
558 using HostView = typename View::host_mirror_type;
559 using HostViewIteratorScalar = Intrepid2::ViewIterator<HostView, Scalar>;
560
561 auto hostView = getHostCopy(view);
562
563 HostViewIteratorScalar vi(hostView);
564
565 bool moreEntries = (vi.nextIncrementRank() != -1);
566 while (moreEntries)
567 {
568 Scalar value = vi.get();
569
570 auto location = vi.getLocation();
571 out << viewName << "(";
572 for (unsigned i=0; i<getFunctorRank(view); i++)
573 {
574 out << location[i];
575 if (i<getFunctorRank(view)-1)
576 {
577 out << ",";
578 }
579 }
580 out << ") " << value << std::endl;
581
582 moreEntries = (vi.increment() != -1);
583 }
584 }
585
586 template <class FunctorType1, class FunctorType2, int rank, typename Scalar=typename FunctorType1::value_type, class ExecutionSpace = typename FunctorType1::execution_space>
588 testFloatingEquality(const FunctorType1 &functor1, const FunctorType2 &functor2, double relTol, double absTol, Teuchos::FancyOStream &out, bool &success,
589 std::string functor1Name = "Functor 1", std::string functor2Name = "Functor 2")
590 {
591 INTREPID2_TEST_FOR_EXCEPTION(true, std::invalid_argument, "testFloatingEquality() called on FunctorType1 or FunctorType2 that does not support the specified rank.\n");
592 }
593
595 template <class FunctorType1, class FunctorType2, int rank, typename Scalar=typename FunctorType1::value_type, class ExecutionSpace = typename FunctorType1::execution_space>
597 testFloatingEquality(const FunctorType1 &functor1, const FunctorType2 &functor2, double relTol, double absTol, Teuchos::FancyOStream &out, bool &success,
598 std::string functor1Name = "Functor 1", std::string functor2Name = "Functor 2")
599 {
600 static_assert( supports_rank<FunctorType1,rank>::value, "Functor1 must support the specified rank through operator().");
601 static_assert( supports_rank<FunctorType2,rank>::value, "Functor2 must support the specified rank through operator().");
602
603 using Functor1IteratorScalar = FunctorIterator<FunctorType1, Scalar, rank>;
604 using Functor2IteratorScalar = FunctorIterator<FunctorType2, Scalar, rank>;
605
606 // check that rank/size match
607 TEUCHOS_TEST_FOR_EXCEPTION(getFunctorRank(functor1) != rank, std::invalid_argument, "functor1 and functor2 must agree in rank"); // Kokkos::View does not provide rank() method; getFunctorRank() provides common interface
608 TEUCHOS_TEST_FOR_EXCEPTION(getFunctorRank(functor2) != rank, std::invalid_argument, "functor1 and functor2 must agree in rank"); // Kokkos::View does not provide rank() method; getFunctorRank() provides common interface
609
610 int entryCount = 1;
611 for (unsigned i=0; i<getFunctorRank(functor1); i++)
612 {
613 TEUCHOS_TEST_FOR_EXCEPTION(functor1.extent_int(i) != functor2.extent_int(i), std::invalid_argument,
614 "functor1 and functor2 must agree in size in each dimension; functor1 has extent "
615 + std::to_string(functor1.extent_int(i)) + " in dimension " + std::to_string(i)
616 + "; functor2 has extent " + std::to_string(functor2.extent_int(i)) );
617 entryCount *= functor1.extent_int(i);
618 }
619 if (entryCount == 0) return; // nothing to test
620
621 ViewType<bool,ExecutionSpace> valuesMatch = getView<bool,ExecutionSpace>("valuesMatch", entryCount);
622
623 Kokkos::RangePolicy < ExecutionSpace > policy(0,entryCount);
624 Kokkos::parallel_for( policy,
625 KOKKOS_LAMBDA (const int &enumerationIndex )
626 {
627 Functor1IteratorScalar vi1(functor1);
628 Functor2IteratorScalar vi2(functor2);
629
630 vi1.setEnumerationIndex(enumerationIndex);
631 vi2.setEnumerationIndex(enumerationIndex);
632
633 const Scalar & value1 = vi1.get();
634 const Scalar & value2 = vi2.get();
635
636 bool errMeetsTol = errMeetsAbsAndRelTol(value1, value2, relTol, absTol);
637 valuesMatch(enumerationIndex) = errMeetsTol;
638 }
639 );
640
641 int failureCount = 0;
642 Kokkos::RangePolicy<ExecutionSpace > reducePolicy(0, entryCount);
643 Kokkos::parallel_reduce( reducePolicy,
644 KOKKOS_LAMBDA( const int &enumerationIndex, int &reducedValue )
645 {
646 reducedValue += valuesMatch(enumerationIndex) ? 0 : 1;
647 }, failureCount);
648
649 const bool allValuesMatch = (failureCount == 0);
650 success = success && allValuesMatch;
651
652 if (!allValuesMatch)
653 {
654 // copy functors to host views
655 auto functor1CopyHost = copyFunctorToHostView<FunctorType1,Scalar,rank>(functor1);
656 auto functor2CopyHost = copyFunctorToHostView<FunctorType2,Scalar,rank>(functor2);
657
658 auto valuesMatchHost = getHostCopy(valuesMatch);
659
660 FunctorIterator<decltype(functor1CopyHost),Scalar,rank> vi1(functor1CopyHost);
661 FunctorIterator<decltype(functor2CopyHost),Scalar,rank> vi2(functor2CopyHost);
662 Intrepid2::ViewIterator<decltype(valuesMatchHost), bool> viMatch(valuesMatchHost);
663
664 bool moreEntries = true;
665 while (moreEntries)
666 {
667 bool errMeetsTol = viMatch.get();
668
669 if (!errMeetsTol)
670 {
671 const Scalar value1 = vi1.get();
672 const Scalar value2 = vi2.get();
673
674 success = false;
675 auto location = vi1.getLocation();
676 out << "At location (";
677 for (unsigned i=0; i<getFunctorRank(functor1); i++)
678 {
679 out << location[i];
680 if (i<getFunctorRank(functor1)-1)
681 {
682 out << ",";
683 }
684 }
685 out << ") " << functor1Name << " value != " << functor2Name << " value (";
686 out << value1 << " != " << value2 << "); diff is " << std::abs(value1-value2) << std::endl;
687 }
688
689 moreEntries = (vi1.increment() != -1);
690 moreEntries = moreEntries && (vi2.increment() != -1);
691 moreEntries = moreEntries && (viMatch.increment() != -1);
692 }
693 }
694 }
695
696 template <class ViewType, class FunctorType>
697 void testFloatingEquality1(const ViewType &view, const FunctorType &functor, double relTol, double absTol, Teuchos::FancyOStream &out, bool &success,
698 std::string view1Name = "View", std::string view2Name = "Functor")
699 {
700 testFloatingEquality<ViewType, FunctorType, 1>(view, functor, relTol, absTol, out, success, view1Name, view2Name);
701 }
702
703 template <class ViewType, class FunctorType>
704 void testFloatingEquality2(const ViewType &view, const FunctorType &functor, double relTol, double absTol, Teuchos::FancyOStream &out, bool &success,
705 std::string view1Name = "View", std::string view2Name = "Functor")
706 {
707 testFloatingEquality<ViewType, FunctorType, 2>(view, functor, relTol, absTol, out, success, view1Name, view2Name);
708 }
709
710 template <class ViewType, class FunctorType>
711 void testFloatingEquality3(const ViewType &view, const FunctorType &functor, double relTol, double absTol, Teuchos::FancyOStream &out, bool &success,
712 std::string view1Name = "View", std::string view2Name = "Functor")
713 {
714 testFloatingEquality<ViewType, FunctorType, 3>(view, functor, relTol, absTol, out, success, view1Name, view2Name);
715 }
716
717 template <class ViewType, class FunctorType>
718 void testFloatingEquality4(const ViewType &view, const FunctorType &functor, double relTol, double absTol, Teuchos::FancyOStream &out, bool &success,
719 std::string view1Name = "View", std::string view2Name = "Functor")
720 {
721 testFloatingEquality<ViewType, FunctorType, 4>(view, functor, relTol, absTol, out, success, view1Name, view2Name);
722 }
723
724 template <class ViewType, class FunctorType>
725 void testFloatingEquality5(const ViewType &view, const FunctorType &functor, double relTol, double absTol, Teuchos::FancyOStream &out, bool &success,
726 std::string view1Name = "View", std::string view2Name = "Functor")
727 {
728 testFloatingEquality<ViewType, FunctorType, 5>(view, functor, relTol, absTol, out, success, view1Name, view2Name);
729 }
730
731 template <class ViewType, class FunctorType>
732 void testFloatingEquality6(const ViewType &view, const FunctorType &functor, double relTol, double absTol, Teuchos::FancyOStream &out, bool &success,
733 std::string view1Name = "View", std::string view2Name = "Functor")
734 {
735 testFloatingEquality<ViewType, FunctorType, 6>(view, functor, relTol, absTol, out, success, view1Name, view2Name);
736 }
737
738 template <class ViewType, class FunctorType>
739 void testFloatingEquality7(const ViewType &view, const FunctorType &functor, double relTol, double absTol, Teuchos::FancyOStream &out, bool &success,
740 std::string view1Name = "View", std::string view2Name = "Functor")
741 {
742 testFloatingEquality<ViewType, FunctorType, 7>(view, functor, relTol, absTol, out, success, view1Name, view2Name);
743 }
744
745 template <class ViewType1, class ViewType2>
746 void testViewFloatingEquality(const ViewType1 &view1, const ViewType2 &view2, double relTol, double absTol, Teuchos::FancyOStream &out, bool &success,
747 std::string view1Name = "View 1", std::string view2Name = "View 2")
748 {
749 // check that rank/size match
750 TEUCHOS_TEST_FOR_EXCEPTION(view1.rank() != view2.rank(), std::invalid_argument, "views must agree in rank");
751 for (unsigned i=0; i<view1.rank(); i++)
752 {
753 TEUCHOS_TEST_FOR_EXCEPTION(view1.extent_int(i) != view2.extent_int(i), std::invalid_argument, "views must agree in size in each dimension");
754 }
755
756 if (view1.size() == 0) return; // nothing to test
757
758 const int rank = view1.rank();
759 switch (rank)
760 {
761 case 0: testFloatingEquality<ViewType1, ViewType2, 0>(view1, view2, relTol, absTol, out, success, view1Name, view2Name); break;
762 case 1: testFloatingEquality<ViewType1, ViewType2, 1>(view1, view2, relTol, absTol, out, success, view1Name, view2Name); break;
763 case 2: testFloatingEquality<ViewType1, ViewType2, 2>(view1, view2, relTol, absTol, out, success, view1Name, view2Name); break;
764 case 3: testFloatingEquality<ViewType1, ViewType2, 3>(view1, view2, relTol, absTol, out, success, view1Name, view2Name); break;
765 case 4: testFloatingEquality<ViewType1, ViewType2, 4>(view1, view2, relTol, absTol, out, success, view1Name, view2Name); break;
766 case 5: testFloatingEquality<ViewType1, ViewType2, 5>(view1, view2, relTol, absTol, out, success, view1Name, view2Name); break;
767 case 6: testFloatingEquality<ViewType1, ViewType2, 6>(view1, view2, relTol, absTol, out, success, view1Name, view2Name); break;
768 case 7: testFloatingEquality<ViewType1, ViewType2, 7>(view1, view2, relTol, absTol, out, success, view1Name, view2Name); break;
769 default: INTREPID2_TEST_FOR_EXCEPTION(true, std::invalid_argument, "Unsupported rank");
770 }
771 }
772
773} // namespace Intrepid2
774
775#ifdef HAVE_INTREPID2_SACADO
776using Sacado_Fad_DFadType = Sacado::Fad::DFad<double>;
777#define INTREPID2_POINTSCALAR_TEST_INSTANT(GROUP_NAME, TEST_NAME) \
778\
779TEUCHOS_UNIT_TEST_TEMPLATE_1_INSTANT( GROUP_NAME, TEST_NAME, double ) \
780\
781TEUCHOS_UNIT_TEST_TEMPLATE_1_INSTANT( GROUP_NAME, TEST_NAME, Sacado_Fad_DFadType ) \
782
783#define INTREPID2_OUTPUTSCALAR_POINTSCALAR_TEST_INSTANT(GROUP_NAME, TEST_NAME) \
784\
785TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( GROUP_NAME, TEST_NAME, double, double ) \
786\
787TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( GROUP_NAME, TEST_NAME, Sacado_Fad_DFadType, Sacado_Fad_DFadType ) \
788
789#else
790#define INTREPID2_POINTSCALAR_TEST_INSTANT(GROUP_NAME, TEST_NAME) \
791\
792TEUCHOS_UNIT_TEST_TEMPLATE_1_INSTANT( GROUP_NAME, TEST_NAME, double ) \
793
794#define INTREPID2_OUTPUTSCALAR_POINTSCALAR_TEST_INSTANT(GROUP_NAME, TEST_NAME) \
795\
796TEUCHOS_UNIT_TEST_TEMPLATE_2_INSTANT( GROUP_NAME, TEST_NAME, double, double ) \
797
798#endif
799
800#endif /* Intrepid2_TestUtils_h */
Header file for the abstract base class Intrepid2::Basis.
Teuchos::RCP< Basis< DeviceType, OutputType, PointType > > BasisPtr
Basis Pointer.
KOKKOS_INLINE_FUNCTION ordinal_type getDkCardinality(const EOperator operatorType, const ordinal_type spaceDim)
Returns cardinality of Dk, i.e., the number of all derivatives of order k.
Stateless class representing a family of basis functions, templated on H(vol) and H(grad) on the line...
Defines the Intrepid2::FunctorIterator class, as well as the Intrepid2::functor_returns_ref SFINAE he...
Stateless classes that act as factories for two families of hierarchical bases. HierarchicalBasisFami...
Header file for Intrepid2::PointTools class to provide utilities for barycentric coordinates,...
Header file to include all Sacado headers that are required if using Intrepid2 with Sacado types.
ViewType< PointValueType, DeviceType > getInputPointsView(shards::CellTopology &cellTopo, int numPoints_1D)
Returns a DynRankView containing regularly-spaced points on the specified cell topology.
typename Kokkos::DefaultExecutionSpace::device_type DefaultTestDeviceType
Default Kokkos::Device to use for tests; depends on platform.
ViewType< Scalar, DefaultTestDeviceType >::host_mirror_type copyFunctorToHostView(const Functor &deviceFunctor)
Copy the values for the specified functor.
constexpr int MAX_FAD_DERIVATIVES_FOR_TESTS
Maximum number of derivatives to track for Fad types in tests.
const Teuchos::ScalarTraits< Scalar >::magnitudeType smallNumber()
Use Teuchos small number determination on host; pass this to Intrepid2::relErr() on device.
KOKKOS_INLINE_FUNCTION bool relErrMeetsTol(const Scalar1 &s1, const Scalar2 &s2, const typename Teuchos::ScalarTraits< typename std::common_type< Scalar1, Scalar2 >::type >::magnitudeType &smallNumber, const double &tol)
Adapted from Teuchos::relErr(); for use in code that may be executed on device.
Header function for Intrepid2::Util class and other utility functions.
enable_if_t< has_rank_method< Functor >::value, unsigned > KOKKOS_INLINE_FUNCTION getFunctorRank(const Functor &functor)
A family of basis functions, constructed from H(vol) and H(grad) bases on the line.
essentially, a read-only variant of ViewIterator, for a general functor (extent_int() and rank() supp...
Basis defining integrated Legendre basis on the line, a polynomial subspace of H(grad) on the line.
Basis defining Legendre basis on the line, a polynomial subspace of L^2 (a.k.a. H(vol)) on the line.
static ordinal_type getLatticeSize(const shards::CellTopology cellType, const ordinal_type order, const ordinal_type offset=0)
Computes the number of points in a lattice of a given order on a simplex (currently disabled for othe...
static void getLattice(Kokkos::DynRankView< pointValueType, pointProperties... > points, const shards::CellTopology cellType, const ordinal_type order, const ordinal_type offset=0, const EPointType pointType=POINTTYPE_EQUISPACED)
Computes a lattice of points of a given order on a reference simplex, quadrilateral or hexahedron (cu...
A helper class that allows iteration over some part of a Kokkos View, while allowing the calling code...
SFINAE helper to detect whether a type supports a rank-integral-argument operator().
Helper to get Scalar[*+] where the number of *'s matches the given rank.