mmengine/docs/zh_cn/api/device.rst

.. role:: hidden
    :class: hidden-section

mmengine.device
===================================

.. currentmodule:: mmengine.device

.. autosummary::
   :toctree: generated
   :nosignatures:

   get_device
   get_max_cuda_memory
   is_cuda_available
   is_npu_available
   is_mlu_available
   is_mps_available
[Docs] Make API of docs clearer (#469) * [Docs] Make API of docs clearer * minor fix * fix md format 2022-08-30 16:27:20 +08:00			`.. role:: hidden`
			`:class: hidden-section`

			`mmengine.device`
			`===================================`

			`.. currentmodule:: mmengine.device`

			`.. autosummary::`
			`:toctree: generated`
			`:nosignatures:`

			`get_device`
			`get_max_cuda_memory`
			`is_cuda_available`
[Feature] Support engine with NPU backend. (#572) * init npu * Update mmengine/optim/optimizer/amp_optimizer_wrapper.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmengine/dist/dist.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * change to is_hccl_backend * Update mmengine/optim/optimizer/amp_optimizer_wrapper.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * add comment with AmpOptimWrapper * Update mmengine/runner/amp.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmengine/runner/amp.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * add npu fn in base_model * Update mmengine/optim/optimizer/amp_optimizer_wrapper.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * clean lint * Update mmengine/optim/optimizer/amp_optimizer_wrapper.py Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmengine/model/base_model/base_model.py Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com> * add is_npu_available * try to fix * Add comments * Refine grammar Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com> Co-authored-by: HAOCHENYE <21724054@zju.edu.cn> 2022-10-24 11:42:29 +08:00			`is_npu_available`
[Docs] Make API of docs clearer (#469) * [Docs] Make API of docs clearer * minor fix * fix md format 2022-08-30 16:27:20 +08:00			`is_mlu_available`
			`is_mps_available`