From 5d5c2ad4cca20627d4fbc85226c257aca4be7466 Mon Sep 17 00:00:00 2001
From: Rakesh Sarma <126173968+r-sarma@users.noreply.github.com>
Date: Thu, 12 Dec 2024 17:12:49 +0100
Subject: [PATCH] Update _core.py

---
 use-cases/lattice-qcd/src/device/_core.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/use-cases/lattice-qcd/src/device/_core.py b/use-cases/lattice-qcd/src/device/_core.py
index f4073bc2..973d9b0a 100644
--- a/use-cases/lattice-qcd/src/device/_core.py
+++ b/use-cases/lattice-qcd/src/device/_core.py
@@ -36,6 +36,7 @@ def __init__(self, model):
     def set_seed(self, seed):
         self.seed = seed
         torch.manual_seed(self.seed)
+        print(f"Seed set for rank {self.rank}: {self.seed}")
 
     def to(self, *args, **kwargs):
         self._model.net_.to(*args, **kwargs)