Skip to content
GitLab
Explore
Sign in
Register
wanggh
apex
Repository
apex
apex
transformer
functional
fused_softmax.py
Find file
Blame
History
Permalink
Remove `custom_fwd`/`custom_bwd` from fused softmax (#1188)
· 14ccf598
Masaki Kozuki
authored
Oct 09, 2021
* run backward * remove custom_fwd/custom_bwd
14ccf598