Commits · 6f674c7ea6a307a62c0ea74cc60e55570de31f6b · OpenDAS / MMCV

08 Jun, 2022 2 commits

[Feature] Support PSAMask with cambricon MLU backend (#2024) · 5601427b

zhouchenyang authored Jun 08, 2022

* [Feature] Support Psamask with cambricon MLU backend

* [Fix] refine the format according to lint check

* Replace std:max/min with conditional operators

* [Fix] Format head files

5601427b

[Feature] Support RoiAlignRotated with cambricon MLU backend (#2033) · fb5062ca

zhouchenyang authored Jun 08, 2022

* [Feature] Support RoiAlignRotated with cambricon MLU backend

* [Fix] Remove std lib in mlu files

* [Fix] replace std::min/max with conditional operators

fb5062ca

14 Apr, 2022 1 commit

[Feature] Support PyTorch backend on MLU (#1770) · 4826a9b7

Qiza-lyhm authored Apr 14, 2022



* feat(MLU): Support PyTorch backend on MLU

  * MMCV support PyTorch backend on MLU

  * Add MLUDataParallel and MLUDistributedDataParallel

  * Add MLU operator support

* [Fix]: Fix PR comments and add IS_MLU to get device available check

* [Fix]: fix PR comments of dist_utils.py

* [Doc] Rewrite annotations of functions.

* [Docs] Rewrite annotation in distributed.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* [Docs] Fix lint
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

4826a9b7