hanhaowen-mt
|
3d8a611eec
|
[Feature] Add the support for musa device support (#1453)
|
2024-01-11 16:25:01 +08:00 |
|
wangjiangben-hw
|
601db12d38
|
[Feature] Support engine with NPU backend. (#572)
* init npu
* Update mmengine/optim/optimizer/amp_optimizer_wrapper.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmengine/dist/dist.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* change to is_hccl_backend
* Update mmengine/optim/optimizer/amp_optimizer_wrapper.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* add comment with AmpOptimWrapper
* Update mmengine/runner/amp.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmengine/runner/amp.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* add npu fn in base_model
* Update mmengine/optim/optimizer/amp_optimizer_wrapper.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* clean lint
* Update mmengine/optim/optimizer/amp_optimizer_wrapper.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmengine/model/base_model/base_model.py
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
* add is_npu_available
* try to fix
* Add comments
* Refine grammar
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Co-authored-by: HAOCHENYE <21724054@zju.edu.cn>
|
2022-10-24 11:42:29 +08:00 |
|
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 |
|