more interp catches

2017-02-20 21:49:47 +00:00 · 2017-02-20 21:49:47 +00:00 · b92252cd5c
commit b92252cd5c
parent 242b4d6e23
1 changed files with 3 additions and 1 deletions
--- a/src/backend/mxnet/graph.jl
+++ b/src/backend/mxnet/graph.jl
@ -74,8 +74,10 @@ function graph(ctx::Context, model, args...)
  interpret(ctx, g, args...)
 end

+graph′(ctx::Context, args...) = @ithrow ctx graph(ctx, args...)
+
 function tograph(model, args...)
-  ctx = Context(mux(iline, ilambda, imap, iargs, ituple, graph),
+  ctx = Context(mux(iline, ilambda, imap, iargs, ituple, graph′),
                params = Dict(), stacks = Dict())
  out = @icatch graph(ctx, model, args...)
  return ctx[:params], ctx[:stacks], out