昇腾社区首页
中文
注册

atb/operation.h

定义

1
2
Status SetExecuteStreamId(Operation *operation, uint32_t streamId);
uint32_t GetExecuteStreamId(Operation *operation);

参数列表

成员名称

参数

返回值

描述

SetExecuteStreamId

operation

streamId

Status

该接口用于设置Operation的StreamId。Operation执行时会根据设置的streamId去Context索引对应的实体Stream进行执行。该接口对于单Op与图Op的修改行为略有差异,单Op调用该接口时修改单Op本身的streamId;图Op调用该接口时会修改图中所有未设置streamId的Op的streamId。

GetExecuteStreamId

operation

uint32_t

该接口用于获取Operation的StreamId。该接口对于单Op与图Op的获取行为同样略有差异。单op调用该接口获取的是当前op的streamId,图op调用该接口时无意义,因为图中可能存在streamId不同的节点,图op的streamId无法代表整体的streamId。