Fix heap buffer overflow when running simulation

aliyun · Nov 8, 2024 · f9484d9 · f9484d9
1 parent 16b21a5
commit f9484d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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;
     }
   }
-}
+}