开发者
资源
[object Object][object Object][object Object]undefined
[object Object]

创建一条同步下发NPUStream,在该流上下发的任务不再使用taskqueue异步下发。在集群场景某一设备出现故障,其他设备保存checkpoint时,可使用此同步下发流保存。

[object Object]
[object Object]
[object Object]

device(Any):可以为设备数字id或者字符串“npu:0”,默认值为None(即当前线程对应的设备id)。

[object Object]

一条创建好的NPUStream,在该流上下发任务不再使用taskqueue异步下发。

[object Object]
  • 由于不再下发到taskqueue,因此该流的下发性能相比普通流有所降低,建议在集群训练时某些节点出现故障,其他节点保存ckpt时使用。
  • 同步下发流资源池只有4条,创建超过4条时将会循环从资源池中获取。
[object Object]
[object Object]