I have a vector ArrayPartition that looks like this: 5-element Vector{ArrayPartition{Float32, Tuple{CuArray{Float32, 2, CUDA.Mem.DeviceBuffer}, CuArray{Float32, 2, CUDA.Mem.DeviceBuffer}}}} I want to reshape this into an CuArray that I can pass to the next layer or compute the loss on. using: r…

Converting/Reshaping a vector of ArrayPartition to an Array on the gpu

ChrisRackauckas February 28, 2024, 7:50pm 2

Try converting it to a VectorOfArray first? Or try using Adapt.jl.

Open an issue and we can figure this one out.

Topic		Replies	Views
GPU Map without reduction on multiple arrays indices GPU	1	706	February 8, 2019
The performance difference of transferring (SubArray, ReshapedArray) Array to GPU GPU flux	2	662	November 20, 2019
Using vectors of vectors with CuArrays GPU	2	1544	July 9, 2019
Passing a wrapped array to a kernel GPU	2	552	May 27, 2020
Is it possible to index a CuArray with a CuArray? GPU question	1	854	January 11, 2019