Move cuda with_device guard as common utility instead of torch one - Githubissues

Bluefog-Lib / bluefog

Distributed and decentralized training framework for PyTorch over graph

https://bluefog-lib.github.io/bluefog/

Apache License 2.0

291 stars 71 forks source link

Move cuda with_device guard as common utility instead of torch one #8

Closed BichengYing closed 4 years ago

BichengYing commented 4 years ago

Done