Move

Author: gdalle

DifferentiationInterface/test/Back/Enzyme/hints.jl

@@ -148,3 +148,56 @@ end
         f_nocontext, AutoEnzyme(; mode=Enzyme.Reverse), rand(10), ConstantOrCache(nothing)
     )
 end
+
+@testset "Hints" begin
+    @testset "MutabilityError" begin
+        f = let
+            cache = [0.0]
+            x -> sum(copyto!(cache, x))
+        end
+
+        e = nothing
+        try
+            gradient(f, AutoEnzyme(), [1.0])
+        catch e
+        end
+        msg = sprint(showerror, e)
+        @test occursin("AutoEnzyme", msg)
+        @test occursin("function_annotation", msg)
+        @test occursin("ADTypes", msg)
+        @test occursin("DifferentiationInterface", msg)
+    end
+
+    @testset "RuntimeActivityError" begin
+        function g(active_var, constant_var, cond)
+            if cond
+                return active_var
+            else
+                return constant_var
+            end
+        end
+
+        function h(active_var, constant_var, cond)
+            return [g(active_var, constant_var, cond), g(active_var, constant_var, cond)]
+        end
+
+        e = nothing
+        try
+            pushforward(
+                h,
+                AutoEnzyme(; mode=Enzyme.Forward),
+                [1.0],
+                ([1.0],),
+                Constant([1.0]),
+                Constant(true),
+            )
+        catch e
+        end
+        msg = sprint(showerror, e)
+        @test occursin("AutoEnzyme", msg)
+        @test occursin("mode", msg)
+        @test occursin("set_runtime_activity", msg)
+        @test occursin("ADTypes", msg)
+        @test occursin("DifferentiationInterface", msg)
+    end
+end