[Feature] Support MLU backend (#1075)
* support mlu device * support mlu device * fix lint error * fix lint error builder.py * fix lint error in amp.py * fix lint errors * fix data type in instance_data.py
Showing
- mmengine/dist/dist.py 5 additions, 1 deletionmmengine/dist/dist.py
- mmengine/model/base_model/base_model.py 14 additions, 0 deletionsmmengine/model/base_model/base_model.py
- mmengine/model/base_model/data_preprocessor.py 9 additions, 0 deletionsmmengine/model/base_model/data_preprocessor.py
- mmengine/optim/optimizer/amp_optimizer_wrapper.py 7 additions, 3 deletionsmmengine/optim/optimizer/amp_optimizer_wrapper.py
- mmengine/runner/amp.py 10 additions, 4 deletionsmmengine/runner/amp.py
- mmengine/structures/base_data_element.py 10 additions, 0 deletionsmmengine/structures/base_data_element.py
- mmengine/structures/instance_data.py 3 additions, 0 deletionsmmengine/structures/instance_data.py
- tests/test_runner/test_amp.py 17 additions, 2 deletionstests/test_runner/test_amp.py
Loading
Please register or sign in to comment