DeepSpeed
Enable fused_lamb_cuda_kernel on ROCm
#2148
Merged

Commits
  • Enable coalesced_group and coalesced_threads on ROCm
    rraminen committed 3 years ago
  • Merge branch 'master' into fused_lamb_kernel_rocm
    jeffra committed 3 years ago
  • Merge branch 'master' into fused_lamb_kernel_rocm
    tjruwase committed 3 years ago
  • Merge branch 'master' into fused_lamb_kernel_rocm
    tjruwase committed 3 years ago
  • Enable coalesced_group and coalesced_threads on ROCm 5.2 and above
    rraminen committed 3 years ago
  • Merge branch 'master' into fused_lamb_kernel_rocm
    tjruwase committed 3 years ago
  • Merge branch 'master' into fused_lamb_kernel_rocm
    mrwyattii committed 3 years ago
  • Merge branch 'master' into fused_lamb_kernel_rocm
    jeffra committed 3 years ago
Loading