diff --git a/Project.toml b/Project.toml index b648109..62d7a64 100644 --- a/Project.toml +++ b/Project.toml @@ -10,8 +10,15 @@ MKL_jll = "856f044c-d86e-5d09-b602-aeab76dc8ba7" Preferences = "21216c6a-2e73-6563-6e65-726566657250" Reexport = "189a3867-3050-52da-a836-e630ba90ab69" +[weakdeps] +Enzyme = "7da242da-08ed-463a-9acd-ee780be4f1d9" + +[extensions] +FFTWEnzymeExt = "Enzyme" + [compat] AbstractFFTs = "1.5" +Enzyme = "0.12, 0.13" FFTW_jll = "3.3.9" LinearAlgebra = "<0.0.1, 1" MKL_jll = "2019.0.117, 2020, 2021, 2022, 2023, 2024" diff --git a/ext/FFTWEnzymeExt.jl b/ext/FFTWEnzymeExt.jl new file mode 100644 index 0000000..f05cd96 --- /dev/null +++ b/ext/FFTWEnzymeExt.jl @@ -0,0 +1,8 @@ +module FFTWEnzymeExt + +using Enzyme, FFTW + +Enzyme.EnzymeRules.inactive_noinl(::typeof(FFTW.assert_applicable), x...) = true + +end # module +