Zaida Zhou 17dbac1812

[Enhancement] Handle the device type of inputs in functions (#137) ...

* [Enhancement] Handle the device type of inputs in functions

* rename and move three fucntions to dist/utils.py

* minor refinement

* rename dist to torch_dist in utils.py

* update unit tests

* refine unit tests

* add unit tests

* fix unit tests

* replace Sequence with list and tuple

* rename get_backend_device to get_comm_device

* fix unit tests

* fix unit tests

* refactor and add more unit tests

* cast_data_device does not support set type

2022-04-27 19:46:13 +08:00

..

test_dist.py

[Enhancement] Handle the device type of inputs in functions (#137)

2022-04-27 19:46:13 +08:00

test_utils.py

[Enhancement] Handle the device type of inputs in functions (#137)

2022-04-27 19:46:13 +08:00