user has full control of state_vars now

lanl · Feb 7, 2024 · 20dad80 · 20dad80
1 parent c6b8506
commit 20dad80
Show file tree

Hide file tree

Showing 8 changed files with 11 additions and 9 deletions.
diff --git a/src/EVP/src/init_state_vars.cpp b/src/EVP/src/init_state_vars.cpp
@@ -32,7 +32,7 @@ std::vector<std::string> split (std::string s, std::string delimiter);
 void init_state_vars(
   const DCArrayKokkos <material_t> &material,
   const DViewCArrayKokkos <size_t> &elem_mat_id,
-  const DCArrayKokkos <double> &state_vars,
+  DCArrayKokkos <double> &state_vars,
   const DCArrayKokkos <double> &global_vars,
   const DCArrayKokkos <double> &elem_user_output_vars,
   const size_t num_elems)
@@ -412,7 +412,7 @@ void init_state_vars(
   //});
 	}
   Kokkos::fence();
-  //state_vars.update_host();
+  state_vars.update_host();
 
     printf("user_mat_init completed\n");
 

diff --git a/src/EVPFFT/src/Fierro-EVPFFT-Link/init_state_vars.cpp b/src/EVPFFT/src/Fierro-EVPFFT-Link/init_state_vars.cpp
@@ -6,7 +6,7 @@ using namespace mtr;
 void init_state_vars(
   const DCArrayKokkos <material_t> &material,
   const DViewCArrayKokkos <size_t> &elem_mat_id,
-  const DCArrayKokkos <double> &state_vars,
+  DCArrayKokkos <double> &state_vars,
   const DCArrayKokkos <double> &global_vars,
   const DCArrayKokkos <double> &elem_user_output_vars,
   const size_t num_elems)
@@ -20,6 +20,7 @@ void init_state_vars(
             state_vars.host(elem_gid,var) = 0.0;
         }
     }
+    state_vars.update_device();
 
 	  return;
 }
diff --git a/src/LS-EVPFFT/src/Fierro-LS-EVPFFT-Link/init_state_vars.cpp b/src/LS-EVPFFT/src/Fierro-LS-EVPFFT-Link/init_state_vars.cpp
@@ -6,7 +6,7 @@ using namespace mtr;
 void init_state_vars(
   const DCArrayKokkos <material_t> &material,
   const DViewCArrayKokkos <size_t> &elem_mat_id,
-  const DCArrayKokkos <double> &state_vars,
+  DCArrayKokkos <double> &state_vars,
   const DCArrayKokkos <double> &global_vars,
   const DCArrayKokkos <double> &elem_user_output_vars,
   const size_t num_elems)
@@ -20,6 +20,7 @@ void init_state_vars(
             state_vars.host(elem_gid,var) = 0.0;
         }
     }
+    state_vars.update_device();
 
 	  return;
 }
diff --git a/src/Parallel-Solvers/Material-Models/Ideal-Gas-Models/init_state_vars.cpp b/src/Parallel-Solvers/Material-Models/Ideal-Gas-Models/init_state_vars.cpp
@@ -6,7 +6,7 @@ using namespace mtr;
 void init_state_vars(
   const DCArrayKokkos <material_t> &material,
   const DViewCArrayKokkos <size_t> &elem_mat_id,
-  const DCArrayKokkos <double> &state_vars,
+  DCArrayKokkos <double> &state_vars,
   const DCArrayKokkos <double> &global_vars,
   const DCArrayKokkos <double> &elem_user_output_vars,
   const size_t num_elems)
@@ -20,6 +20,7 @@ void init_state_vars(
             state_vars.host(elem_gid,var) = 0.0;
         }
     }
+    state_vars.update_device();
 
 	  return;
 }
diff --git a/src/Parallel-Solvers/Material-Models/User-Material-Models/init_state_vars.cpp b/src/Parallel-Solvers/Material-Models/User-Material-Models/init_state_vars.cpp
@@ -6,7 +6,7 @@ using namespace mtr;
 void init_state_vars(
   const DCArrayKokkos <material_t> &material,
   const DViewCArrayKokkos <size_t> &elem_mat_id,
-  const DCArrayKokkos <double> &state_vars,
+  DCArrayKokkos <double> &state_vars,
   const DCArrayKokkos <double> &global_vars,
   const DCArrayKokkos <double> &elem_user_output_vars,
   const size_t num_elems)
@@ -20,6 +20,7 @@ void init_state_vars(
             state_vars.host(elem_gid,var) = 0.0;
         }
     }
+    state_vars.update_device();
 
 	  return;
 }
diff --git a/src/Parallel-Solvers/Material-Models/material_models.h b/src/Parallel-Solvers/Material-Models/material_models.h
@@ -143,7 +143,7 @@ struct strength_t {
 void init_state_vars(
   const DCArrayKokkos <material_t> &material,
   const DViewCArrayKokkos <size_t> &elem_mat_id,
-  const DCArrayKokkos <double> &state_vars,
+  DCArrayKokkos <double> &state_vars,
   const DCArrayKokkos <double> &global_vars,
   const DCArrayKokkos <double> &elem_user_output_vars,
   const size_t num_elems);

diff --git a/...rallel-Solvers/Parallel-Explicit/Dynamic_Elastic_Solver/FEA_Module_Dynamic_Elasticity.cpp b/...rallel-Solvers/Parallel-Explicit/Dynamic_Elastic_Solver/FEA_Module_Dynamic_Elasticity.cpp
@@ -1054,7 +1054,6 @@ void FEA_Module_Dynamic_Elasticity::setup(){
                     global_vars,
                     elem_user_output_vars,
                     rnum_elem);
-    state_vars.update_device();
 
     // initialize strength model
     init_strength_model(elem_strength,

diff --git a/src/Parallel-Solvers/Parallel-Explicit/SGH_Solver/FEA_Module_SGH.cpp b/src/Parallel-Solvers/Parallel-Explicit/SGH_Solver/FEA_Module_SGH.cpp
@@ -1094,7 +1094,6 @@ void FEA_Module_SGH::setup(){
                     global_vars,
                     elem_user_output_vars,
                     rnum_elem);
-    state_vars.update_device();
 
     // initialize strength model
     init_strength_model(elem_strength,