WIP

mlcommons · May 2, 2024 · 2ee9e68 · 2ee9e68
1 parent 8916878
commit 2ee9e68
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/et_converter/pytorch2chakra_converter.py b/et_converter/pytorch2chakra_converter.py
@@ -319,6 +319,9 @@ def get_collective_comm_type(self, name: str) -> int:
             "allgather": ALL_GATHER,
             "reducescatter": REDUCE_SCATTER,
             "broadcast": BROADCAST,
+            "sendrecv": BROADCAST, # TODO
+            "send": BROADCAST, # TODO
+            "recv": BROADCAST, # TODO
             # Additional cases can be added here
         }