Corrected some syntax for selecting GPU/device with MPI

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

查看文件

@@ -151,8 +151,8 @@ class MPIContext(Context):
# Check GPU resources versus number of MPI tasks
if executor.is_master():
if config.sim_config.general['cuda']:
if executor.size - 1 > len(config.sim_config.cuda['gpus']):
if config.sim_config.general['solver'] == 'cuda':
if executor.size - 1 > len(config.sim_config.devices['devs']):
logger.exception('Not enough GPU resources for number of '
'MPI tasks requested. Number of MPI tasks '
'should be equal to number of GPUs + 1.')