* [Feature] Support MPS device. * Add `auto_select_device` * Add unit tests |
||
---|---|---|
.. | ||
__init__.py | ||
collect_env.py | ||
device.py | ||
distribution.py | ||
logger.py | ||
setup_env.py |
* [Feature] Support MPS device. * Add `auto_select_device` * Add unit tests |
||
---|---|---|
.. | ||
__init__.py | ||
collect_env.py | ||
device.py | ||
distribution.py | ||
logger.py | ||
setup_env.py |