Skip to content

Commit 4a74662

Browse files
authored
add some benchmarks for remake_zero! (#2608)
1 parent a11d667 commit 4a74662

File tree

1 file changed

+20
-2
lines changed

1 file changed

+20
-2
lines changed

benchmark/benchmarks.jl

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# To run:
22
# using PkgBenchmark, Enzyme
3-
# result = benchmarkpkg(KernelAbstractions)
3+
# result = benchmarkpkg(Enzyme)
44
# export_markdown("benchmark/perf.md", result)
55

66
# Note: if you change this file you will need to delete an regenerate tune.json
@@ -13,4 +13,22 @@ const SUITE = BenchmarkGroup()
1313

1414
SUITE["basics"] = BenchmarkGroup()
1515

16-
SUITE["basics"]["overhead"] = @benchmarkable Enzyme.autodiff(Forward, identity, Const(1.0))
16+
SUITE["basics"]["overhead"] = @benchmarkable Enzyme.autodiff(Forward, identity, Const(1.0))
17+
18+
SUITE["basics"]["make_zero"] = BenchmarkGroup()
19+
SUITE["basics"]["remake_zero!"] = BenchmarkGroup()
20+
21+
p = (; x = 1.0, y = zeros(3))
22+
23+
SUITE["basics"]["make_zero"]["namedtuple"] = @benchmarkable Enzyme.make_zero($p)
24+
SUITE["basics"]["remake_zero!"]["namedtuple"] = @benchmarkable Enzyme.remake_zero!(dp) setup = (dp = Enzyme.make_zero(p))
25+
26+
struct MyStruct
27+
x::Float64
28+
y::Vector{Float64}
29+
end
30+
31+
x = MyStruct(1.0, zeros(3))
32+
33+
SUITE["basics"]["make_zero"]["struct"] = @benchmarkable Enzyme.make_zero($x)
34+
SUITE["basics"]["remake_zero!"]["struct"] = @benchmarkable Enzyme.remake_zero!(dx) setup = (dx = Enzyme.make_zero(x))

0 commit comments

Comments
 (0)