File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 2525 @test g_gpu isa CuArray
2626 @test g_gpu |> collect ≈ g
2727
28- # https://github.com/FluxML/Zygote.jl/issues/1027
28+ # https://github.com/FluxML/Zygote.jl/issues/1027 # status on Zygote v0.6.14, CUDA v3.3.0 in comments:
2929 @test gradient (x -> sum (x .!= 0 ), a_gpu) == (nothing ,) # was MethodError: no method matching iterate(::Nothing)
30+ @test gradient (x -> sum (x .> 3 ), a_gpu) == (nothing ,)
3031 g3 = gradient (x -> sum (x .^ 3 ) / count (x .> 3 ), a)[1 ] # was Can't differentiate gc_preserve_end expression
31- @test cu (g3) ≈ gradient (x -> sum (x .^ 3 ) / sum (x .> 3 ), a_gpu)[1 ] # was KernelException -- Zygote v0.6.14, CUDA v3.3.0
32+ @test_skip cu (g3) ≈ gradient (x -> sum (x .^ 3 ) / sum (x .> 3 ), a_gpu)[1 ] # was KernelException -- not fixed by PR #1018
33+ @test cu (g3) ≈ gradient (x -> sum (x .^ 3 ) / count (x .> 3 ), a_gpu)[1 ]
3234end
3335
3436@testset " sum(f, x)" begin
You can’t perform that action at this time.
0 commit comments