search

bytedance / byteps

A high performance and generic framework for distributed DNN training
Other
3.63k stars 488 forks source link

torch: fix hang after int tensor push_pull #358

Closed pleasantrabbit closed 3 years ago

pleasantrabbit commented 3 years ago

mark task done after averaging an int tensor. This fixes a bug introduced in 46944e8.

Signed-off-by: Yulu Jia yulu.jia@bytedance.com