fix tests on LTS version

Author: hakkelt

.github/workflows/tests.yml

@@ -30,8 +30,27 @@ jobs:
           version: ${{ matrix.julia-version }}
           arch: ${{ matrix.julia-arch }}
       - uses: julia-actions/cache@v2
+      - name: Prepare LTS test environment
+        if: matrix.julia-version == 'lts'
+        shell: bash
+        run: |
+          julia --project=test -e '
+            using Pkg
+            Pkg.develop(path = pwd())
+            for pkg in ("DSPOperators", "FFTWOperators", "NFFTOperators", "WaveletOperators")
+                Pkg.develop(path = joinpath(pwd(), pkg))
+            end
+            Pkg.instantiate()
+            Pkg.build()
+            Pkg.precompile()
+          '
       - uses: julia-actions/julia-buildpkg@v1
+        if: matrix.julia-version != 'lts'
+      - name: Run tests on LTS
+        if: matrix.julia-version == 'lts'
+        run: julia --project=test --check-bounds=yes --compiled-modules=yes --depwarn=yes --code-coverage=user test/runtests.jl
       - uses: julia-actions/julia-runtest@v1
+        if: matrix.julia-version != 'lts'
         with:
           coverage: true
       - uses: julia-actions/julia-processcoverage@v1

src/utils.jl

@@ -8,12 +8,29 @@ const thread_count_functions = Ref{Vector{Pair{Function, Function}}}(
     ]
 )
 
+# Non-inlined helpers so that the abstract Function dispatch is contained in
+# AbstractOperators and not inlined into the calling module (which would cause
+# JET @test_opt findings when target_modules excludes AbstractOperators).
+@noinline function _save_thread_counts()
+    return [pair.first() for pair in thread_count_functions[]]
+end
+
+@noinline function _apply_thread_counts(n::Int)
+    for pair in thread_count_functions[]
+        pair.second(n)
+    end
+end
+
+@noinline function _restore_thread_counts(prev::Vector)
+    for (i, pair) in enumerate(thread_count_functions[])
+        pair.second(prev[i])
+    end
+end
+
 function set_thread_counts_expr(thread_count_expr, body_expr)
     return quote
-        local prev_thread_counts = [pair.first() for pair in AbstractOperators.thread_count_functions[]]
-        for pair in AbstractOperators.thread_count_functions[]
-            pair.second($thread_count_expr)
-        end
+        local prev_thread_counts = AbstractOperators._save_thread_counts()
+        AbstractOperators._apply_thread_counts($thread_count_expr)
         local res
         try
             if $thread_count_expr == 1
@@ -26,9 +43,7 @@ function set_thread_counts_expr(thread_count_expr, body_expr)
             end
         finally
             # Restore previous thread counts
-            for (i, pair) in enumerate(AbstractOperators.thread_count_functions[])
-                pair.second(prev_thread_counts[i])
-            end
+            AbstractOperators._restore_thread_counts(prev_thread_counts)
         end
         res
     end

test/Project.toml

@@ -29,7 +29,7 @@ Wavelets = "29a6e085-ba6d-5f35-a997-948ac2efa89a"
 Aqua = "0.8"
 BenchmarkTools = "1"
 Documenter = "1.8"
-JET = "0.10"
+JET = "0.8, 0.9, 0.10, 0.11"
 JLArrays = "0.2.0"
 LinearAlgebra = "1"
 LinearMaps = "3.11.4"