We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
chunk
See the discussion in FluxML/Metalhead.jl#165 and JuliaGPU/CUDA.jl#1542.
The problem is that we don't want wrapped cuarray types since it will affect the following dispatches, right @theabhirath @darsnack?
using CUDA, MLUtils julia> x = rand(2, 10) |> cu; julia> a = chunk(x, 2)[1] # wrapped cuarray 2×2 view(::CuArray{Float32, 2, CUDA.Mem.DeviceBuffer}, :, 1:2) with eltype Float32: 0.545259 0.936495 0.174116 0.514381 julia> view(x, :, 1:2) # just a cuarray 2×2 CuArray{Float32, 2, CUDA.Mem.DeviceBuffer}: 0.545259 0.936495 0.174116 0.514381
The text was updated successfully, but these errors were encountered:
Yeah that's the issue. Is there a difference between what selectdim returns vs. a manually written view?
selectdim
view
Sorry, something went wrong.
There is, all the gory details (and a possible fix) are in the CUDA.jl issue.
Successfully merging a pull request may close this issue.
See the discussion in FluxML/Metalhead.jl#165
and JuliaGPU/CUDA.jl#1542.
The problem is that we don't want wrapped cuarray types since it will affect the following dispatches, right @theabhirath @darsnack?
The text was updated successfully, but these errors were encountered: