Yinlei Sun
0beaf69047
[Enhance] Enable full precision training on Ascend NPU. ( #3085 )
...
## Motivation
We will support full precision training on the next generation Ascend
NPU, so there is no need to enable mixed precision by default.
## Modification
Determine whether the current chip supports full precision training, and
automatically enable mixed precision.
## BC-breaking (Optional)
Not affected.
## Use cases (Optional)
We have verified the correctness on the Ascend NPU.
2023-06-07 15:30:24 +08:00
luomaoling
3a14c35974
[Docs] Add NPU support page and add set device. ( #2920 )
2023-04-19 17:15:30 +08:00
luomaoling
ae78cb9d53
[Feature] Support mmseg with NPU backend. ( #2768 )
...
## Motivation
Added ascending device support in mmseg.
## Modification
The main modification points are as follows:
We added an NPU device in the DDP scenario and DP scenario when using
the NPU.
## BC-breaking (Optional)
None
## Use cases (Optional)
We tested
[fcn_unet_s5-d16_4x4_512x1024_160k_cityscapes.py](https://github.com/open-mmlab/mmsegmentation/blob/master/configs/unet/fcn_unet_s5-d16_4x4_512x1024_160k_cityscapes.py )
.
2023-03-23 19:42:49 +08:00
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
