Skip to content

Commit 9bb1b8e

Browse files
authored
Add enzyme to benchmark tests
1 parent 7249158 commit 9bb1b8e

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

benchmarks/benchmarks.jl

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,13 +43,21 @@ chosen_combinations = [
4343
("Smorgasbord", smorgasbord_instance, :simple_dict, :forwarddiff, true),
4444
("Smorgasbord", smorgasbord_instance, :typed, :reversediff, true),
4545
("Smorgasbord", smorgasbord_instance, :typed, :mooncake, true),
46+
("Smorgasbord", smorgasbord_instance, :typed, :enzyme, true),
4647
("Loop univariate 1k", loop_univariate1k, :typed, :mooncake, true),
48+
("Loop univariate 1k", loop_univariate1k, :typed, :enzyme, true),
4749
("Multivariate 1k", multivariate1k, :typed, :mooncake, true),
50+
("Multivariate 1k", multivariate1k, :typed, :enzyme, true),
4851
("Loop univariate 10k", loop_univariate10k, :typed, :mooncake, true),
52+
("Loop univariate 10k", loop_univariate10k, :typed, :enzyme, true),
4953
("Multivariate 10k", multivariate10k, :typed, :mooncake, true),
54+
("Multivariate 10k", multivariate10k, :typed, :enzyme, true),
5055
("Dynamic", Models.dynamic(), :typed, :mooncake, true),
56+
("Dynamic", Models.dynamic(), :typed, :enzyme, true),
5157
("Submodel", Models.parent(randn(rng)), :typed, :mooncake, true),
58+
("Submodel", Models.parent(randn(rng)), :typed, :enzyme, true),
5259
("LDA", lda_instance, :typed, :reversediff, true),
60+
("LDA", lda_instance, :typed, :enzyme, true),
5361
]
5462

5563
# Time running a model-like function that does not use DynamicPPL, as a reference point.

0 commit comments

Comments
 (0)