parallel test instead of serial

Author: pcarruscag

TestCases/parallel_regression_AD.py

@@ -408,6 +408,19 @@ def main():
     test_list.append(pywrapper_FEA_AD_FlowLoad)
     pass_list.append(pywrapper_FEA_AD_FlowLoad.run_test())
 
+    # FEA unsteady AD Load Sensitivity
+    pywrapper_Unst_FEA_AD = TestCase('pywrapper_Unst_FEA_AD')
+    pywrapper_Unst_FEA_AD.cfg_dir = "py_wrapper/custom_load_fea"
+    pywrapper_Unst_FEA_AD.cfg_file = "config.cfg"
+    pywrapper_Unst_FEA_AD.test_iter = 100
+    pywrapper_Unst_FEA_AD.test_vals = [0.2569257, 0.2569264, 0.03166391, 0.03169108]
+    pywrapper_Unst_FEA_AD.command = TestCase.Command("mpirun -n 2", "python", "run_ad.py")
+    pywrapper_Unst_FEA_AD.timeout = 1600
+    pywrapper_Unst_FEA_AD.tol = 0.00001
+    pywrapper_Unst_FEA_AD.new_output = False
+    test_list.append(pywrapper_Unst_FEA_AD)
+    pass_list.append(pywrapper_Unst_FEA_AD.run_test())
+
     # Flow AD Mesh Displacement Sensitivity
     pywrapper_CFD_AD_MeshDisp               = TestCase('pywrapper_CFD_AD_MeshDisp')
     pywrapper_CFD_AD_MeshDisp.cfg_dir       = "py_wrapper/disc_adj_flow/mesh_disp_sens"

TestCases/py_wrapper/custom_load_fea/run_ad.py

@@ -190,10 +190,13 @@ def RunAdjoint(density, peak_load):
   # Finalize the solver and exit cleanly.
   driver.Finalize()
 
-  return load_sens
+  return comm.allreduce(load_sens)
 
 
 def main():
+  comm = MPI.COMM_WORLD
+  rank = comm.Get_rank()
+
   # Run the primal with 2 loads to compute the sensitivity via finite differences.
   obj_pert_load = RunPrimal(1, 2.002)
   obj_pert_rho = RunPrimal(1.0001, 2)
@@ -206,15 +209,18 @@ def main():
   with open('Results_Reverse_Adjoint.txt') as f:
     sens_rho = float(f.readlines()[-1].strip().split('\t')[-1])
 
-  print("      Finite Differences\tDiscrete Adjoint")
-  print(f"Load  {sens_load_fd}\t{sens_load}")
-  print(f"Rho   {sens_rho_fd}\t{sens_rho}")
+  if rank == 0:
+    print("      Finite Differences\tDiscrete Adjoint")
+    print(f"Load  {sens_load_fd}\t{sens_load}")
+    print(f"Rho   {sens_rho_fd}\t{sens_rho}")
 
   assert abs(sens_load / sens_load_fd - 1) < 1e-4, "Error in load derivative."
   assert abs(sens_rho / sens_rho_fd - 1) < 1e-3, "Error in density derivative."
 
   # Print results for the regression script to check.
-  print(100, 100, sens_load_fd, sens_load, sens_rho_fd, sens_rho)
+  if rank == 0:
+    print("\n")
+    print(100, 100, sens_load_fd, sens_load, sens_rho_fd, sens_rho)
 
 
 if __name__ == '__main__':

TestCases/serial_regression_AD.py

@@ -343,19 +343,6 @@ def main():
     test_list.append(pywrapper_FEA_AD_FlowLoad)
     pass_list.append(pywrapper_FEA_AD_FlowLoad.run_test())
 
-    # FEA unsteady AD Load Sensitivity
-    pywrapper_Unst_FEA_AD = TestCase('pywrapper_Unst_FEA_AD')
-    pywrapper_Unst_FEA_AD.cfg_dir = "py_wrapper/custom_load_fea"
-    pywrapper_Unst_FEA_AD.cfg_file = "config.cfg"
-    pywrapper_Unst_FEA_AD.test_iter = 100
-    pywrapper_Unst_FEA_AD.test_vals = [0.2569256889872473, 0.256926360722213, 0.03166391031539373, 0.0316910814265723]
-    pywrapper_Unst_FEA_AD.command = TestCase.Command(exec = "python", param = "run_ad.py")
-    pywrapper_Unst_FEA_AD.timeout = 1600
-    pywrapper_Unst_FEA_AD.tol = 0.00001
-    pywrapper_Unst_FEA_AD.new_output = False
-    test_list.append(pywrapper_Unst_FEA_AD)
-    pass_list.append(pywrapper_Unst_FEA_AD.run_test())
-
     # Flow AD Mesh Displacement Sensitivity
     pywrapper_CFD_AD_MeshDisp               = TestCase('pywrapper_CFD_AD_MeshDisp')
     pywrapper_CFD_AD_MeshDisp.cfg_dir       = "py_wrapper/disc_adj_flow/mesh_disp_sens"