From 3c02faa78ba6af32f8c94341f2fd2e5cfad183b9 Mon Sep 17 00:00:00 2001
From: StrongAdam <adam@strongcompute.com>
Date: Mon, 12 May 2025 14:34:20 +1000
Subject: [PATCH] fix: global strategy dtype fix

---
 cycling_utils/saving.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cycling_utils/saving.py b/cycling_utils/saving.py
index 495b214..93e0c28 100644
--- a/cycling_utils/saving.py
+++ b/cycling_utils/saving.py
@@ -256,7 +256,7 @@ def prepare_checkpoint_directory(self, force_save=False):
         if self.strategy in ["sync_any", "sync_all"]:
             global_force = torch.tensor(
                 1 if force_save else 0,
-                dtype=torch.int16,
+                dtype=torch.int64,
                 requires_grad=False,
                 device="cuda",
             )