mindspore.runtime
运行时封装了执行、内存、流、事件的接口。MindSpore从不同的后端抽象出相应的模块,允许用户在Python层调度硬件资源。
执行
使能线程级绑核功能,给MindSpore的主要模块(主线程、pynative、runtime、minddata)分配特定的CPU核,防止MindSpore线程抢占CPU导致的性能不稳定情况。 |
|
表示是否启动设备同步执行。 |
|
设置运行时要使用的线程数。 |
内存
返回从进程启动开始,内存池真实被Tensor占用的内存大小的峰值。 |
|
返回从进程启动开始,内存池管理的内存总量的峰值。 |
|
返回当前真实被Tensor占用的内存大小。 |
|
返回内存池当前管理的内存总量。 |
|
返回从内存池查询到的状态信息。 |
|
返回可读的内存池状态信息。 |
|
重置被内存池所管理的内存的峰值。 |
|
重置内存池真实被Tensor占用的内存大小的峰值。 |
|
重置内存池中的全部峰值。 |
|
清理内存池中的内存碎片,优化内存排布。 |
|
设置使用内存池实现的运行时设备内存管理的内存参数。 |
流
返回此设备上的通信流。 |
|
返回此设备上正在使用的流。 |
|
返回此设备上的默认流。 |
|
设置当前流,这是用于设置流的包装器API。 |
|
同步当前设备上的所有流。 |
|
基于设备流的封装器。 |
|
上下文管理器,用于选择给定的流。 |
事件
设备事件的封装器。 |