Fix loopinfo expression

Author: willtebbutt

src/copyable_task.jl

@@ -704,6 +704,8 @@ function derive_copyable_task_ir(ir::BBCode)::Tuple{BBCode,Tuple,Vector{Any}}
                     push!(inst_pairs, (ID(), new_inst(set_expr)))
                 elseif stmt isa IDPhiNode
                     # do nothing -- we've already handled any `PhiNode`s.
+                elseif Meta.isexpr(stmt, :loopinfo)
+                    push!(inst_pairs, (id, inst))
                 else
                     throw(error("Unhandled stmt $stmt of type $(typeof(stmt))"))
                 end

src/test_utils.jl

@@ -202,6 +202,14 @@ function test_cases()
             [1, 2],
             allocs,
         ),
+        Testcase(
+            "rosenbrock",
+            nothing,
+            (rosenbrock, rand(100_000), nothing),
+            nothing,
+            [],
+            none,
+        ),
     ]
 end
 
@@ -354,4 +362,13 @@ function final_statement_produce()
     return produce(2)
 end
 
+# Produces a `:loopinfo` expression.
+function rosenbrock(x, callback=nothing)
+    i = x[2:end]
+    j = x[1:(end - 1)]
+    ret = sum((1 .- j) .^ 2 + 100 * (i - j .^ 2) .^ 2)
+    callback !== nothing && callback(ret)
+    return ret
+end
+
 end