testing

Author: mcabbott

src/rulesets/Base/broadcast.jl

@@ -12,7 +12,7 @@ function rrule(::typeof(Broadcast.instantiate), bc::Broadcasted)
     return Broadcast.instantiate(bc), uninstantiate
 end
 
-_print(args...) = nothing # println(join(args, " ")) # 
+_print(args...) = printstyled("CR: ", join(args, " "), color=:magenta) # nothing # 
 
 #####
 ##### Split broadcasting

test/rulesets/Base/broadcast.jl

@@ -52,6 +52,10 @@ using Base.Broadcast: broadcasted
         test_rrule(copy∘broadcasted, |>, rand(3), sin, check_inferred=false)
         _call(f, x...) = f(x...)
         test_rrule(copy∘broadcasted, _call, atan, rand(3), rand(4)', check_inferred=false)
+        
+        test_rrule(copy∘broadcasted, getindex, [rand(3) for _ in 1:2], [3,1], check_inferred=false)
+        # test_rrule(copy∘broadcasted, getindex, [rand(3) for _ in 1:2], (3,1), check_inferred=false)
+        # test_rrule(copy∘broadcasted, getindex, [rand(3) for _ in 1:2], Ref(CartesianIndex(2)), check_inferred=false)
 
         # Protected by Ref/Tuple:
         test_rrule(copy∘broadcasted, *, rand(3), Ref(rand(2)), check_inferred=false)

test/rulesets/Base/mapreduce.jl

@@ -83,6 +83,7 @@ struct SumRuleConfig <: RuleConfig{Union{HasReverseMode}} end
 
         # inference fails for array of arrays
         test_rrule(sum, sum, [[2.0, 4.0], [4.0,1.9]]; check_inferred=false)
+        test_rrule(sum, norm, collect.(eachcol(rand(3,4))); check_inferred=false)
 
         # dims kwarg
         test_rrule(sum, abs, [-2.0 4.0; 5.0 1.9]; fkwargs=(;dims=1))