mindspore.runtime
运行时封装了流、事件、内存、执行的接口。MindSpore从不同的后端抽象出相应的模块,允许用户在Python层调度硬件资源。
流
返回此设备上的通信流。 |
|
返回此设备上正在使用的流。 |
|
返回此设备上的默认流。 |
|
设置当前流,这是用于设置流的包装器API。 |
|
同步当前设备上的所有流。 |
|
基于设备流的封装器。 |
|
上下文管理器,用于选择给定的流。 |
事件
设备事件的封装器。 |
内存
返回从进程启动开始,内存池真实被Tensor占用的内存大小的峰值。 |
|
返回从进程启动开始,内存池管理的内存总量的峰值。 |
|
返回当前真实被Tensor占用的内存大小。 |
|
返回内存池当前管理的内存总量。 |
|
返回从内存池查询到的状态信息。 |
|
返回可读的内存池状态信息。 |
|
重置被内存池所管理的内存的峰值。 |
|
重置内存池真实被Tensor占用的内存大小的峰值。 |
|
重置内存池中的全部峰值。 |
|
清理内存池中的内存碎片,优化内存排布。 |
|
设置使用内存池实现的运行时设备内存管理的内存参数。 |
执行
API Name |
Description |
Supported Platforms |
Enable thread-level core binding to assign specific CPU cores to MindSpore's main modules (main thread, pynative, runtime, minddata), to prevent unstable performance caused by MindSpore's threads seizing CPU. |
|
|
Whether to enable synchronous execution. |
|
|
Set the threads number of runtime used. |
|