Commit Graph

2 Commits (a9ad09bded3e573ee2f7811ee7052cfa2cf0b6c7)

Author SHA1 Message Date
Alex Yang 2994195be2
[Feat] Support training on MPS (#331)
* [Feat] Support mps

* fix docstring
2022-06-23 16:53:19 +08:00
Jiazhen Wang d0d7174274
[Feature] Support MLU Devices (#288)
* support mlu

* add ut and refine docstring
2022-06-16 20:28:09 +08:00