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	1147	August 16, 2020
Mapping functions using CuArrays GPU cuarrays	2	1867	September 9, 2020
CuArray is Row Major or Column Major? GPU	1	3768	November 30, 2017
How to set the diagonal part of a GPU Array GPU	6	1360	November 12, 2018
Bug in CUDA, CuArray, or something I just don't know? GPU	3	264	December 25, 2022