I have the following which gives the correct result, but I need to keep the result on the device. diffs = CuArray([-0.1 0.2 0.1;-0.2 -0.2 -0.3]) firstmatch = map(x->CUDA.findfirst(y->y<0,x),CUDA.eachslice(diffs,dims=2)) The output is no longer on the device 3-element Array{Int64,1}: 1 2 2 Cas…

CuArray find first negative value along columns

dpsanders August 15, 2020, 11:06pm 2

I think you can / should do this by writing a kernel similar to CUDA.findfirst.

Column-wise reduction on a CUDA.CuArray matrix

Topic		Replies	Views
Column-wise reduction on a CUDA.CuArray matrix Performance question , cudanative , indexing , cuda , cuarrays	0	1166	August 16, 2020
CuArray/CUDAnative argmin paradoxical performance GPU	2	861	January 31, 2019
CuArrays: error calling CuArray() (ERROR_INVALID_DEVICE) GPU question	25	3789	February 16, 2020
Store CuArrays on a mutable struct? GPU	5	1450	July 2, 2018
Strange behavior of CuDeviceArrays General Usage gpu , cuda	7	869	June 10, 2021