昇腾社区首页
中文
注册
开发者
下载

SetStreamId

函数功能

设置Task的流id。

函数原型

1
void SetStreamId(uint32_t stream_id)

参数说明

参数

输入/输出

说明

stream_id

输入

流id。

返回值说明

约束说明

调用示例

1
2
3
4
5
6
7
graphStatus Mc2GenTaskCallback(const gert::ExeResGenerationContext *context,
    std::vector<std::vector<uint8_t>> &tasks) {
  ...
  auto wait_task = KernelLaunchInfo::CreateHcomWaitTask(context);
  wait_task.SetStreamId(attach_stream_id);
  ...
}