opentau.policies.flash\_attn\_cuda
==================================

.. automodule:: opentau.policies.flash_attn_cuda

   
   .. rubric:: Functions

   .. autosummary::
   
      flash_attn_blockmask
      make_att_block_ids