New issue
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
Missing functionalities - Metal with Conv and ConvTranspose layers #2278
Labels
Comments
4SAnalyticsnModelling
changed the title
Missing functionalities - Metal with Conv layers
Missing functionalities - Metal with Conv and ConvTranspose layers
Jun 21, 2023
This is more of an issue for Metal.jl. in short, someone needs to add support for https://developer.apple.com/documentation/metalperformanceshaders/convolutional_neural_network_kernels and plumb it through to NNlib.jl. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I was getting ERROR: TaskFailedException while trying the following codes:
I get the same error when I try a ConvTranspose layer instead of a Conv layer
Here's my
macOS 13.4.0, Darwin 22.5.0
Toolchain:
Julia packages:
1 device:
And here's my error trace for the first chunk of codes:
ERROR: TaskFailedException
Stacktrace:
[1] sync_end(c::Channel{Any})
@ Base ./task.jl:445
[2] macro expansion
@ ./task.jl:477 [inlined]
[3] conv!(out::MtlArray{Float32, 5}, in1::MtlArray{Float32, 5}, in2::Array{Float32, 5}, cdims::DenseConvDims{3, 3, 3, 6, 3}; kwargs::Base.Pairs{Symbol, Union{}, Tuple{}, NamedTuple{(), Tuple{}}})
@ NNlib ~/.julia/packages/NNlib/Fg3DQ/src/conv.jl:205
[4] conv!
@ ~/.julia/packages/NNlib/Fg3DQ/src/conv.jl:185 [inlined]
[5] conv!(y::MtlArray{Float32, 4}, x::MtlArray{Float32, 4}, w::Array{Float32, 4}, cdims::DenseConvDims{2, 2, 2, 4, 2}; kwargs::Base.Pairs{Symbol, Union{}, Tuple{}, NamedTuple{(), Tuple{}}})
@ NNlib ~/.julia/packages/NNlib/Fg3DQ/src/conv.jl:145
[6] conv!
@ ~/.julia/packages/NNlib/Fg3DQ/src/conv.jl:140 [inlined]
[7] conv(x::MtlArray{Float32, 4}, w::Array{Float32, 4}, cdims::DenseConvDims{2, 2, 2, 4, 2}; kwargs::Base.Pairs{Symbol, Union{}, Tuple{}, NamedTuple{(), Tuple{}}})
@ NNlib ~/.julia/packages/NNlib/Fg3DQ/src/conv.jl:88
[8] conv
@ ~/.julia/packages/NNlib/Fg3DQ/src/conv.jl:83 [inlined]
[9] (::Conv{2, 4, typeof(identity), Array{Float32, 4}, Vector{Float32}})(x::MtlArray{Float32, 4})
@ Flux ~/.julia/packages/Flux/n3cOc/src/layers/conv.jl:202
[10] top-level scope
@ ~/Documents/ecosys_June19_2023/ecosys_julia_test_run/src/test_run.jl:253
The text was updated successfully, but these errors were encountered: