Device内的数据传输
同步内存复制
调用接口后,需增加异常处理的分支,并记录报错日志、提示日志,此处不一一列举。以下是关键步骤的代码示例,不可以直接拷贝运行,仅供参考,示例代码如下:
import acl # ...... # 1.申请内存。 size = 1 * 1024 * 1024 dev_ptr_a, ret = acl.rt.malloc(size, ACL_MEM_MALLOC_NORMAL_ONLY) dev_ptr_b, ret = acl.rt.malloc(size, ACL_MEM_MALLOC_NORMAL_ONLY) # 2.申请内存后,可向内存中读入数据,该自定义函数read_file由用户实现。 read_file(fileName, dev_ptr_a, size) # 3.同步内存复制。 # 同步内存复制,dev_ptr_a表示Device上源内存地址指针,dev_ptr_b表示Device上目的内存地址指针,size表示内存大小。 # ACL_MEMCPY_DEVICE_TO_DEVICE = 3。 ret = acl.rt.memcpy(dev_ptr_b, size, dev_ptr_a, size, ACL_MEMCPY_DEVICE_TO_DEVICE) # 4.使用完内存中的数据后,需及时释放资源。 ret = acl.rt.free(dev_ptr_a) ret = acl.rt.free(dev_ptr_b) # ......
异步内存复制
调用接口后,需增加异常处理的分支,并记录报错日志、提示日志,此处不一一列举。以下是关键步骤的代码示例,不可以直接拷贝运行,仅供参考,示例代码如下:
import acl # ...... # 1.申请内存。 size = 1 * 1024 * 1024 dev_ptr_a, ret = acl.rt.malloc(size, ACL_MEM_MALLOC_NORMAL_ONLY) dev_ptr_b, ret = acl.rt.malloc(size, ACL_MEM_MALLOC_NORMAL_ONLY) # 2.申请内存后,可向内存中读入数据,该自定义函数read_file由用户实现。 read_file(fileName, dev_ptr_a, size) # 3.异步内存复制。 # 异步内存复制,dev_ptr_a表示Device上源内存地址指针,dev_ptr_b表示Device上目的内存地址指针,size表示内存大小。 # ACL_MEMCPY_DEVICE_TO_DEVICE = 3。 stream = acl.rt_create_stream() ret = acl.rt.memcpy_async(dev_ptr_b, size, dev_ptr_a, size, ACL_MEMCPY_DEVICE_TO_DEVICE, stream) ret = acl.rt.synchronize_stream(stream) # 4.使用完内存中的数据后,需及时释放资源。 ret = acl.rt.destroy_stream(stream) ret = acl.rt.free(dev_ptr_a) ret = acl.rt.free(dev_ptr_b) # ......
父主题: 数据传输