gpt4 book ai didi

c++ - OpenCL:从 ' float* ' 到 ' cl_mem ' 的转换

转载 作者:太空宇宙 更新时间:2023-11-04 13:05:57 24 4
gpt4 key购买 nike

在 caffe 源代码(分支 opencl)中,有时我可以看到这样的一些(例如在 src/caffe/layers/base_conv_layer.cpp line 400):

void func(cl_mem a, ...)
{...}

void func2(float* a, ...)
{
...
func( (cl_mem) a, ...);
...
}

我觉得这很奇怪,所以我用谷歌搜索了一下,但一无所获。

然后我写了一个简单的测试代码来弄清楚发生了什么,但是我发现这个转换是没有意义的:

std::cout << long((void*)a) << " " << long((void*)(cl_mem(a))) << std::endl;

他们的值(value)是一样的!

所以我的问题是为什么这种转换在 caffe opencl 中没有问题?此转换中发生了什么?

最佳答案

如果您查看 SyncedMemory 类(在 include/caffe/syncedmem.hppsrc/caffe/syncedmem.cpp 中),您会发现clCreateBuffer首先创建了一个内存缓冲区,并将句柄存储为一个cl_mem对象。然后在调用 gpu_data()mutable_gpu_data() 时将句柄转换为 DType* 类型。因此它可以转换回 cl_mem 类型,因为它首先是一个 cl_mem 对象。

关于c++ - OpenCL:从 ' float* ' 到 ' cl_mem ' 的转换,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42359848/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com