add explicit errors for 2nd order

FluxML · Nov 23, 2024 · 21d05c0 · 21d05c0
1 parent 349dc99
commit 21d05c0
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/gradient.jl b/src/gradient.jl
@@ -35,6 +35,12 @@ function gradient(f, args...; zero::Bool=true)
     for a in args
         _ensure_noenzyme(a)
     end
+    if Zygote.isderiving()
+        error("""`Flux.gradient` does not support use within a Zygote gradient.
+            If what you are doing worked on Flux < 0.14, then calling `Zygote.gradiet` directly should still work.
+            If you are writing new code, then Zygote over Zygote is heavily discouraged.
+            """)
+    end
     Zygote.gradient(f, args...)
 end
 
@@ -167,6 +173,12 @@ function withgradient(f, args...; zero::Bool=true)
     for a in args
         _ensure_noenzyme(a)
     end
+    if Zygote.isderiving()
+        error("""`Flux.withgradient` does not support use within a Zygote gradient.
+            If what you are doing worked on Flux < 0.14, then calling `Zygote.gradiet` directly should still work.
+            If you are writing new code, then Zygote over Zygote is heavily discouraged.
+            """)
+    end
     Zygote.withgradient(f, args...)
 end