GPU 中结构体计算

在提问之前请确定你已经努力阅读了文档,并且尝试自己在互联网上搜索。

请尽可能提供你的demo代码或者GitHub的gist地址。

# code

请问怎么把结构体转移到GPU里计算呢?
结构体中每个元素对应的是不定长的数组,因此没有办法将其数据转变为CuArray,
除了CuArray还有其他库包或函数可以将CPU结构体直接转移到GPU 上计算吗?
十分着急,万分感谢!
image

如果把内部的数组作为GPU上的array能满足你的需求的话:

julia> using CUDA
julia> mutable struct partial2
           ID::Int16
           postition::CuArray{Float64, 1}
       end

julia> pt = partial2(1, Float64[1, 2, 3])

julia> pt.postition
3-element CuArray{Float64, 1}:
 1.0
 2.0
 3.0

京ICP备17009874号-2