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

SetAttachedStreamInfos

函数功能

为当前算子附着一组Stream信息,用于指定该算子在哪个执行流(Stream)上运行。

函数原型

1
ge::graphStatus SetAttachedStreamInfos(std::vector<StreamInfo> &stream_info_vec) const

参数说明

参数

输入/输出

说明

stream_info_vec

输入

执行流信息。std::vector<StreamInfo>类型。

返回值说明

返回graphStatus状态码。

约束说明

调用示例

1
2
3
4
5
ge::graphStatus SetAttachedStreamInfos(ExeResGenerationContext* context) {
  std::vector<StreamInfo> stream_info_vec;
  auto status = context->SetAttachedStreamInfos(stream_info_vec);
  ...
}