ITADN

Question about how to use on gpu

#195Closedyolhan83 创建于 2024-11-16
Y
yolhan83commented
Hello, I wonder why this is not working on gpu : ```julia julia> using CUDA,FiniteDiff julia> f(x) = sum(abs2,x) f (generic function with 1 method) julia> x = rand(100,100) |> CuArray; julia> f(x) 3298.8592709432687 julia> FiniteDiff.finite_difference_gradient(f,x); ERROR: MethodError: no method matching compute_epsilon(::Val{:central}, ::CuArray{Float64, 2, CUDA.DeviceMemory}, ::Float64, ::Float64, ::Bool) The function `compute_epsilon` exists, but no method is defined for this combination of argument types. julia> FiniteDiff.finite_difference_gradient(f,collect(x)); # works ```
关闭于 2024-11-18 5 条评论