From 51ecbd5663e517c11e79d53a72b1ba7546d91757 Mon Sep 17 00:00:00 2001
From: Lester Hedges <lester.hedges@gmail.com>
Date: Mon, 4 Nov 2024 12:08:23 +0000
Subject: [PATCH] Log GPU device index for each replica. [ci skip]

---
 src/somd2/runner/_repex.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/somd2/runner/_repex.py b/src/somd2/runner/_repex.py
index 62fbded..b018200 100644
--- a/src/somd2/runner/_repex.py
+++ b/src/somd2/runner/_repex.py
@@ -103,7 +103,7 @@ def __init__(self, system, lambdas, num_gpus, dynamics_kwargs):
             # Append the dynamics object.
             self._dynamics.append(dynamics)
 
-            _logger.info(f"Created dynamics object for lambda {lam:.5f}")
+            _logger.info(f"Created dynamics object for lambda {lam:.5f} on device {device}")
 
     def get(self, index):
         """