etha.comm.comm_methods
======================

.. py:module:: etha.comm.comm_methods

.. autoapi-nested-parse::

   Communication utilities for Etha.



Attributes
----------

.. autoapisummary::

   etha.comm.comm_methods.logger


Functions
---------

.. autoapisummary::

   etha.comm.comm_methods.bucket_comm
   etha.comm.comm_methods.chunk_comm
   etha.comm.comm_methods.gather_broadcast_comm


Module Contents
---------------

.. py:function:: bucket_comm(buckets: list[etha.comm.ir.Bucket]) -> None

   Execute bucketized communication.


.. py:function:: chunk_comm(chunks: list[etha.comm.ir.Chunk]) -> None

   Execute chunked communication.


.. py:function:: gather_broadcast_comm(target_mesh: torch.distributed._tensor.DeviceMesh, target_specs: tuple[torch.distributed.tensor.placement_types.Placement, Ellipsis], local_tensor: torch.distributed._tensor.DTensor, origin_tensor: torch.Tensor, source_world_size: int)

   Performs data redistribution using the Gather-Broadcast method.


.. py:data:: logger