如何在主机 C++ 函数之间传递 cl::Buffer 对象？

Question

我将 OpenCL 与 C++ 包装器一起使用，使用 GPU 进行繁重的计算。

数据在 cl::Buffer 个对象中，例如

std::vector<float> host_z_vec(100, 0.0f);
cl::Buffer device_z_vec;
device_z_vec = cl::Buffer(*context_ptr,
                          CL_MEM_READ_WRITE | CL_MEM_COPY_HOST_PTR, 
                          sizeof(float)*host_z_vec.size(),   
                          &host_z_vec.at(0),
                          &status);

我应该如何在主机 C++ 函数之间传递 cl::Buffer 对象 device_z_vec？ - 我不想制作 device_z_vec 的副本，只是传递它。

void run_on_gpu(cl::Buffer device_z_vec){ ... }

或

void run_on_gpu(const cl::Buffer& device_z_vec){ ... }

?

这样可以吗？

class StoreBuffer{
    cl::Buffer device_z_vec;
public:
    cl::Buffer provide_access_to_device_z_vec(){ return device_z_vec; };
}

或者我应该传递指向 cl::Buffer 对象的指针吗？

Answer 1

按值将 cl::Buffer 传递给函数或将其存储为 class 成员都可以。 cl::Buffer class 的复制构造函数只是增加底层 OpenCL 分配的引用计数（通过 clRetainMemObject）。 cl.hpp.

中的所有其他 OpenCL 对象 classes 也是如此

如何在主机 C++ 函数之间传递 cl::Buffer 对象？

How do I pass cl::Buffer object between host c++ functions?

c++

opencl