Skip to content

[bugfix] pin MeanMetric all_reduce tensor to float32 to avoid cross-rank dtype mismatch#9551

Merged
Jintao-Huang merged 2 commits into
modelscope:mainfrom
HaozheZhang6:fix/meanmetric-allreduce-dtype
Jun 13, 2026
Merged

[bugfix] pin MeanMetric all_reduce tensor to float32 to avoid cross-rank dtype mismatch#9551
Jintao-Huang merged 2 commits into
modelscope:mainfrom
HaozheZhang6:fix/meanmetric-allreduce-dtype

use float32 for all_reduce tensor (some backends e.g. NPU/HCCL don't …

9823a8f
Select commit
Loading
Failed to load commit list.
Sign in for the full log view

Annotations

1 warning
unittest
succeeded Jun 13, 2026 in 52m 18s