"tests/git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "c1de4c9bbf427b3786546eb16caf37f29cc39651"
[Fix] Fix the output of scatter on MLU device (#1881)
To have the same shape of MLU copy and Scatter.forward, a 1-dim list has to be added. Origin MMDP / MMDDP with this scatter_gather returns a list of tensor for each targets, and here MLU copy returns the same shape Tensor in a list of just one default target.
Showing
Please register or sign in to comment