test all structures

SCOREC · Nov 24, 2024 · 6ef4dac · 6ef4dac
1 parent e2aaf60
commit 6ef4dac
Showing 1 changed file with 64 additions and 46 deletions.
diff --git a/particle_structs/test/smoke_test_particle.cpp b/particle_structs/test/smoke_test_particle.cpp
@@ -2,6 +2,8 @@
 #include "read_particles.hpp"
 #include "team_policy.hpp"
 
+void testStructure(PS* structure, int comm_rank, int comm_size, kkLidView failures);
+
 int main(int argc, char* argv[]) {
   Kokkos::initialize(argc, argv);
   MPI_Init(&argc, &argv);
@@ -14,6 +16,8 @@ int main(int argc, char* argv[]) {
 
     int num_elems = 5;
     int num_ptcls = 5;
+    lid_t V = 1024;
+    lid_t sigma = INT_MAX;
     kkLidView ppe = kkLidView("particles_per_element", num_elems);
     kkGidView element_gids = kkGidView("elemnt_gids", num_elems);
     kkLidView particle_elements = kkLidView("element_of_particle", num_ptcls);
@@ -25,61 +29,75 @@ int main(int argc, char* argv[]) {
     ps::hostToDevice(ppe, ppe_h);
     ps::hostToDevice(element_gids, element_gids_h);
     ps::hostToDevice(particle_elements, particle_elements_h);
+    Kokkos::TeamPolicy<ExeSpace> policy = pumipic::TeamPolicyAuto(num_elems,32);
     kkLidView failures("fails", 1);
 
-    //Init
-    Kokkos::TeamPolicy<ExeSpace> policy = pumipic::TeamPolicyAuto(num_elems,32);
-    PS* structure = new ps::DPS<Types, MemSpace>(policy, num_elems, num_ptcls, ppe,
-                                      element_gids, particle_elements, particle_info);
+    PS* structure = new ps::SellCSigma<Types, MemSpace>(policy, sigma, V, num_elems, num_ptcls, ppe,
+                                                 element_gids, particle_elements, particle_info);
+    testStructure(structure, comm_rank, comm_size, failures);
+    delete structure;
 
-    auto checkInit = PS_LAMBDA(const lid_t& e, const lid_t& p, const bool& mask) {
-      if (mask && e != p){
-        printf("Structure not initalized at Elm: %d, Particle: %d\n", e, p);
-        failures(0) = 1;
-      }
-    };
-    ps::parallel_for(structure, checkInit, "checkInit");
-
-    //Rebuild
-    kkLidView new_element("new_element", structure->capacity());
-    auto setNewElm = PS_LAMBDA(const lid_t& e, const lid_t& p, const bool& mask) {
-      if (mask) new_element(p) = comm_rank;
-      else new_element(p) = -1;
-    };
-    ps::parallel_for(structure, setNewElm, "setNewElm");
-    structure->rebuild(new_element);
-
-    auto checkRebuild = PS_LAMBDA(const lid_t& e, const lid_t& p, const bool& mask) {
-      if (mask && e != comm_rank){
-        printf("Particle not set to correct element at Ptcl: %d, Elm: %d\n", p, e);
-        failures(0) = 1;
-      }
-    };
-    ps::parallel_for(structure, checkRebuild, "checkRebuild");
-
-    //Migrate
-    kkLidView new_process("new_process", structure->capacity());
-    auto setNewProcess = PS_LAMBDA(const lid_t& e, const lid_t& p, const bool& mask) {
-      if (mask) new_process(p) = (comm_rank + 1) % comm_size;
-      else new_process(p) = comm_rank;
-    };
-    ps::parallel_for(structure, setNewProcess, "setNewProcess");
-    structure->migrate(new_element, new_process);
-
-    auto checkMigrate = PS_LAMBDA(const lid_t& e, const lid_t& p, const bool& mask) {
-      if (mask && e == comm_rank){
-        printf("Particle in correct rank at Ptcl: %d, Elm: %d\n", p, e);
-        failures(0) = 1;
-      }
-    };
-    ps::parallel_for(structure, checkMigrate, "checkMigrate");
+    structure = new ps::CSR<Types, MemSpace>(policy, num_elems, num_ptcls, ppe,
+                                          element_gids, particle_elements, particle_info);
+    testStructure(structure, comm_rank, comm_size, failures);
+    delete structure;
 
-    test_failed = ps::getLastValue(failures);
+    structure = new ps::DPS<Types, MemSpace>(policy, num_elems, num_ptcls, ppe,
+                                      element_gids, particle_elements, particle_info);
+    testStructure(structure, comm_rank, comm_size, failures);
     delete structure;
+
+    test_failed = ps::getLastValue(failures);
     ps::destroyViews<Types>(particle_info);
   }
   Kokkos::finalize();
   MPI_Finalize();
 
   if (test_failed) exit(1);
+}
+
+
+void testStructure(PS* structure, int comm_rank, int comm_size, kkLidView failures) {
+  //Init
+  auto checkInit = PS_LAMBDA(const lid_t& e, const lid_t& p, const bool& mask) {
+    if (mask && e != p){
+      printf("Structure not initalized at Particle: %d, Elm: %d\n", p, e);
+      failures(0) = 1;
+    }
+  };
+  ps::parallel_for(structure, checkInit, "checkInit");
+
+  //Rebuild
+  kkLidView new_element("new_element", structure->capacity());
+  auto setNewElm = PS_LAMBDA(const lid_t& e, const lid_t& p, const bool& mask) {
+    if (mask) new_element(p) = comm_rank;
+    else new_element(p) = -1;
+  };
+  ps::parallel_for(structure, setNewElm, "setNewElm");
+  structure->rebuild(new_element);
+
+  auto checkRebuild = PS_LAMBDA(const lid_t& e, const lid_t& p, const bool& mask) {
+    if (mask && e != comm_rank){
+      printf("Structure failed to rebuild at Ptcl: %d, Elm: %d\n", p, e);
+      failures(0) = 1;
+    }
+  };
+  ps::parallel_for(structure, checkRebuild, "checkRebuild");
+
+  //Migrate
+  kkLidView new_process("new_process", structure->capacity());
+  auto setNewProcess = PS_LAMBDA(const lid_t& e, const lid_t& p, const bool& mask) {
+    if (mask) new_process(p) = (comm_rank + 1) % comm_size;
+    else new_process(p) = comm_rank;
+  };
+  ps::parallel_for(structure, setNewProcess, "setNewProcess");
+  structure->migrate(new_element, new_process);
+
+  auto checkMigrate = PS_LAMBDA(const lid_t& e, const lid_t& p, const bool& mask) {
+    if (mask && e == comm_rank){
+      printf("Structure failed to migrate ranks at Ptcl: %d, Elm: %d\n", p, e);
+      failures(0) = 1;
+    }
+  };
+  ps::parallel_for(structure, checkMigrate, "checkMigrate");
 }