昇腾社区首页
中文
注册

Context

说明

加速库上下文类,主要用于管理Operation运行所需要的全局资源。

Context类会管理任务流队列比如Operation执行以及TilingCopy,管理tiling内存的申请与释放。

定义

class Context {
public:
    Context() = default;
    virtual ~Context() = default;
    virtual Status SetExecuteStream(aclrtStream stream) = 0;
    virtual aclrtStream GetExecuteStream() const = 0;
    virtual Status SetAsyncTilingCopyStatus(bool enable) = 0;
    virtual bool GetAsyncTilingCopyStatus() const = 0;
};

成员

成员名称

描述

SetExecuteStream

将传入“stream”队列设置为当前执行队列。然后再去执行对应的Operation。

如果设置成功,返回NO_ERROR。

GetExecuteStream

获取当前执行“stream”队列。执行流队列。

SetAsyncTilingCopyStatus

设置异步拷贝tiling信息功能是否开启,默认为是,表示创建Stream和Event来进行tiling拷贝过程。

如果设置成功,返回NO_ERROR。

GetAsyncTilingCopyStatus

获取tiling拷贝状态,如果获取成功,返回True。