Skip to content

fix(datasets): build the DP all_reduce tensor on the mesh device instead of assuming CUDA - #2127

Open
li-lizhe wants to merge 2 commits into
InternLM:mainfrom
li-lizhe:fix/dp-allreduce-device
Open

fix(datasets): build the DP all_reduce tensor on the mesh device instead of assuming CUDA#2127
li-lizhe wants to merge 2 commits into
InternLM:mainfrom
li-lizhe:fix/dp-allreduce-device

Merge branch 'main' into fix/dp-allreduce-device

6728cb1
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs