From b9d232c7206498bed60bcc572ecb300bb27f8a94 Mon Sep 17 00:00:00 2001
From: Craig Warren <craig.warren@northumbria.ac.uk>
Date: Fri, 28 Oct 2022 13:03:23 +0100
Subject: [PATCH] Corrected some syntax for selecting GPU/device with MPI.

---
 gprMax/contexts.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/gprMax/contexts.py b/gprMax/contexts.py
index 84371df6..a014de6e 100644
--- a/gprMax/contexts.py
+++ b/gprMax/contexts.py
@@ -121,9 +121,9 @@ class MPIContext(Context):
         config.model_num = work['i']
         model_config = config.ModelConfig()
         # Set GPU deviceID according to worker rank
-        if config.sim_config.general['cuda']:
-            model_config.cuda = {'gpu': config.sim_config.cuda['gpus'][self.rank - 1],
-                                 'snapsgpu2cpu': False}
+        if config.sim_config.general['solver'] == 'cuda':
+            model_config.device = {'dev': config.sim_config.devices['devs'][self.rank - 1],
+                                   'snapsgpu2cpu': False}
         config.model_configs = model_config
 
         G = create_G()