From 711d73262cffe6f3c095b9a96ec1b41fb577a8d0 Mon Sep 17 00:00:00 2001
From: AtlantaPepsi <timhu102@amd.com>
Date: Tue, 26 May 2026 04:12:45 +0000
Subject: [PATCH] removing secondary reordering

---
 src/client/Presets/PodAllToAll.hpp | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/src/client/Presets/PodAllToAll.hpp b/src/client/Presets/PodAllToAll.hpp
index 03da167..be1ee08 100644
--- a/src/client/Presets/PodAllToAll.hpp
+++ b/src/client/Presets/PodAllToAll.hpp
@@ -154,15 +154,16 @@ int PodAllToAllPreset(EnvVars&          ev,
       return ERR_FATAL;
     }
     int groupSize = n / numGroups;
-    std::vector<MemDevice> devices(n);
     std::vector<int> indices(n);
     for (int k = 0; k < n; k++) indices[k] = k;
     Utils::StrideGenerate(indices, groupStride);
-    int idx = 0;
-    for (int rank : ranks) {
-      for (int devIdx = 0; devIdx < numGpus; devIdx++) {
-        devices[indices[idx++]] = {memType, devIdx, rank};
-      }
+
+    std::vector<MemDevice> devices(n);
+    for (int i = 0; i < n; i++) {
+      int const globalIdx = indices[i];
+      int const rankIdx   = globalIdx / numGpus;
+      int const devIdx    = globalIdx % numGpus;
+      devices[i] = {memType, devIdx, ranks[rankIdx]};
     }
 
     // Build transfers for every group, then run once per pod so all groups share the same