mstar.utils.flashinfer\_utils
=============================

.. automodule:: mstar.utils.flashinfer_utils

   
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
   
      run_attention
      run_rms_norm
   
   

   
   
   .. rubric:: Classes

   .. autosummary::
   
      FlashInferDecodeWrapper
      FlashInferPrefillWrapper