68 lines
1.6 KiB
ReStructuredText
68 lines
1.6 KiB
ReStructuredText
mindspore.hal
|
|
=============
|
|
|
|
Hal encapsulates interfaces for device, stream, event, and memory. MindSpore abstracts the corresponding modules from different backends, allowing users to schedule hardware resources at the Python layer.
|
|
|
|
Device
|
|
-----------
|
|
|
|
.. autosummary::
|
|
:toctree: hal
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.hal.device_count
|
|
mindspore.hal.get_arch_list
|
|
mindspore.hal.get_device_capability
|
|
mindspore.hal.get_device_name
|
|
mindspore.hal.get_device_properties
|
|
mindspore.hal.is_available
|
|
mindspore.hal.is_initialized
|
|
|
|
Stream
|
|
---------
|
|
|
|
.. autosummary::
|
|
:toctree: hal
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.hal.communication_stream
|
|
mindspore.hal.current_stream
|
|
mindspore.hal.default_stream
|
|
mindspore.hal.set_cur_stream
|
|
mindspore.hal.synchronize
|
|
mindspore.hal.Stream
|
|
mindspore.hal.StreamCtx
|
|
|
|
Event
|
|
---------
|
|
|
|
.. autosummary::
|
|
:toctree: hal
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.hal.Event
|
|
|
|
Memory
|
|
------------
|
|
|
|
.. autosummary::
|
|
:toctree: hal
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.hal.contiguous_tensors_handle.combine_tensor_list_contiguous
|
|
mindspore.hal.contiguous_tensors_handle.ContiguousTensorsHandle
|
|
mindspore.hal.max_memory_allocated
|
|
mindspore.hal.max_memory_reserved
|
|
mindspore.hal.memory_allocated
|
|
mindspore.hal.memory_reserved
|
|
mindspore.hal.memory_stats
|
|
mindspore.hal.memory_summary
|
|
mindspore.hal.reset_max_memory_reserved
|
|
mindspore.hal.reset_max_memory_allocated
|
|
mindspore.hal.reset_peak_memory_stats
|
|
mindspore.hal.empty_cache
|