From 82e80ab12ce9cfb81323249a39a4956b9f8b5890 Mon Sep 17 00:00:00 2001
From: craig-warren <craig_warren@me.com>
Date: Sun, 10 Dec 2023 00:10:41 +0000
Subject: [PATCH] Print GPU device ID with MPI

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

diff --git a/gprMax/contexts.py b/gprMax/contexts.py
index 708b4fe9..66828dc8 100644
--- a/gprMax/contexts.py
+++ b/gprMax/contexts.py
@@ -130,7 +130,9 @@ class MPIContext(Context):
         model_config = config.ModelConfig()
         # Set GPU deviceID according to worker rank
         if config.sim_config.general["solver"] == "cuda":
-            model_config.device = {"dev": config.sim_config.devices["devs"][self.rank - 1], "snapsgpu2cpu": False}
+            model_config.device = {"dev": config.sim_config.devices["devs"][self.rank - 1], 
+                                   "deviceID": self.rank - 1,
+                                   "snapsgpu2cpu": False}
         config.model_configs = model_config
 
         G = create_G()