创建一条同步下发NPUStream,在该流上下发的任务不再使用taskqueue异步下发。在集群场景某一设备出现故障,其他设备保存checkpoint时,可使用此同步下发流保存。
[object Object]
device(Any):可以为设备数字id或者字符串“npu:0”,默认值为None(即当前线程对应的设备id)。
一条创建好的NPUStream,在该流上下发任务不再使用taskqueue异步下发。
- 由于不再下发到taskqueue,因此该流的下发性能相比普通流有所降低,建议在集群训练时某些节点出现故障,其他节点保存ckpt时使用。
- 同步下发流资源池只有4条,创建超过4条时将会循环从资源池中获取。
[object Object]