Skip to content

Commit 24b63fd

Browse files
committed
perf: update cos testing
1 parent 2bf3aec commit 24b63fd

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

benchmark/benchmarks.jl

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,14 @@ SUITE["comptime"]["basics"] = BenchmarkGroup()
1111
SUITE["comptime"]["basics"]["2D sum"] = @benchmarkable Reactant.compile(sum, (a,)) setup = (
1212
a = Reactant.ConcreteRArray(ones(2, 10))
1313
)
14-
SUITE["comptime"]["basics"]["Basic cos"] = @benchmarkable Reactant.compile(cos, (a,)) setup = (
14+
15+
bcast_cos(x) = cos.(x)
16+
17+
SUITE["comptime"]["basics"]["cos.(x)"] = @benchmarkable begin
18+
Reactant.compile(bcast_cos, (a,))
19+
end setup = begin
1520
a = Reactant.ConcreteRArray(ones(2, 10))
16-
)
21+
end
1722

1823
SUITE["comptime"]["lux neural networks"] = BenchmarkGroup()
1924

0 commit comments

Comments
 (0)