Commit Graph

5 Commits (4ec25097152bd7c21b1fd087695df6fea57ddc23)

Author SHA1 Message Date
LRJKD 4ec2509715
[Improve] Enable binary operators on the NPU. (#1714)
* [Improve]Enable binary operators on the NPU.

* Update docs link

---------

Co-authored-by: mzr1996 <mzr1996@163.com>
2023-07-19 18:03:49 +08:00
LRJKD 748ab7aa7d
[Fix]Fix the issue where 1p cannot specify a card number. (#1650) 2023-06-21 17:47:08 +08:00
TangYueran dc8691e889
[Feature] Support MLU backend. (#1159)
* Training on MLU is available
2022-11-15 17:02:16 +08:00
wangjiangben-hw 17ed870fd1
[Feature] Support mmcls with NPU backend. (#1072)
* init npu

* Avoid to import latest MMCV code to be compatible with old verisons.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-10-24 11:45:14 +08:00
Ma Zerun c03efeeea4
[Feature] Support MPS device. (#894)
* [Feature] Support MPS device.

* Add `auto_select_device`

* Add unit tests
2022-07-28 12:28:51 +08:00