Revert "allgather -> gather"

This reverts commit 8eb34bb.
mlcommons · Oct 2, 2024 · 9e0882b · 9e0882b
1 parent 6622521
commit 9e0882b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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