Enzyme to downstream test

Author: ChrisRackauckas

test/downstream/Project.toml

@@ -3,6 +3,7 @@ Calculus = "49dc2e85-a5d0-5ad3-a950-438e2897f1b9"
 DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
 DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"
 DiffEqCallbacks = "459566f4-90b8-5000-8ac3-15dfb0a30def"
+Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 Measurements = "eff96d63-e80a-5855-80a2-b1b0885c5ab7"
 ModelingToolkit = "961ee093-0014-501f-94e3-6117800e7a78"

test/downstream/enzyme.jl

@@ -0,0 +1,23 @@
+using Enzyme, EnzymeTestUtils
+using DiffEqBase: fastlog2, fastpow
+using Test
+
+@testset "Fast pow - Enzyme forward rule" begin
+    @testset for RT in (Duplicated, DuplicatedNoNeed),
+        Tx in (Const, Duplicated),
+        Ty in (Const, Duplicated)
+        x = 3.0
+        y = 2.0
+        test_forward(fastpow, RT, (x, Tx), (y, Ty), atol=0.005, rtol=0.005)
+    end
+end
+
+@testset "Fast pow - Enzyme reverse rule" begin
+    @testset for RT in (Active,),
+        Tx in (Active,),
+        Ty in (Active,)
+        x = 2.0
+        y = 3.0
+        test_reverse(fastpow, RT, (x, Tx), (y, Ty), atol=0.001, rtol=0.001)
+    end
+end

test/fastpow.jl

@@ -1,5 +1,4 @@
 using DiffEqBase: fastlog2, fastpow
-using Enzyme, EnzymeTestUtils
 using Test
 
 @testset "Fast log2" begin
@@ -13,24 +12,4 @@ end
     @test fastpow(1.0, 1.0) isa Float32
     errors = [abs(^(x, y) - fastpow(x, y)) for x in 0.001:0.001:1, y in 0.08:0.001:0.5]
     @test maximum(errors) < 1e-4
-end
-
-@testset "Fast pow - Enzyme forward rule" begin
-    @testset for RT in (Duplicated, DuplicatedNoNeed),
-        Tx in (Const, Duplicated),
-        Ty in (Const, Duplicated)
-        x = 3.0
-        y = 2.0
-        test_forward(fastpow, RT, (x, Tx), (y, Ty), atol=0.005, rtol=0.005)
-    end
-end
-
-@testset "Fast pow - Enzyme reverse rule" begin
-    @testset for RT in (Active,),
-        Tx in (Active,),
-        Ty in (Active,)
-        x = 2.0
-        y = 3.0
-        test_reverse(fastpow, RT, (x, Tx), (y, Ty), atol=0.001, rtol=0.001)
-    end
 end