From 9e0882b5abd047d1e5f1286d93466cd4ba04b917 Mon Sep 17 00:00:00 2001
From: Florian Heilmann <florian.heilmann@turba.ai>
Date: Wed, 2 Oct 2024 16:15:56 +0200
Subject: [PATCH] Revert "allgather -> gather"

This reverts commit 8eb34bbf72bec1d1404e29dcdead0eb0a73d0611.
---
 src/converter/pytorch_converter.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/converter/pytorch_converter.py b/src/converter/pytorch_converter.py
index e7c49d9b..2371dccb 100644
--- a/src/converter/pytorch_converter.py
+++ b/src/converter/pytorch_converter.py
@@ -362,7 +362,7 @@ def get_collective_comm_type(self, name: str) -> int:
         comm_type_mapping = {
             "allreduce": ALL_REDUCE,
             "alltoall": ALL_TO_ALL,
-            "gather": ALL_GATHER,
+            "allgather": ALL_GATHER,
             "reducescatter": REDUCE_SCATTER,
             "broadcast": BROADCAST,
             # Additional cases can be added here