def PrepareWorkers(vms) -> None:
if FLAGS.mpi_vendor == 'intel':
intelmpi.NfsExportIntelDirectory(vms)
else:
for vm in vms:
vm.Install('openmpi')
nfs_service.NfsExportAndMount(vms, _RUN_DIR)
_TestInstall(vms)
def PrepareBinaries(vms: List[linux_vm.BaseLinuxVirtualMachine]) -> None:
"""Prepare binaries on all vms."""
if hpcc.USE_INTEL_COMPILED_HPL.value:
intelmpi.NfsExportIntelDirectory(vms)
vm_util.RunThreaded(lambda vm: vm.Install('numactl'), vms)
return
headnode_vm = vms[0]
if FLAGS.hpcc_binary:
headnode_vm.PushFile(data.ResourcePath(FLAGS.hpcc_binary), './hpcc')
else:
headnode_vm.RemoteCommand(f'cp {hpcc.HPCC_DIR}/hpcc hpcc')
vm_util.RunThreaded(lambda vm: _PrepareBinaries(headnode_vm, vm), vms[1:])
def PrepareWorkers(vms) -> None:
intelmpi.NfsExportIntelDirectory(vms)
nfs_service.NfsExportAndMount(vms, _RUN_DIR)
_TestInstall(vms)