昇腾社区首页
中文
注册

atb/context.h

定义

1
2
virtual Status SetExecuteStreams(const std::vector<aclrtStream> &streams) = 0;
virtual std::vector<aclrtStream> GetExecuteStreams() = 0;

参数列表

成员名称

参数

返回值

描述

SetExecuteStreams

streams

Status

设置实际的执行流,Operation执行时会根据配置的streamId从Context匹配对应的实际执行流。

  • streams:需要设置的一组stream。

状态值,如果设置成功,返回NO_ERROR。如果失败,请参见ErrorType

GetExecuteStreams

-

std::vector<aclrtStream>

获取Context中当前设置的一组执行流。

如果设置成功,返回Context当前设置的一组执行流。