@@ -69,23 +69,24 @@ def Prepare(benchmark_spec):
6969 background_tasks .RunThreaded (lambda vm : vm .Install ('build_tools' ), vms )
7070 background_tasks .RunThreaded (lambda vm : vm .AuthenticateVm (), vms )
7171 hpc_util .CreateMachineFile (vms )
72- headnode . InstallPackages ('pkg-config' )
72+ background_tasks . RunThreaded ( lambda vm : vm . InstallPackages ('pkg-config' ), vms )
7373 # io500 needs to build a group of binaries.
74- # Among them, the pined commit for pfind is currently broken,
74+ # Among them, the pinned commit for pfind is currently broken,
7575 # where it looks for *.o files. This commit aaba722 fixes the script
7676 # to look for *.a files.
77- headnode .RemoteCommand (
77+ # Install binary on all vms.
78+ background_tasks .RunThreaded (lambda vm : vm .RemoteCommand (
7879 f'cd { BENCHMARK_DIR } && git clone { GIT_REPO } -b io500-isc24 && '
7980 'cd io500 && '
8081 'sed -i "s/778dca8/aaba722/g" prepare.sh && '
81- './prepare.sh' )
82+ './prepare.sh' ), vms )
8283 # TODO(yuyanting) Make this a flag to accept other configs.
8384 local_path = data .ResourcePath ('io500/io500.ini.j2' )
8485 remote_path = f'{ BENCHMARK_DIR } /io500.ini'
85- headnode .RenderTemplate (
86+ background_tasks . RunThreaded ( lambda vm : vm .RenderTemplate (
8687 template_path = local_path , remote_path = remote_path ,
8788 context = {'directory' : headnode .scratch_disks [0 ].mount_point }
88- )
89+ ), vms )
8990
9091
9192def _Run (headnode , ranks , ppn ):
0 commit comments