--gres=gpu:0 added to srun options, necessary to run on cpus of leonardo booster and dcgp partitions

Author: Maochao Xiao

src/smartflow/runtime.py

@@ -254,11 +254,12 @@ def _validate_args(arg, n):
                     }
                 elif launcher == 'srun':
                     run_args = {
-                        # 'mpi': 'pmix',
+                        # 'mpi': 'pmix_v3',
                         'nodelist': ','.join(self.launch_config.hosts_per_exe[i]),
                         'distribution': 'block:block:block,Pack',
-                        'cpu-bind': 'verbose',
+                        'cpu-bind': 'cores,verbose',
                         'exclusive': None,
+                        'gres': 'gpu:0',
                     }
                 run_settings = self.exp.create_run_settings(
                     exe=exe[i],