You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository has been archived by the owner on Mar 12, 2021. It is now read-only.
var/std seem to fall back on indexing the CuArray, instead of a vectorized implementation:
julia> var(c,dims=1)
┌ Warning: Performing scalar operations on GPU arrays: This is very slow, consider disallowing these operations with `allowscalar(false)`
└ @ GPUArrays ~/.julia/packages/GPUArrays/J4c3Q/src/indexing.jl:16
1×4 CuArray{Float32,2}:
0.0646661 0.167767 0.0156315 0.14948
This in reference to FluxML/Flux.jl#981 var and std is defined in Statistics.jl. If I want to add functions specific to CuArray where should I add them in this package. @MikeInnes
var/std seem to fall back on indexing the CuArray, instead of a vectorized implementation:
This makes it slower than cpu:
The text was updated successfully, but these errors were encountered: