Corrected some syntax for selecting GPU/device with MPI.

这个提交包含在:
Craig Warren
2022-10-28 13:03:23 +01:00
父节点 eb25fac027
当前提交 b9d232c720

查看文件

@@ -121,9 +121,9 @@ class MPIContext(Context):
config.model_num = work['i'] config.model_num = work['i']
model_config = config.ModelConfig() model_config = config.ModelConfig()
# Set GPU deviceID according to worker rank # Set GPU deviceID according to worker rank
if config.sim_config.general['cuda']: if config.sim_config.general['solver'] == 'cuda':
model_config.cuda = {'gpu': config.sim_config.cuda['gpus'][self.rank - 1], model_config.device = {'dev': config.sim_config.devices['devs'][self.rank - 1],
'snapsgpu2cpu': False} 'snapsgpu2cpu': False}
config.model_configs = model_config config.model_configs = model_config
G = create_G() G = create_G()