From dd03b634d29cc1f22cccc640e292b536e3b0ea9f Mon Sep 17 00:00:00 2001 From: Preston Waters Date: Wed, 6 Mar 2024 09:29:35 -0500 Subject: [PATCH] formatting testSerialize --- test/testSerialize.cpp | 76 +++++++++++++++++++++++------------------- 1 file changed, 42 insertions(+), 34 deletions(-) diff --git a/test/testSerialize.cpp b/test/testSerialize.cpp index 5d59de5..b122f5e 100644 --- a/test/testSerialize.cpp +++ b/test/testSerialize.cpp @@ -1,15 +1,15 @@ -#include "MeshField.hpp" #include "CabanaController.hpp" #include "KokkosController.hpp" +#include "MeshField.hpp" #include "MeshField_Macros.hpp" #include "MeshField_Utility.hpp" #include #include -#include -#include #include +#include +#include using ExecutionSpace = Kokkos::DefaultExecutionSpace; using MemorySpace = Kokkos::DefaultExecutionSpace::memory_space; @@ -18,9 +18,11 @@ int main(int argc, char *argv[]) { Kokkos::ScopeGuard scope_guard(argc, argv); const int N = 10; - using kok1 = Controller::KokkosController; - kok1 c1({N,N,N,N,N,N,N,N,N,N,N,N,N,N,N}); - + using kok1 = + Controller::KokkosController; + kok1 c1({N, N, N, N, N, N, N, N, N, N, N, N, N, N, N}); + MeshField::MeshField mf(c1); MeshField::Field field1 = mf.makeField<0>(); MeshField::Field field2 = mf.makeField<1>(); @@ -28,48 +30,54 @@ int main(int argc, char *argv[]) { MeshField::Field field4 = mf.makeField<3>(); MeshField::Field field5 = mf.makeField<4>(); - Kokkos::View view1("1",N); - Kokkos::View view2("2",N,N); - Kokkos::View view3("3",N,N,N); - Kokkos::View view4("4",N,N,N,N); - Kokkos::View view5("5",N,N,N,N,N); + Kokkos::View view1("1", N); + Kokkos::View view2("2", N, N); + Kokkos::View view3("3", N, N, N); + Kokkos::View view4("4", N, N, N, N); + Kokkos::View view5("5", N, N, N, N, N); - Kokkos::Array start = MeshFieldUtil::to_kokkos_array<5>({0,0,0,0,0}); - Kokkos::Array end = MeshFieldUtil::to_kokkos_array<5>({N,N,N,N,N}); - Kokkos::MDRangePolicy> p(start,end); + Kokkos::Array start = MeshFieldUtil::to_kokkos_array<5>({0, 0, 0, 0, 0}); + Kokkos::Array end = MeshFieldUtil::to_kokkos_array<5>({N, N, N, N, N}); + Kokkos::MDRangePolicy> p(start, end); - Kokkos::parallel_for( "",p,KOKKOS_LAMBDA(const int& i,const int& j, const int& k, const int& l, const int& m){ - view1(i) = i; - view2(i,j) = i+j; - view3(i,j,k) = i+j+k; - view4(i,j,k,l) = i+j+k+l; - view5(i,j,k,l,m) = i+j+k+l+m; - }); + Kokkos::parallel_for( + "", p, + KOKKOS_LAMBDA(const int &i, const int &j, const int &k, const int &l, + const int &m) { + view1(i) = i; + view2(i, j) = i + j; + view3(i, j, k) = i + j + k; + view4(i, j, k, l) = i + j + k + l; + view5(i, j, k, l, m) = i + j + k + l + m; + }); - mf.setField(field1,view1); - mf.setField(field2,view2); - mf.setField(field3,view3); - mf.setField(field4,view4); - mf.setField(field5,view5); + mf.setField(field1, view1); + mf.setField(field2, view2); + mf.setField(field3, view3); + mf.setField(field4, view4); + mf.setField(field5, view5); auto serialized1 = field1.serialize(); auto serialized2 = field2.serialize(); auto serialized3 = field3.serialize(); auto serialized4 = field4.serialize(); auto serialized5 = field5.serialize(); - + field1.deserialize(serialized1); field2.deserialize(serialized2); field3.deserialize(serialized3); field4.deserialize(serialized4); field5.deserialize(serialized5); - Kokkos::parallel_for( "",p,KOKKOS_LAMBDA(const int& i,const int& j, const int& k, const int& l, const int& m){ - assert(view1(i) == field1(i)); - assert(view2(i, j) == field2(i, j)); - assert(view3(i, j, k) == field3(i, j, k)); - assert(view4(i, j, k, l) == field4(i, j, k, l)); - assert(view5(i, j, k, l, m) == field5(i, j, k, l, m)); - }); + Kokkos::parallel_for( + "", p, + KOKKOS_LAMBDA(const int &i, const int &j, const int &k, const int &l, + const int &m) { + assert(view1(i) == field1(i)); + assert(view2(i, j) == field2(i, j)); + assert(view3(i, j, k) == field3(i, j, k)); + assert(view4(i, j, k, l) == field4(i, j, k, l)); + assert(view5(i, j, k, l, m) == field5(i, j, k, l, m)); + }); return 0; }