下载
中文
注册

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

aclrtStream stream

Status

stream表示了Device侧的执行流, 每个流具体表现为一条执行队列。 通过这个函数设置使用传入的流执行对应的Operation。

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

GetExecuteStream

NA

aclrtStream

获取当前执行stream。

SetAsyncTilingCopyStatus

bool enable

Status

设置异步拷贝tiling信息功能是否开启,如果是,则创建Stream和Event来进行tiling拷贝过程。

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

GetAsyncTilingCopyStatus

NA

bool

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