Skip to content

Commit ac78d25

Browse files
committed
Mark assert applicable as non-differentiable
1 parent f888022 commit ac78d25

File tree

2 files changed

+15
-0
lines changed

2 files changed

+15
-0
lines changed

Project.toml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,15 @@ MKL_jll = "856f044c-d86e-5d09-b602-aeab76dc8ba7"
1010
Preferences = "21216c6a-2e73-6563-6e65-726566657250"
1111
Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
1212

13+
[weakdeps]
14+
Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9"
15+
16+
[extensions]
17+
FFTWEnzymeExt = "Enzyme"
18+
1319
[compat]
1420
AbstractFFTs = "1.5"
21+
Enzyme = "0.12, 0.13"
1522
FFTW_jll = "3.3.9"
1623
LinearAlgebra = "<0.0.1, 1"
1724
MKL_jll = "2019.0.117, 2020, 2021, 2022, 2023, 2024"

ext/FFTWEnzymeExt.jl

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
module FFTWEnzymeExt
2+
3+
using Enzyme, FFTW
4+
5+
Enzyme.EnzymeRules.inactive_noinl(typeof(assert_applicable), x...) = true
6+
7+
end # module
8+

0 commit comments

Comments
 (0)