thermomech gpu in work

Wells-Group · Dec 21, 2023 · a315ceb · a315ceb
1 parent c1eb9fe
commit a315ceb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/demo/thermomechanical_dlrbnicsx_multigpu/dlrbnicsx_thermal_multigpu.py b/demo/thermomechanical_dlrbnicsx_multigpu/dlrbnicsx_thermal_multigpu.py
@@ -490,7 +490,7 @@ def generate_ann_output_set(problem, reduced_problem, input_set,
 
     print("\n")
 
-    thermal_gpu_group0_procs = world_comm.group.Incl([0, 1, 2, 3])
+    thermal_gpu_group0_procs = world_comm.group.Incl([0, 1, 2])
     thermal_gpu_group0_comm = world_comm.Create_group(thermal_gpu_group0_procs)
 
     # ANN model
@@ -499,7 +499,7 @@ def generate_ann_output_set(problem, reduced_problem, input_set,
 
     if thermal_gpu_group0_comm != MPI.COMM_NULL:
 
-        cuda_rank_list = [0, 1, 2, 3]
+        cuda_rank_list = [0, 1, 2]
         init_gpu_process_group(thermal_gpu_group0_comm)
 
         thermal_training_set_indices_gpu = np.arange(thermal_gpu_group0_comm.rank,

diff --git a/demo/thermomechanical_dlrbnicsx_multigpu/test_sbatch_gpu b/demo/thermomechanical_dlrbnicsx_multigpu/test_sbatch_gpu
@@ -5,10 +5,10 @@
 #SBATCH --error=test_job_gpu_%A_%a.err
 
 #SBATCH --nodes=1
-#SBATCH --gres=gpu:4
-#SBATCH --ntasks=8
+#SBATCH --gres=gpu:3
+#SBATCH --ntasks=16
 #SBATCH --time=02:00:00
-#SBATCH --mem=7000mb
+#SBATCH --mem=3500mb
 
 
 #SBATCH -p ampere