[Fix] Add support for Ascend device (#847)
* add npu device support * add comment for torch.npu.set_compile_mode
Showing
- mmengine/device/utils.py 4 additions, 0 deletionsmmengine/device/utils.py
- mmengine/model/base_model/base_model.py 12 additions, 0 deletionsmmengine/model/base_model/base_model.py
- mmengine/model/base_model/data_preprocessor.py 21 additions, 0 deletionsmmengine/model/base_model/data_preprocessor.py
- mmengine/structures/base_data_element.py 11 additions, 0 deletionsmmengine/structures/base_data_element.py
Loading
Please register or sign in to comment