ROCm / DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
https://www.deepspeed.ai/
Apache License 2.0
4 stars 3 forks source link

[DON'T MERGE] Revert hip cooperative groups workaround [REDUX] #22

Open jithunnair-amd opened 3 years ago

jithunnair-amd commented 3 years ago

This is a re-Merge of pull request #21 from rraminen/PR_Revert_HIP_Cooperative_Groups_Workaround, it reverts commit 5b0fac74999ca0fc5fd4e10ecd6bd614eea0343c.

Depends on new Cooperative Groups HIP headers available in ROCm4.3+

H/T: @rraminen for the work in this PR.

jithunnair-amd commented 2 years ago

retest this please