Skip to content

Commit 4fbdf10

Browse files
committed
rename funcs
1 parent d5d5333 commit 4fbdf10

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/static_ir/backprop.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -325,7 +325,7 @@ function back_codegen!(stmts, ir, selected_calls, fwd_marked, back_marked,
325325
end
326326
end
327327

328-
function generate_value_choice_gradient(selected_choices::Set{RandomChoiceNode},
328+
function generate_value_gradient_choices(selected_choices::Set{RandomChoiceNode},
329329
selected_calls::Set{GenerativeFunctionCallNode},
330330
value_choices::Symbol, gradient_choices::Symbol)
331331
selected_choices_vec = collect(selected_choices)
@@ -432,7 +432,7 @@ function codegen_choice_gradients(trace_type::Type{T}, selection_type::Type,
432432
# assemble value_choices and gradient_choices
433433
value_choices = gensym("value_choices")
434434
gradient_choices = gensym("gradient_choices")
435-
push!(stmts, generate_value_choice_gradient(selected_choices, selected_calls,
435+
push!(stmts, generate_value_gradient_choices(selected_choices, selected_calls,
436436
value_choices, gradient_choices))
437437

438438
# gradients with respect to inputs

0 commit comments

Comments
 (0)