o ½Ù¾iìã@s<ddlZddlmZddlmZejdedefdd„ƒZdS)éN)ÚInt32ÚxÚreturncCs@tdƒ}d}t d¡D]}dd|>|@r|st|ƒ}d}q |S)Né FééT)rÚcutlassÚrange)rÚresÚdoneÚi©r úT/home/ubuntu/.local/lib/python3.10/site-packages/flash_attn_origin/cute/fast_math.pyÚclzs€r)rÚ cutlass.cuteÚcuterÚjitrr r r rÚs