Add network_interface for Orchestrator as user parameter

flexi-framework · Jul 24, 2024 · 6304690 · 6304690
1 parent ab0c2f3
commit 6304690
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 14 deletions.
diff --git a/examples/HIT_24_DOF/prm.yaml b/examples/HIT_24_DOF/prm.yaml
@@ -14,8 +14,6 @@ environment:
   executable_path: ../../../flexi-extensions/build/bin/flexi
   parameter_file: ./simulation_files/parameter_flexi.ini
   mesh_file: ./simulation_files/CART_HEX_PERIODIC_004_mesh.h5
-  #local_dir: /var/tmp
-  mpi_launch_mpmd: False
   env_launcher: local
 
 # All parameters for setting up the reward
@@ -64,6 +62,5 @@ performance:
 # SmartSim
 smartsim:
   smartsim_port: 6780
-  smartsim_num_dbs: 1
-  smartsim_launcher: pbs
-  smartsim_orchestrator: pbs
+  smartsim_network_interface: local
+  smartsim_orchestrator: auto
diff --git a/examples/HIT_32_DOF/prm.yaml b/examples/HIT_32_DOF/prm.yaml
@@ -14,8 +14,6 @@ environment:
   executable_path: ../../../flexi-extensions/build/bin/flexi
   parameter_file: ./simulation_files/parameter_flexi.ini
   mesh_file: ./simulation_files/CART_HEX_PERIODIC_004_mesh.h5
-  #local_dir: /var/tmp
-  mpi_launch_mpmd: False
   env_launcher: local
 
 # All parameters for setting up the reward
@@ -64,6 +62,5 @@ performance:
 # SmartSim
 smartsim:
   smartsim_port: 6780
-  smartsim_num_dbs: 1
-  smartsim_launcher: pbs
-  smartsim_orchestrator: pbs
+  smartsim_network_interface: local
+  smartsim_orchestrator: auto
diff --git a/src/relexi/rl/ppo/train.py b/src/relexi/rl/ppo/train.py
@@ -62,9 +62,9 @@ def train( config_file
           ,do_profile  = False
           ,smartsim_port    = 6780
           ,smartsim_num_dbs = 1
-          ,smartsim_launcher = "local"
-          ,smartsim_orchestrator = "local"
-          ,env_launcher = "mpirun"
+          ,smartsim_orchestrator = 'local'
+          ,smartsim_network_interface = 'local'
+          ,env_launcher = 'mpirun'
           ,mpi_launch_mpmd = False
           ,local_dir = None
           ,n_procs_per_node=128 # Hawk
@@ -110,7 +110,7 @@ def train( config_file
     runtime = relexi.runtime.Runtime(
                                     type_=smartsim_orchestrator,
                                     db_port=smartsim_port,
-                                    db_network_interface='ibp94s0',
+                                    db_network_interface=smartsim_network_interface,
                                     )
     runtime.info()