SetExecuteStreamId
功能说明
设置Operation使用的streamId,这里的streamId为调用Context的SetExecuteStreams接口时传入streams中的序号。(例如如果需要op1在stream2上运行(streams={stream1,stream2,stream3}),那么这里调用为SetExecuteStreamId(op1, 1))。
定义
Status SetExecuteStreamId(Operation *operation, uint32_t streamId);
参数列表
参数名称 |
输入/输出 |
描述 |
---|---|---|
operation |
输入 |
被设置的Operation指针。 |
streamId |
输入 |
需要设置的streamId。 |
返回值说明
类型 |
输入/输出 |
说明 |
---|---|---|
Status |
输出 |
状态值,返回NO_ERROR表示成功,返回其它值表示失败。 |
约束说明
若自定义插件算子(PluginOperation)需要使用该功能,PluginOperation需要继承自OperationInfra接口基类。

该接口对单算子与图算子的设置行为略有差异,单算子调用该接口时修改单算子本身的StreamId,图算子调用该接口会修改图中所有streamId为0的Operation。默认的streamId为0。
父主题: atb/operation.h