Megatron-DeepSpeed icon indicating copy to clipboard operation
Megatron-DeepSpeed copied to clipboard
verified publisher icon bigscience-workshop

Checking we use fused kernels to compute scaled masked softmax on prefix lm

Open thomasw21 opened this issue 4 years ago • 0 comments

  • Related to: #209

Basically re-opening the PR as it seems to pass locally but not CI.

thomasw21 avatar Nov 29 '21 13:11 thomasw21