Switch to use cutlass.utils.get_smem_capacity_in_bytes#1906

tridao merged 1 commit into

Dao-AILab:mainDao-AILab/flash-attention:mainfrom

brandon-yujie-sun:update_smem_cap_apibrandon-yujie-sun/flash-attention:update_smem_cap_apiCopy head branch name to clipboard