ColossalAI icon indicating copy to clipboard operation
ColossalAI copied to clipboard

Published 20 hours ago • verified publisher icon hpcaitech

[WIP] updated flash attention api

Open kurisusnowdeng opened this issue 2 years ago • 0 comments

  • Added MaskedFlashAttention for easier use

kurisusnowdeng avatar Nov 14 '22 09:11 kurisusnowdeng