Print more type info in JET tests

Author: charleskawczynski

test/MatrixFields/matrix_field_test_utils.jl

@@ -28,7 +28,7 @@ macro test_all(expression)
         local test_func() = $(esc(expression))
         @test test_func()                   # correctness
         @test (@allocated test_func()) == 0 # allocations
-        @test_opt test_func()               # type instabilities
+        @test_opt stacktrace_types_limit=4 test_func()               # type instabilities
     end
 end
 
@@ -117,14 +117,14 @@ function test_field_broadcast(;
         # Test get_result and set_result! for type instabilities, and test
         # set_result! for allocations. Ignore the type instabilities in CUDA and
         # the allocations they incur.
-        @test_opt ignored_modules = cuda_frames materialize(get_result)
-        @test_opt ignored_modules = cuda_frames materialize!(result, set_result)
+        @test_opt stacktrace_types_limit=4 ignored_modules = cuda_frames materialize(get_result)
+        @test_opt stacktrace_types_limit=4 ignored_modules = cuda_frames materialize!(result, set_result)
         using_cuda || @test (@allocated materialize!(result, set_result)) == 0
 
         if !isnothing(ref_set_result)
             # Test ref_set_result! for type instabilities and allocations to
             # ensure that the performance comparison is fair.
-            @test_opt ignored_modules = cuda_frames materialize!(
+            @test_opt stacktrace_types_limit=4  ignored_modules = cuda_frames materialize!(
                 ref_result,
                 ref_set_result,
             )

test/Operators/integrals.jl

@@ -48,7 +48,7 @@ function test_column_integral_definite!(center_space)
     @test max_relative_error <= 0.006 # Less than 0.6% error.
 
     cuda = (AnyFrameModule(CUDA),)
-    @test_opt ignored_modules = cuda column_integral_definite!(∫u_test, ᶜu)
+    @test_opt stacktrace_types_limit=4 ignored_modules = cuda column_integral_definite!(∫u_test, ᶜu)
 
     test_allocs(@allocated column_integral_definite!(∫u_test, ᶜu))
 end
@@ -70,7 +70,7 @@ function test_column_integral_indefinite!(center_space)
     @test max_relative_error <= 0.006 # Less than 0.6% error at the top level.
 
     cuda = (AnyFrameModule(CUDA),)
-    @test_opt ignored_modules = cuda column_integral_indefinite!(ᶠ∫u_test, ᶜu)
+    @test_opt stacktrace_types_limit=4 ignored_modules = cuda column_integral_indefinite!(ᶠ∫u_test, ᶜu)
 
     test_allocs(@allocated column_integral_indefinite!(ᶠ∫u_test, ᶜu))
 end
@@ -93,7 +93,7 @@ function test_column_integral_indefinite_fn!(center_space)
         @test max_relative_error <= 0.006 # Less than 0.6% error at the top level.
 
         cuda = (AnyFrameModule(CUDA),)
-        @test_opt ignored_modules = cuda column_integral_indefinite!(
+        @test_opt stacktrace_types_limit=4 ignored_modules = cuda column_integral_indefinite!(
             fn,
             ᶠ∫u_test,
         )
@@ -132,7 +132,7 @@ function test_column_reduce_and_accumulate!(center_space)
         set_output! = () -> column_reduce!(f, output, input; init, transform)
         set_output!()
         @test output == reference_output
-        @test_opt ignored_modules = (AnyFrameModule(CUDA),) set_output!()
+        @test_opt stacktrace_types_limit=4 ignored_modules = (AnyFrameModule(CUDA),) set_output!()
         test_allocs(@allocated set_output!())
     end
 
@@ -148,7 +148,7 @@ function test_column_reduce_and_accumulate!(center_space)
             () -> column_accumulate!(f, output, input; init, transform)
         set_output!()
         @test output == reference_output
-        @test_opt ignored_modules = (AnyFrameModule(CUDA),) set_output!()
+        @test_opt stacktrace_types_limit=4 ignored_modules = (AnyFrameModule(CUDA),) set_output!()
         test_allocs(@allocated set_output!())
     end
 end