diff --git a/test/rulesets/Base/sort.jl b/test/rulesets/Base/sort.jl index 564817feb..d06067bd2 100644 --- a/test/rulesets/Base/sort.jl +++ b/test/rulesets/Base/sort.jl @@ -8,14 +8,16 @@ test_rrule(sort, a) test_rrule(sort, a; fkwargs=(;rev=true)) - a = rand(5, 4) - for dims in (1, 2) - # fwd - test_frule(sort, a; fkwargs=(;dims)) - test_frule(sort, a; fkwargs=(;dims, rev=true)) - # rev - test_rrule(sort, a; fkwargs=(;dims)) - test_rrule(sort, a; fkwargs=(;dims, rev=true)) + if VERSION ≥ v"1.9" + a = rand(5, 4) + for dims in (1, 2) + # fwd + test_frule(sort, a; fkwargs=(;dims)) + test_frule(sort, a; fkwargs=(;dims, rev=true)) + # rev + test_rrule(sort, a; fkwargs=(;dims)) + test_rrule(sort, a; fkwargs=(;dims, rev=true)) + end end end @testset "partialsort" begin