将cuda。@原子枚举扩展到自定义结构

Question

我想知道天气可能会将cuda。@原子操作扩展到自定义类型。

这是我要做的事情的一个示例：

using CUDA

struct Dual
    x
    y
end 

cu0 = CuArray([Dual(1, 2), Dual(2,3)])
cu1 = CuArray([Dual(1, 2), Dual(2,3)])

indexes = CuArray([1, 1])


function my_kernel(dst, src, idx)
   index = threadIdx().x + (blockIdx().x - 1) * blockDim().x

    @inbounds if index <= length(idx)
       CUDA.@atomic dst[idx[index]] = dst[idx[index]] + src[index]
    end
    return nothing
end

@cuda threads = 100 my_kernel(cu0, cu1, indexes)

该代码的问题是cuda。@atomic调用仅支持基本类型 int，浮动或真实。

我需要它与自己的结构一起工作。

如果有人知道这是怎么可能的，那会很好。

Answer 1

CUDA的基础PTX指令集提供了原子商店的子集，交换，添加/减，增量/减少，最小/最大和比较全局和共享存储器位置（并非所有架构都支持所有操作都与所有架构一起使用所有操作， POD类型，并且有证据表明，并非所有操作都在所有架构上实施）。

所有这些指令的共同点是它们仅在原子上执行一个操作。我完全不熟悉Julia，但是如果是

CUDA.@atomic dst[idx[index]] = dst[idx[index]] + src[index]

“ atmanthys”添加src []。x和src []。y to dst []。x和dst []。y“这是不可能的，因为这意味着在一个原子操作中的单独存储位置上有两个添加。如果您的结构成员可以被包装成兼容类型（例如，一个32位或64位未签名的整数），则可以在CUDA中执行原子商店，交换或比较。但不是算术。

如果您咨询此部分编程指南，您可以使用紧密循环中的比较和集合看到蛮力双精度添加实现的示例。 如果您的结构可以包装成可以用比较和集合来操纵的东西，那么可能会以自定义类型滚动自己的原子添加（最多只有64位） 。

在朱莉娅（Julia）中，您如何处理绝对是读者的练习。