Skip to content

Commit

Permalink
delete redundancy
Browse files Browse the repository at this point in the history
  • Loading branch information
ZhiyiLi committed Dec 19, 2023
1 parent f35f2be commit 5dff476
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions src/backend/compiler_python.jl
Original file line number Diff line number Diff line change
Expand Up @@ -115,13 +115,12 @@ function to_python_str(graphs::AbstractVector{<:AbstractGraph}, framework::Symbo
end
head *= "def graphfunc(root,leaf):\n"
tail = "\n"

if framework == :jax
tail *="graphfunc_jit = jit(graphfunc)"
end
expr = head * body * tail
# println(expr)
# return head * body * tail

return expr, leafidx , gid_to_leafid
end
function compile_python(graphs::AbstractVector{<:AbstractGraph}, framework::Symbol=:jax, filename::String="GraphFunc.py")
Expand Down

0 comments on commit 5dff476

Please sign in to comment.