From 24fef63cb29af8056154e960ec6881f3b9d200fb Mon Sep 17 00:00:00 2001 From: Craig Warren Date: Fri, 28 Oct 2022 13:07:10 +0100 Subject: [PATCH] Corrected some syntax for selecting GPU/device with MPI --- gprMax/contexts.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gprMax/contexts.py b/gprMax/contexts.py index a014de6e..fdb92d91 100644 --- a/gprMax/contexts.py +++ b/gprMax/contexts.py @@ -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.')