fix mha_bwd: set dtype of softmax_lse to fp32 - Githubissues

DeepLink-org / DeepLinkExt

BSD 3-Clause "New" or "Revised" License

11 stars 0 forks source link

fix mha_bwd: set dtype of softmax_lse to fp32 #22

Closed lljbash closed 10 months ago