Aston.He
|
7628a61f92
|
feat(mlu): Support PyTorch backend on MLU. (#1515)
* feat(mlu): Support PyTorch backend on MLU.
* fix redundant device variable.
* Update mmseg/apis/train.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update comments.
* Update mmseg/apis/train.py
* Update is_mlu_available flag.
* align util_distribution.py to mmdet.
* align util_distribution.py to mmdet.
* add build_dp, build_ddp testcase.
* Update mmseg/utils/util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmseg/utils/util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmseg/utils/util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* add mmcv version check for mlu device.
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
|
2022-05-25 18:11:42 +08:00 |