From f9484d99dd3f1d7ba0ee900a413e5f5996b4bed5 Mon Sep 17 00:00:00 2001
From: Heran Yang <heran55@126.com>
Date: Fri, 8 Nov 2024 09:52:11 +0800
Subject: [PATCH] Fix heap buffer overflow when running simulation

---
 astra-sim-alibabacloud/astra-sim/system/MockNcclGroup.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/astra-sim-alibabacloud/astra-sim/system/MockNcclGroup.cc b/astra-sim-alibabacloud/astra-sim/system/MockNcclGroup.cc
index 3bcec1e0..046e6027 100644
--- a/astra-sim-alibabacloud/astra-sim/system/MockNcclGroup.cc
+++ b/astra-sim-alibabacloud/astra-sim/system/MockNcclGroup.cc
@@ -2035,7 +2035,7 @@ namespace MockNccl {
         q.push(current->right);
       }
     }
-    return node2treenode[(nodes[rank2index[root->node]+1]) % nodes.size()];
+    return node2treenode[nodes[(rank2index[root->node]+1) % nodes.size()]];
   }
 
   ncclInfo* MockNcclGroup::get_algo_proto_info(
@@ -2102,4 +2102,4 @@ namespace MockNccl {
     return info;
     }
   }
-}
\ No newline at end of file
+}