after merging with Niklas

Author: pevnak

docs/src/lecture_09/irtools.jl

@@ -27,13 +27,11 @@ profile_fun(f::Core.Builtin, args...) = f(args...)
 
 @dynamo function profile_fun(f, args...)
     ir = IRTools.Inner.IR(f, args...)
-    for b in IRTools.blocks(ir)
-        for (v, ex) in b
-            if timable(ex.expr)
-                fname = exportname(ex.expr)
-                insert!(b, v, xcall(Main, :record_start, fname))
-                insertafter!(b, v, xcall(Main, :record_end, fname))
-            end
+    for (v, ex) in ir
+        if timable(ex.expr)
+            fname = exportname(ex.expr)
+            insert!(ir, v, xcall(Main, :record_start, fname))
+            insertafter!(ir, v, xcall(Main, :record_end, fname))
         end
     end
     for (x, st) in ir
@@ -44,6 +42,13 @@ profile_fun(f::Core.Builtin, args...) = f(args...)
     return ir
 end
 
+function foo(x, y)
+  z =  x * y
+  z + sin(y)
+end
+
 reset!(to)
 profile_fun(foo, 1.0, 1.0)
-to
+to
+
+@record foo(1.0, 1.0) => profile_fun(foo, 1.0, 1.0)

docs/src/lecture_09/lecture.md

@@ -22,6 +22,7 @@ end
 ```
 now observe the output
 ```julia
+julia> genplus(1.0, 1.0) == 1.0 + 1.0
 generating genplus(x, y)
 (x, y, typeof(x), typeof(y)) = (Float64, Float64, DataType, DataType)
 executing generated genplus(x, y)