DeepSpeed
2e3769a1 - Enable fused_lamb_cuda_kernel on ROCm (#2148)
Commit
View On GitHub
Commit
2 years ago
Enable fused_lamb_cuda_kernel on ROCm (#2148) Co-authored-by: Jeff Rasley <jerasley@microsoft.com> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com>
Author
rraminen rraminen
Parents
e419f7cb
  • csrc/lamb
    • File
      fused_lamb_cuda_kernel.cu
FAQ Terms Privacy Refunds Impressum