55 lines
2.0 KiB
ReStructuredText
55 lines
2.0 KiB
ReStructuredText
mindspore.device_context
|
|
===========================
|
|
|
|
The device_context encapsulates the interface for querying the number of devices with whether the currently specified backend is available.
|
|
|
|
Cpu Device Backend Management
|
|
-----------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: device_context
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.device_context.cpu.device_count
|
|
mindspore.device_context.cpu.is_available
|
|
mindspore.device_context.cpu.op_tuning.threads_num
|
|
|
|
Gpu Device Backend Management
|
|
------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: device_context
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.device_context.gpu.device_count
|
|
mindspore.device_context.gpu.is_available
|
|
mindspore.device_context.gpu.op_precision.conv_allow_tf32
|
|
mindspore.device_context.gpu.op_precision.matmul_allow_tf32
|
|
mindspore.device_context.gpu.op_tuning.conv_dgrad_algo
|
|
mindspore.device_context.gpu.op_tuning.conv_fprop_algo
|
|
mindspore.device_context.gpu.op_tuning.conv_wgrad_algo
|
|
|
|
Ascend Device Backend Management
|
|
---------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: device_context
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.device_context.ascend.device_count
|
|
mindspore.device_context.ascend.is_available
|
|
mindspore.device_context.ascend.op_precision.conv_allow_hf32
|
|
mindspore.device_context.ascend.op_precision.matmul_allow_hf32
|
|
mindspore.device_context.ascend.op_precision.precision_mode
|
|
mindspore.device_context.ascend.op_precision.op_precision_mode
|
|
mindspore.device_context.ascend.op_debug.execute_timeout
|
|
mindspore.device_context.ascend.op_debug.debug_option
|
|
mindspore.device_context.ascend.op_debug.aclinit_config
|
|
mindspore.device_context.ascend.op_tuning.op_compile
|
|
mindspore.device_context.ascend.op_tuning.aclnn_cache
|
|
mindspore.device_context.ascend.op_tuning.aoe_tune_mode
|
|
mindspore.device_context.ascend.op_tuning.aoe_job_type
|