cassette_perf.jl

using Cassette

function rosenbrock(x)
    a = one(eltype(x))
    b = 100.0
    result = zero(a)
    for i in 1:length(x)-1
        result += (a - x[i])^2 + b*(x[i+1] - x[i]^2)^2
    end
    return result
end

Cassette.@context Ctx

f(x) = Cassette.@execute Ctx rosenbrock(x)

x = rand(1000)

# ideally, f(x) should have nearly the same performance as rosenbrock(x)