Use FFTW.ESTIMATE in precompilation

This may slightly reduce package precompilation time.
jipolanco · Oct 5, 2024 · f3b5839 · f3b5839
1 parent 19e04d9
commit f3b5839
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/NonuniformFFTs.jl b/src/NonuniformFFTs.jl
@@ -432,6 +432,7 @@ end
     # Ts = [Float32, Float64, ComplexF32, ComplexF64]
     Ts = (Float64,)
     σ = 1.25
+    fftw_flags = FFTW.ESTIMATE
     for kernel ∈ kernels, m ∈ ms, T ∈ Ts, ndims ∈ ndims_all
         dims = ntuple(_ -> 13, ndims)  # floor(σN) = 16
         Np = 16  # number of non-uniform points
@@ -446,7 +447,7 @@ end
             qs = ntuple(_ -> randn(T, Np), ntransforms)
             uhat = ntuple(_ -> Array{C}(undef, Ns), ntransforms)
             @compile_workload begin
-                plan = PlanNUFFT(T, dims; ntransforms, m, σ, kernel)
+                plan = PlanNUFFT(T, dims; ntransforms, fftw_flags, m, σ, kernel)
                 set_points!(plan, xs)
                 exec_type1!(uhat, plan, qs)
                 exec_type2!(qs, plan, uhat)