torchacc.ops.context\_parallel package
======================================

Submodules
----------

torchacc.ops.context\_parallel.context\_parallel\_2d module
-----------------------------------------------------------

.. automodule:: torchacc.ops.context_parallel.context_parallel_2d
   :members:
   :undoc-members:
   :show-inheritance:

torchacc.ops.context\_parallel.init\_group module
-------------------------------------------------

.. automodule:: torchacc.ops.context_parallel.init_group
   :members:
   :undoc-members:
   :show-inheritance:

torchacc.ops.context\_parallel.ring\_attn module
------------------------------------------------

.. automodule:: torchacc.ops.context_parallel.ring_attn
   :members:
   :undoc-members:
   :show-inheritance:

torchacc.ops.context\_parallel.ulysses module
---------------------------------------------

.. automodule:: torchacc.ops.context_parallel.ulysses
   :members:
   :undoc-members:
   :show-inheritance:

torchacc.ops.context\_parallel.utils module
-------------------------------------------

.. automodule:: torchacc.ops.context_parallel.utils
   :members:
   :undoc-members:
   :show-inheritance:

Module contents
---------------

.. automodule:: torchacc.ops.context_parallel
   :members:
   :undoc-members:
   :show-inheritance: