thermomech gpu in work

Wells-Group · Dec 21, 2023 · 2f7f063 · 2f7f063
1 parent a315ceb
commit 2f7f063
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo/thermomechanical_dlrbnicsx_multigpu/dlrbnicsx_thermal_multigpu.py b/demo/thermomechanical_dlrbnicsx_multigpu/dlrbnicsx_thermal_multigpu.py
@@ -714,7 +714,7 @@ def generate_ann_output_set(problem, reduced_problem, input_set,
                 thermal_solution_file.write_function(thermal_projection_error_function_plot)
 
 
-    if thermal_cpu_group0_comm != MPI.COMM_NULL:
+    if thermal_gpu_group0_comm != MPI.COMM_NULL:
         print(f"Training time (Thermal): {thermal_elapsed_time}")
 
     if world_comm.rank == 0: