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

fixes

Browse Source
This commit is contained in:
Mike J Innes 2017-02-01 17:06:42 +05:30
parent 7c450c3200
commit 00a5acde46
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/compiler/interp.jl
View File

@ -26,14 +26,17 @@ end
imap(f, args...) = f(args...)
function interp(ctx, f, xs...)
@ithrow ctx f(xs...)
g = graph(f)
@ithrow(ctx, g ≠ nothing ?
interpret(ctx, reifyparams(g), xs...) :
f(xs...))
end
# TODO: batching should be secondary
function interpmodel(m, args::Batch...)
ctx = Context(mux(iline, ilambda, iconst, iargs, ituple, interp))
rebatch(@icatch interpret(ctx, reifyparams(graph(m)), map(rawbatch, args)...))
rebatch(@icatch interp(ctx, m, map(rawbatch, args)...))
end
interpmodel(m, args...) = unbatchone(interpmodel(m, batchone(args)...))