Context
说明
Context接口类,主要用于设置Operation上下文,如stream,tiling属性等。
定义
class Context { public: Context() = default; virtual ~Context() = default; virtual Status SetExecuteStream(aclrtStream stream) = 0; virtual aclrtStream GetExecuteStream() const = 0; };
成员
成员名称 |
描述 |
---|---|
SetExecuteStream |
设置算子执行的运行Stream。 |
GetExecuteStream |
获取算子执行的Stream。 |
父主题: atb/context.h