Skip to content

Commit 1149a4c

Browse files
authored
Drive by fix: NewVarNode in forward mode
1 parent 302478a commit 1149a4c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/stage1/recurse_fwd.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,8 @@ function transform_fwd!(ci, meth, nargs, sparams, N)
5555
return SlotNumber(2 + stmt.id)
5656
elseif isa(stmt, Argument)
5757
return SlotNumber(2 + stmt.n)
58+
elseif isa(stmt, NewvarNode)
59+
return NewvarNode(SlotNumber(2 + stmt.slot.id))
5860
elseif isa(stmt, ReturnNode)
5961
return ReturnNode(emit!(mapstmt!(stmt.val)))
6062
elseif isa(stmt, GotoNode)

0 commit comments

Comments
 (0)