translate literal variables

TuringLang · Oct 22, 2024 · b18752b · b18752b
1 parent 7d73c27
commit b18752b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/tapedfunction.jl b/src/tapedfunction.jl
@@ -368,6 +368,14 @@ function translate!!(var::IRVar, line::Core.SlotNumber,
     return Instruction(func, input, output)
 end
 
+function translate!!(var::IRVar, line::Number, # literal vars
+                     bindings::Bindings, isconst::Bool, ir)
+    func = identity
+    input = (bind_var!(line, bindings, ir),)
+    output =  bind_var!(var, bindings, ir)
+    return Instruction(func, input, output)
+end
+
 function translate!!(var::IRVar, line::NTuple{N, Symbol},
                      bindings::Bindings, isconst::Bool, ir) where {N}
     # for syntax (; x, y, z), see Turing.jl#1873