friendlier error message for gradient failure

mcabbott · Dec 19, 2020 · f389f2b · f389f2b
1 parent 1d1221d
commit f389f2b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/grad/reverse.jl b/src/grad/reverse.jl
@@ -8,7 +8,7 @@ using .ReverseDiff
 ReverseDiff.@grad function (ev::Eval)(args...)
     Z = ev.fwd(ReverseDiff.value.(args)...)
     Z, Δ -> begin
-        isnothing(ev.rev) && error("no gradient definition here!")
+        ev.rev===nothing && throw("No gradient definition found! Running `@tullio` with keyword `verbose=true` may print the reason")
         ev.rev(ReverseDiff.value(Δ), Z, ReverseDiff.value.(args)...)
     end
 end
diff --git a/src/grad/tracker.jl b/src/grad/tracker.jl
@@ -8,7 +8,7 @@ using .Tracker
 Tracker.@grad function (ev::Eval)(args...)
     Z = ev.fwd(Tracker.data.(args)...)
     Z, Δ -> begin
-        isnothing(ev.rev) && error("no gradient definition here!")
+        ev.rev===nothing && throw("No gradient definition found! Running `@tullio` with keyword `verbose=true` may print the reason")
         tuple(ev.rev(Tracker.data(Δ), Z, Tracker.data.(args)...)...)
     end
 end
diff --git a/src/grad/zygote.jl b/src/grad/zygote.jl
@@ -4,7 +4,7 @@ using .Zygote
 Zygote.@adjoint function (ev::Eval)(args...)
     Z = ev.fwd(args...)
     Z, Δ -> begin
-        isnothing(ev.rev) && error("no gradient definition here!")
+        ev.rev===nothing && throw("No gradient definition found! Running `@tullio` with keyword `verbose=true` may print the reason")
         tuple(nothing, ev.rev(Δ, Z, args...)...)
     end
 end