cuda.jl：错误：loadError：methodError：无方法匹配typeof（fillpixel！）（:: cudevicematrix {rgb {float32}，1}）

Question

我在朱莉娅（Julia）制作了一个非常最小的射线示踪剂，并且正在实现使用CUDA的更快版本。完整的代码太广泛而无法共享，但是这是我认为与问题最相关的部分：

world = World(RGB(1, 1, 1), 5e-6, shapes, lights, 0.2, 4)
camera = Camera((0, -5000, -5000), 1000, (0, 0, 0), 1920, 1080)
canvas = CUDA.fill(world.background, camera.height, camera.width)

function fillpixel!(arr::CuArray)
    height = size(arr)[1]
    for j in 1:length(arr)
        ind = (j % height, ceil(j / height))

        ray = [([ind[2], ind[1]] - [camera.width / 2, camera.height / 2])..., camera.depth]

        (ray[2], ray[3]) = (cos(camera.rotation[1] + atan(ray[3], ray[2])), sin(camera.rotation[1] + atan(ray[3], ray[2]))) .* sqrt(ray[2]^2 + ray[3]^2)
        (ray[1], ray[3]) = (cos(camera.rotation[2] + atan(ray[3], ray[1])), sin(camera.rotation[2] + atan(ray[3], ray[1]))) .* sqrt(ray[1]^2 + ray[3]^2)
        (ray[1], ray[2]) = (cos(camera.rotation[3] + atan(ray[2], ray[1])), sin(camera.rotation[3] + atan(ray[2], ray[1]))) .* sqrt(ray[2]^2 + ray[1]^2)

        v = (Inf, nothing, nothing)

        for object in world.objects
            t = traceray(ray, camera.position, object, mindistance=camera.depth)
            t !== nothing && t[1] < v[1] && (v = (t[1], t[2], object))
        end

        v[1] != Inf && (arr[j] = computecolor(v[3].material, ray, v[1], v[2], world, camera.position .+ v[1] * ray, v[3]))
        return nothing
    end
end

@cuda fillpixel!(canvas)

但是当我运行程序时，它给我以下错误：

CUDA.jl: ERROR: LoadError: MethodError: no method matching typeof(fillpixel!)(::CuDeviceMatrix{RGB{Float32}, 1})

我无法找出导致此错误和的原因我到底做错了什么。

谢谢。

Answer 1

两个注释：fillPixel！（arr :: cuarray）将您的功能仅限于cuarray。 cuda.jl将主机侧表示cuarray转换为设备侧表示`cudeViceArray。因此，如果您放松了类型限制，则不会遇到此问题。

其次，您不想迭代您启动的内核内部的数组。您要么要使用MAP map！之类的函数来实现数据并行主义，要么使用CUDA索引原语。