84 lines
2.0 KiB
ReStructuredText
84 lines
2.0 KiB
ReStructuredText
mindspore.runtime
|
||
==================
|
||
|
||
运行时封装了执行、内存、流、事件的接口。MindSpore从不同的后端抽象出相应的模块，允许用户在Python层调度硬件资源。
|
||
|
||
执行
|
||
------------
|
||
|
||
.. mscnautosummary::
|
||
:toctree: runtime
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.runtime.dispatch_threads_num
|
||
mindspore.runtime.launch_blocking
|
||
mindspore.runtime.set_cpu_affinity
|
||
mindspore.runtime.set_kernel_launch_capture
|
||
mindspore.runtime.set_kernel_launch_group
|
||
|
||
内存
|
||
------------
|
||
|
||
.. mscnautosummary::
|
||
:toctree: runtime
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.runtime.MemPool
|
||
mindspore.runtime.PluggableAllocator
|
||
mindspore.runtime.empty_cache
|
||
mindspore.runtime.max_memory_allocated
|
||
mindspore.runtime.max_memory_reserved
|
||
mindspore.runtime.memory_allocated
|
||
mindspore.runtime.memory_replay
|
||
mindspore.runtime.memory_reserved
|
||
mindspore.runtime.memory_stats
|
||
mindspore.runtime.memory_summary
|
||
mindspore.runtime.reset_max_memory_reserved
|
||
mindspore.runtime.reset_max_memory_allocated
|
||
mindspore.runtime.reset_peak_memory_stats
|
||
mindspore.runtime.set_memory
|
||
mindspore.runtime.use_mem_pool
|
||
|
||
流
|
||
---------
|
||
|
||
.. mscnautosummary::
|
||
:toctree: runtime
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.runtime.communication_stream
|
||
mindspore.runtime.current_stream
|
||
mindspore.runtime.default_stream
|
||
mindspore.runtime.set_cur_stream
|
||
mindspore.runtime.synchronize
|
||
mindspore.runtime.Stream
|
||
mindspore.runtime.StreamCtx
|
||
|
||
事件
|
||
---------
|
||
|
||
.. mscnautosummary::
|
||
:toctree: runtime
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.runtime.Event
|
||
|
||
资源限制
|
||
---------
|
||
|
||
.. mscnautosummary::
|
||
:toctree: runtime
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.runtime.get_device_limit
|
||
mindspore.runtime.get_stream_limit
|
||
mindspore.runtime.reset_stream_limit
|
||
mindspore.runtime.set_device_limit
|
||
mindspore.runtime.set_stream_limit
|
||
mindspore.runtime.StreamLimitCtx
|