You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Checking we use fused kernels to compute scaled masked softmax on prefix lm#213
Open
thomasw21 wants to merge 6 commits intomainbigscience-workshop/Megatron-DeepSpeed:mainfrom thomas/improve_test_to_test_custom_kernelbigscience-workshop/Megatron-DeepSpeed:thomas/improve_test_to_test_custom_kernelCopy head branch name to clipboard