GPUArrays.jl icon indicating copy to clipboard operation
GPUArrays.jl copied to clipboard

Published 20 hours ago • verified publisher icon JuliaGPU

mean fails with views

Open cossio opened this issue 2 years ago • 1 comments 

A = gpu(rand(3, 5, 2))
mean(@views A[1, :, :]; dims=2)
# ERROR: Scalar indexing is disallowed.

However, the following all work:

sum(@views A[1, :, :]; dims=2)
mean(A[1, :, :]; dims=2)
sum(A[1, :, :]; dims=2)

cossio avatar Jun 13 '23 15:06 cossio