Reduce precompilation times

(Precompile less things.)
jipolanco · Jun 12, 2024 · 9c7c3d1 · 9c7c3d1 · jipolanco · Jun 12, 2024
1 parent acccfdd
commit 9c7c3d1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "NonuniformFFTs"
 uuid = "cd96f58b-6017-4a02-bb9e-f4d81626177f"
 authors = ["Juan Ignacio Polanco <[email protected]>"]
-version = "0.3.13"
+version = "0.3.14"
 
 [deps]
 Bessels = "0e736298-9ec6-45e8-9647-e4fc86a2fe38"

diff --git a/src/NonuniformFFTs.jl b/src/NonuniformFFTs.jl
@@ -353,13 +353,14 @@ function copy_deconvolve_to_oversampled!(
     ûs_all
 end
 
-# Precompilation
+# Precompile a small subset of possible static parameter combinations (m, kernel, T, ndims,
+# ntransforms), to avoid huge precompilation times.
 @setup_workload let
     kernels = [default_kernel()]  # only precompile for default kernel
-    ndims_all = 1:3
-    ms = map(HalfSupport, 2:2:8)
+    ndims_all = (1, 2, 3)
+    ms = map(HalfSupport, (4, 8))
     # Ts = [Float32, Float64, ComplexF32, ComplexF64]
-    Ts = [Float64, ComplexF64]
+    Ts = (Float64,)
     σ = 1.25
     for kernel ∈ kernels, m ∈ ms, T ∈ Ts, ndims ∈ ndims_all
         dims = ntuple(_ -> 13, ndims)  # floor(σN) = 16
@@ -370,7 +371,7 @@ end
             (i == 1 && T <: Real) ? ((N + 2) >> 1) : N
         end
         C = complex(T)
-        for ntrans ∈ (1, ndims)  # number of simultaneous transforms
+        for ntrans ∈ (1,)  # number of simultaneous transforms
             ntransforms = Val(ntrans)
             qs = ntuple(_ -> randn(T, Np), ntransforms)
             uhat = ntuple(_ -> Array{C}(undef, Ns), ntransforms)