Thermomechanical in work

Author: niravshah241

demo/thermomechanical_dlrbnicsx/dlrbnicsx_thermal.py

@@ -257,9 +257,10 @@ def norm_error(self, u, v):
     mu_ref = [0.6438, 0.4313, 1., 0.5]  # reference geometry
     mu = [0.8, 0.55, 0.8, 0.4]  # Parametric geometry
 
-    pod_samples = [3, 4, 3, 4]
-    ann_samples = [4, 3, 4, 3]
-    error_analysis_samples = [2, 2, 2, 2]
+    para_dim = 4
+    mechanical_ann_input_samples_num = 640
+    mechanical_error_analysis_samples_num = 144
+    num_snapshots = 700
 
     # FEM solve
     thermal_problem_parametric = \
@@ -290,7 +291,7 @@ def generate_training_set(num_samples, para_dim):
         training_set[:, 3] = (0.60 - 0.40) * training_set[:, 3] + 0.40
         return training_set
 
-    thermal_training_set = rbnicsx.io.on_rank_zero(mesh.comm, generate_training_set)
+    thermal_training_set = generate_training_set(num_snapshots, para_dim)
 
     Nmax = 30
 
@@ -379,7 +380,7 @@ def generate_ann_output_set(problem, reduced_problem,
 
 
     # Training dataset
-    thermal_ann_input_set = generate_ann_input_set(samples=ann_samples)
+    thermal_ann_input_set = generate_ann_input_set(mechanical_ann_input_samples_num)
     # np.random.shuffle(thermal_ann_input_set)
     thermal_ann_output_set = \
         generate_ann_output_set(thermal_problem_parametric,
@@ -469,7 +470,7 @@ def generate_ann_output_set(problem, reduced_problem,
     print("\n")
     print("Generating error analysis (only input/parameters) dataset")
     print("\n")
-    thermal_error_analysis_set = generate_ann_input_set(samples=error_analysis_samples)
+    thermal_error_analysis_set = generate_ann_input_set(mechanical_error_analysis_samples_num)
     thermal_error_numpy = np.zeros(thermal_error_analysis_set.shape[0])
 
     for i in range(thermal_error_analysis_set.shape[0]):

demo/thermomechanical_dlrbnicsx/mesh_data/mesh.py

@@ -1,6 +1,6 @@
+import dolfinx
 import gmsh
 from mpi4py import MPI
-import dolfinx
 
 gmsh.initialize('', False)
 gdim = 2
@@ -146,7 +146,7 @@
 
 gmsh.write("mesh.msh")
 
-gmsh.fltk.run()
+# gmsh.fltk.run()
 
 gmsh.finalize()
 

demo/thermomechanical_dlrbnicsx/test_sbatch_thermal

@@ -0,0 +1,23 @@
+#!/bin/bash
+#SBATCH -J test_job
+#SBATCH -A WELLS-SL3-CPU
+#SBATCH --output=test_job_%A_%a.out
+#SBATCH --error=test_job_%A_%a.err
+
+#SBATCH --nodes=1
+#SBATCH --ntasks=1
+#SBATCH --time=01:00:00
+#SBATCH --mem=3420mb
+
+#SBATCH -p icelake
+#SBATCH --mail-type=ALL
+
+module purge
+source ../fenics_branch/share/spack/setup-env.sh
+spack env activate fenicsx-env
+spack env status
+spack load [email protected]
+cd mesh_data
+mpiexec -n 1 python3 mesh.py
+cd ..
+mpiexec -n 1 python3 dlrbnicsx_thermal.py

demo/thermomechanical_dlrbnicsx/test_sbatch_thermomechanical

@@ -0,0 +1,23 @@
+#!/bin/bash
+#SBATCH -J test_job
+#SBATCH -A WELLS-SL3-CPU
+#SBATCH --output=test_job_%A_%a.out
+#SBATCH --error=test_job_%A_%a.err
+
+#SBATCH --nodes=1
+#SBATCH --ntasks=1
+#SBATCH --time=2:00:00
+#SBATCH --mem=3420mb
+
+#SBATCH -p icelake
+#SBATCH --mail-type=ALL
+
+module purge
+source ../fenics_branch/share/spack/setup-env.sh
+spack env activate fenicsx-env
+spack env status
+spack load [email protected]
+cd mesh_data
+mpiexec -n 1 python3 mesh.py
+cd ..
+mpiexec -n 1 python3 dlrbnicsx_thermomechanical.py