test

Author: jumerckx

test/basic.jl

@@ -56,6 +56,16 @@ sinexpbc(x) = sinexp.(x)
     @test @jit(sinexpbc(a)) ≈ r_res
 end
 
+function throw_if_within_autodiff()
+    if Enzyme.within_autodiff()
+        error("`within_autodiff` returned true even though we are not in an autodiff call.")
+    end
+    return nothing
+end
+@testset "within_autodiff" begin
+    @test isnothing(@jit throw_if_within_autodiff())
+end
+
 sumexp(x) = sum(exp, x)
 
 sum_compare(x) = sum(x) > 0