Skip to content

Commit 158ca75

Browse files
authored
Merge pull request #693 from JuliaDiff/ox/gu_imum_pass
remove broken marking on max/min-imum gpu tests
2 parents 749fb1e + afe8e78 commit 158ca75

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/rulesets/Base/array.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -366,7 +366,7 @@ end
366366
test_rrule(findmax, rand(5,3))
367367
@test [0 0; 0 5] == @inferred unthunk(rrule(findmax, [1 2; 3 4])[2]((5.0, nothing))[2])
368368
@test [0 0; 0 5] == @inferred unthunk(rrule(findmax, [1 2; 3 4])[2]((5.0, NoTangent()))[2])
369-
369+
370370
# Reverse with dims:
371371
@test [0 0; 5 6] == @inferred unthunk(rrule(findmax, [1 2; 3 4], dims=1)[2](([5 6], nothing))[2])
372372
@test [5 0; 6 0] == @inferred unthunk(rrule(findmin, [1 2; 3 4], dims=2)[2]((hcat([5,6]), nothing))[2])
@@ -379,14 +379,14 @@ end
379379
# Forward
380380
test_frule(imum, rand(10))
381381
test_frule(imum, rand(3,4))
382-
@gpu_broken test_frule(imum, rand(3,4), fkwargs=(dims=1,))
382+
@gpu test_frule(imum, rand(3,4), fkwargs=(dims=1,))
383383
test_frule(imum, [rand(2) for _ in 1:3])
384384
test_frule(imum, [rand(2) for _ in 1:3, _ in 1:4]; fkwargs=(dims=1,))
385385

386386
# Reverse
387387
test_rrule(imum, rand(10))
388388
test_rrule(imum, rand(3,4))
389-
@gpu_broken test_rrule(imum, rand(3,4), fkwargs=(dims=1,))
389+
@gpu test_rrule(imum, rand(3,4), fkwargs=(dims=1,))
390390
test_rrule(imum, rand(3,4,5), fkwargs=(dims=(1,3),))
391391

392392
# Arrays of arrays

0 commit comments

Comments
 (0)