torch_npu.npu.SyncLaunchStream
产品支持情况
| 产品 | 是否支持 |
|---|---|
| [object Object]Atlas A3 训练系列产品[object Object] | √ |
| [object Object]Atlas A2 训练系列产品[object Object] | √ |
| [object Object]Atlas 训练系列产品[object Object] | √ |
功能说明
创建一条同步下发NPUStream，在该流上下发的任务不再使用taskqueue异步下发。在集群场景某一设备出现故障，其他设备保存checkpoint时，可使用此同步下发流保存。
函数原型
[object Object]
参数说明
device(Any)：可以为设备数字id或者字符串“npu:0”，默认值为None（即当前线程对应的设备id）。
返回值说明
一条创建好的NPUStream，在该流上下发任务不再使用taskqueue异步下发。
约束说明
- 由于不再下发到taskqueue，因此该流的下发性能相比普通流有所降低，建议在集群训练时某些节点出现故障，其他节点保存ckpt时使用。
- 同步下发流资源池只有4条，创建超过4条时将会循环从资源池中获取。
调用示例
[object Object]