search

BaguaSys / bagua

Bagua Speeds up PyTorch
https://tutorials-8ro.pages.dev/
MIT License
872 stars 83 forks source link

fix broadcast module test case #287

Closed NOBLES5E closed 2 years ago

NOBLES5E commented 2 years ago

https://github.com/BaguaSys/bagua/pull/228/files#diff-a5d7ee32ab8d93a59290923aeb9597cae5becc8fb682daeac373c91bce6061f3R96-R98

Tengxu-Sun commented 2 years ago

This problem is being solved by @wangraying, thx a lot

NOBLES5E commented 2 years ago

the program hangs when using GradientAllReduce algorithm