SetAttachedStreamInfos
函数功能
为当前算子附着一组Stream信息,用于指定该算子在哪个执行流(Stream)上运行。
函数原型
ge::graphStatus SetAttachedStreamInfos(std::vector<StreamInfo> &stream_info_vec) const
参数说明
执行流信息。std::vector<StreamInfo>类型。 |
返回值说明
返回graphStatus状态码。
约束说明
无
调用示例
ge::graphStatus SetAttachedStreamInfos(ExeResGenerationContext* context) {
std::vector<StreamInfo> stream_info_vec;
auto status = context->SetAttachedStreamInfos(stream_info_vec);
...
}