Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wuxk1
Megatron-LM
Repository
"official/r1/resnet/imagenet_main.py" did not exist on "7b5606a5d6e469a090bce3b443915747a01e3521"
76db958327475bfa4a290ec5d782f8213cdb67e9
Switch branch/tag
megatron-lm
megatron
fused_kernels
scaled_masked_softmax.h
Find file
Blame
History
Permalink
support for all mask in fused kernel + avoiding inplace operation in bwd pass
· 76db9583
Vijay Korthikanti
authored
Jul 28, 2022
76db9583
scaled_masked_softmax.h
31.4 KB
Edit
Web IDE
Replace scaled_masked_softmax.h
×
Attach a file by drag & drop or
click to upload
Commit message
Replace scaled_masked_softmax.h
Replace file
Cancel
A new branch will be created in your fork and a new merge request will be started.
