TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

make mxerr more robust

Browse Source
This commit is contained in:
Mike J Innes 2017-02-23 16:58:29 +00:00
parent 21775f8c89
commit f230b7cebf
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/backend/mxnet/graph.jl
View File

@ -105,6 +105,7 @@ macro mxerr(stk, ex)
catch e catch e
(isa(e, mx.MXError) && (node = errnode(e)) != nothing) || rethrow() (isa(e, mx.MXError) && (node = errnode(e)) != nothing) || rethrow()
stk = $(esc(stk)) stk = $(esc(stk))
haskey(stk, node) || rethrow()
throw(Exception(e, totrace(stk[node]))) throw(Exception(e, totrace(stk[node])))
end) end)
end end

1
test/backend.jl
View File

@ -25,6 +25,7 @@ end
e = try mxnet(model, (10, 1)) e = try mxnet(model, (10, 1))
catch e e end catch e e end
@test isa(e, DataFlow.Interpreter.Exception)
@test e.trace[1].func == Symbol("Flux.Affine") @test e.trace[1].func == Symbol("Flux.Affine")
@test e.trace[2].func == :TLP @test e.trace[2].func == :TLP
end end