Merge pull request #788 from SciML/vjp_stacktrace

Re-throw the VJP failure stack traces with a verbose option

src/concrete_solve.jl

@@ -14,7 +14,12 @@ function inplace_vjp(prob, u0, p, verbose)
  nothing
  end
  true
- catch
+ catch e
+ if verbose
+ @warn "EnzymeVJP tried and failed in the automated AD choice algorithm with the following error. (To turn off this printing, add `verbose = false` to the `solve` call)\n"
+ showerror(stderr,e)
+ println()
+ end
  false
  end
  if ez
@@ -41,7 +46,12 @@ function inplace_vjp(prob, u0, p, verbose)
  return vec(du1)
  end
  ReverseDiffVJP(compile)
- catch
+ catch e
+ if verbose
+ @warn "ReverseDiffVJP tried and failed in the automated AD choice algorithm with the following error. (To turn off this printing, add `verbose = false` to the `solve` call)\n"
+ showerror(stderr,e)
+ println()
+ end
  false
  end
 
@@ -65,15 +75,25 @@ function automatic_sensealg_choice(prob::Union{SciMLBase.AbstractODEProblem,
  vjp = try
  Zygote.gradient((u, p) -> sum(prob.f(u, p, prob.tspan[1])), u0, p)
  ZygoteVJP()
- catch
+ catch e
+ if verbose
+ @warn "ZygoteVJP tried and failed in the automated AD choice algorithm with the following error. (To turn off this printing, add `verbose = false` to the `solve` call)\n"
+ showerror(stderr,e)
+ println()
+ end
  false
  end
 
  if vjp == false
  vjp = try
  ReverseDiff.gradient((u, p) -> sum(prob.f(u, p, prob.tspan[1])), u0, p)
  ReverseDiffVJP()
- catch
+ catch e
+ if verbose
+ @warn "ReverseDiffVJP tried and failed in the automated AD choice algorithm with the following error. (To turn off this printing, add `verbose = false` to the `solve` call)\n"
+ showerror(stderr,e)
+ println()
+ end
  false
  end
  end
@@ -82,7 +102,12 @@ function automatic_sensealg_choice(prob::Union{SciMLBase.AbstractODEProblem,
  vjp = try
  Tracker.gradient((u, p) -> sum(prob.f(u, p, prob.tspan[1])), u0, p)
  TrackerVJP()
- catch
+ catch e
+ if verbose
+ @warn "TrackerVJP tried and failed in the automated AD choice algorithm with the following error. (To turn off this printing, add `verbose = false` to the `solve` call)\n"
+ showerror(stderr,e)
+ println()
+ end
  false
  end
  end