DeepSpeed
Enable fused_lamb_cuda_kernel on ROCm
#2148
Merged
Overview
Commits
8
Changes
View On GitHub

Enable fused_lamb_cuda_kernel on ROCm #2148

jeffra merged 8 commits into deepspeedai:master from rraminen:fused_lamb_kernel_rocm
rraminen
rraminen Enable coalesced_group and coalesced_threads on ROCm
4c85557d
rraminen rraminen requested a review from jeffra jeffra 3 years ago
rraminen rraminen requested a review from samyam samyam 3 years ago
rraminen rraminen requested a review from tjruwase tjruwase 3 years ago
rraminen rraminen requested a review from ShadenSmith ShadenSmith 3 years ago
rraminen rraminen requested a review from conglongli conglongli 3 years ago
rraminen rraminen requested a review from awan-10 awan-10 3 years ago
rraminen rraminen requested a review from cli99 cli99 3 years ago
rraminen rraminen requested a review from eltonzheng eltonzheng 3 years ago
rraminen rraminen requested a review from minjiaz minjiaz 3 years ago
rraminen rraminen requested a review from RezaYazdaniAminabadi RezaYazdaniAminabadi 3 years ago
rraminen rraminen requested a review from duli2012 duli2012 3 years ago
rraminen rraminen requested a review from mrwyattii mrwyattii 3 years ago
rraminen rraminen requested a review from yaozhewei yaozhewei 3 years ago
rraminen rraminen requested a review from arashb arashb 3 years ago
rraminen rraminen requested a review from xiaoxiawu-microsoft xiaoxiawu-microsoft 3 years ago
rraminen rraminen requested a review from samadejacobs samadejacobs 3 years ago
rraminen
jeffra
jeffra approved these changes on 2022-07-28
jeffra Merge branch 'master' into fused_lamb_kernel_rocm
a52c40c5
tjruwase Merge branch 'master' into fused_lamb_kernel_rocm
05cb9101
tjruwase Merge branch 'master' into fused_lamb_kernel_rocm
a69ebb20
rraminen Enable coalesced_group and coalesced_threads on ROCm 5.2 and above
0460465d
tjruwase Merge branch 'master' into fused_lamb_kernel_rocm
ba0b41b3
mrwyattii Merge branch 'master' into fused_lamb_kernel_rocm
bbe6ad8a
jeffra Merge branch 'master' into fused_lamb_kernel_rocm
5b278934
jeffra jeffra merged 2e3769a1 into master 3 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum