From 6304690c9c6bd0916c475c02e7e3298b163b1c32 Mon Sep 17 00:00:00 2001 From: Marius Kurz Date: Wed, 24 Jul 2024 08:52:18 +0200 Subject: [PATCH] Add network_interface for Orchestrator as user parameter --- examples/HIT_24_DOF/prm.yaml | 7 ++----- examples/HIT_32_DOF/prm.yaml | 7 ++----- src/relexi/rl/ppo/train.py | 8 ++++---- 3 files changed, 8 insertions(+), 14 deletions(-) diff --git a/examples/HIT_24_DOF/prm.yaml b/examples/HIT_24_DOF/prm.yaml index 0124dbd..bffbfeb 100644 --- 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 index 70c3368..ee16652 100644 --- 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 index eb08961..f2bacc7 100644 --- 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()