函数:set_context
函数:create_context
| C函数原型 | aclError****aclrtCreateContext(aclrtContext *context, int32_t deviceId) |
|---|---|
| Python函数 | context, ret = acl.rt.create_context(device_id) |
| 函数功能 | 在当前进程或线程中显式创建一个Context。同步接口。 对于Atlas 200/500 A2推理产品,该默认Context中包含2个Stream,1个默认Stream和1个执行内部同步的Stream。 |
| 输入说明 | device_id:int,指定需要创建Context的Device设备号。 |
| 返回值说明 | context:int,表示创建的Context的指针地址。 ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 支持以下使用场景: - 若不调用acl.rt.create_context接口显式创建Context,那系统会使用默认Context,该默认Context是在调用acl.rt.set_device接口时隐式创建的。 - 隐式创建Context:适合简单、无复杂交互逻辑的应用,但缺点在于,在多线程编程中,执行结果取决于线程调度的顺序。 - (推荐)显式创建Context:适合大型、复杂交互逻辑的应用,且便于提高程序的可读性、可维护性。 - 若在某一进程内创建多个Context(Context的数量与Stream相关,Stream数量有限制,请参见acl.rt.create_stream),当前线程在同一时刻内只能使用其中一个Context,建议通过acl.rt.set_context接口明确指定当前线程的Context,增加程序的可维护性**。** (说明:) |
| 参考资源 | 接口调用流程与示例,请参见运行管理资源申请与释放、同步等待。 |
父主题: Context管理
函数:destroy_context
| C函数原型 | aclError aclrtDestroyContext(aclrtContext context) |
|---|---|
| Python函数 | ret = acl.rt.destroy_context(context) |
| 函数功能 | 销毁一个Context,释放Context的资源,同步接口。只能销毁通过acl.rt.create_context接口创建的Context。 |
| 输入说明 | context:int,指定需要销毁的Context对象指针地址。 |
| 返回值说明 | ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 无 |
| 参考资源 | 接口调用流程与示例,请参见运行管理资源申请与释放、同步等待。 |
父主题: Context管理
函数:set_context
| C函数原型 | aclError aclrtSetCurrentContext(aclrtContext context) |
|---|---|
| Python函数 | ret = acl.rt.set_context(context) |
| 函数功能 | 设置线程的Context,同步接口。 |
| 输入说明 | context:int,线程当前的Context对象指针地址。 |
| 返回值说明 | ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | - 支持以下场景: - 如果在某线程(例如:thread1)中调用acl.rt.create_context接口显式创建一个Context(例如:ctx1),则可以不调用acl.rt.set_context接口指定该线程的Context,系统默认将ctx1作为thread1的Context。 - 如果没有调用acl.rt.create_context接口显式创建Context,则系统将默认Context作为线程的Context,此时,不能通过acl.rt.destroy_context接口来释放默认Context。 - 如果多次调用acl.rt.set_context接口设置线程的Context,以最后一次为准。 - 若给线程设置的Context所对应的Device已经被复位,则不能将该Context设置为线程的Context,否则会导致业务异常。 - 推荐在某一线程中创建的Context,在该线程中使用。若在线程A中调用acl.rt.create_context接口创建Context,在线程B中使用该Context,则需由用户自行保证两个线程中同一个Context下同一个Stream中任务执行的顺序。 |
| 参考资源 | 接口调用流程,参见多Device场景。 |
父主题: Context管理
函数:get_context
| C函数原型 | aclError aclrtGetCurrentContext(aclrtContext *context) |
|---|---|
| Python函数 | context, ret = acl.rt.get_context() |
| 函数功能 | 获取线程的Context,同步接口。 如果用户多次调用acl.rt.set_context接口设置当前线程的Context,则获取的是最后一次设置的Context。 |
| 输入说明 | 无 |
| 返回值说明 | context:int,表示创建的Context对象指针地址。 ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 无 |
父主题: Context管理
函数:get_overflow_status
| C函数原型 | aclError aclrtGetOverflowStatus(void *outputAddr, size_t outputSize, aclrtStream stream) |
|---|---|
| Python函数 | ret = acl.rt.get_overflow_status(outputAddr, outputSize, stream) |
| 函数功能 | 获取当前Context下所有Stream上任务的溢出状态,并将状态值拷贝到用户申请的Device内存中。异步接口。 |
| 输入说明 | outputAddr:int,用户申请的Device内存,需通过acl.rt.malloc接口申请。如果需要在Host侧查看数据,可调用acl.rt.memcpy或acl.rt.memcpy_async接口,将Device侧的数据传输到Host侧。 outputSize:int,需申请的Device内存大小,单位Byte,固定大小为64Byte。 stream:int,指定Stream的指针地址,用于下发溢出状态查询任务。 |
| 返回值说明 | ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 本接口是异步接口,在调用之后,需调用acl.rt.synchronize_stream接口阻塞应用程序运行,直到指定Stream中的溢出状态查询任务已完成。 |
父主题: Context管理
函数:reset_overflow_status
| C函数原型 | aclError aclrtResetOverflowStatus(aclrtStream stream) |
|---|---|
| Python函数 | ret = acl.rt.reset_overflow_status(stream) |
| 函数功能 | 清除当前Context下所有Stream上任务的溢出状态。异步接口。 |
| 输入说明 | stream:int,指定Stream的指针地址,用于下发溢出状态复位任务。 |
| 返回值说明 | ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 本接口是异步接口,在调用之后,需调用acl.rt.synchronize_stream接口阻塞应用程序运行,直到指定Stream中的溢出状态复位任务已完成。 |
父主题: Context管理
在线提单