From fd8db7354fd3a4fb82e6728d6bf1e39baff17ab4 Mon Sep 17 00:00:00 2001
From: Avik Pal <avikpal@iitk.ac.in>
Date: Sat, 27 Jul 2024 23:40:47 -0400
Subject: [PATCH] fix: skip FiniteDiff on GPU too slow

---
 src/autodiff.jl | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/autodiff.jl b/src/autodiff.jl
index 6e2b66d..bdc4d2a 100644
--- a/src/autodiff.jl
+++ b/src/autodiff.jl
@@ -119,13 +119,14 @@ function test_gradients(f, args...; skip_backends=[], broken_backends=[], kwargs
         if AutoForwardDiff() ∉ skip_backends && total_length ≤ 100
             push!(backends, AutoForwardDiff())
         end
+        if AutoFiniteDiff() ∉ skip_backends && total_length ≤ 100
+            push!(backends, AutoFiniteDiff())
+        end
+        # TODO: Move Enzyme out of here once it supports GPUs
         if AutoEnzyme() ∉ skip_backends && ENZYME_TESTING_ENABLED
             push!(backends, AutoEnzyme())
         end
     end
-    if AutoFiniteDiff() ∉ skip_backends && total_length ≤ 100
-        push!(backends, AutoFiniteDiff())
-    end
     AutoTracker() ∉ skip_backends && push!(backends, AutoTracker())
 
     # Test the gradients