colossalai.kernel.cuda\_native.multihead\_attention
===================================================

.. automodule:: colossalai.kernel.cuda_native.multihead_attention
   :members: