fixup! activate test_allocations for GPU tests

Author: vchuravy

test/test_cuda_2d.jl

@@ -21,6 +21,7 @@ isdir(outdir) && rm(outdir, recursive = true)
                         linf=6.627000273229378e-5)
     # Ensure that we do not have excessive memory allocations
     # (e.g., from type instabilities)
+    semi = ode.p # `semidiscretize` adapts the semi, so we need to obtain it from the ODE problem.
     @test_allocations(Trixi.rhs!, semi, sol, 1000)
     @test real(ode.p.solver) == Float64
     @test real(ode.p.solver.basis) == Float64
@@ -49,6 +50,7 @@ end
                         storage_type=CuArray)
     # Ensure that we do not have excessive memory allocations
     # (e.g., from type instabilities)
+    semi = ode.p # `semidiscretize` adapts the semi, so we need to obtain it from the ODE problem.
     @test_allocations(Trixi.rhs!, semi, sol, 1000)
     @test real(ode.p.solver) == Float32
     @test real(ode.p.solver.basis) == Float32

test/test_cuda_3d.jl

@@ -21,6 +21,7 @@ isdir(outdir) && rm(outdir, recursive = true)
                         linf=[0.0014537194925779984])
     # Ensure that we do not have excessive memory allocations
     # (e.g., from type instabilities)
+    semi = ode.p # `semidiscretize` adapts the semi, so we need to obtain it from the ODE problem.
     @test_allocations(Trixi.rhs!, semi, sol, 1000)
     @test real(ode.p.solver) == Float64
     @test real(ode.p.solver.basis) == Float64
@@ -49,6 +50,7 @@ end
                         storage_type=CuArray)
     # Ensure that we do not have excessive memory allocations
     # (e.g., from type instabilities)
+    semi = ode.p # `semidiscretize` adapts the semi, so we need to obtain it from the ODE problem.
     @test_allocations(Trixi.rhs!, semi, sol, 1000)
     @test real(ode.p.solver) == Float32
     @test real(ode.p.solver.basis) == Float32

test/test_kernelabstractions.jl

@@ -27,7 +27,8 @@ end
                         linf=6.627000273229378e-5)
     # Ensure that we do not have excessive memory allocations
     # (e.g., from type instabilities)
-    @test_allocations(Trixi.rhs!, semi, sol, 75_000)
+    semi = ode.p # `semidiscretize` adapts the semi, so we need to obtain it from the ODE problem.
+    @test_allocations(Trixi.rhs!, ode.p, sol, 1000)
 end
 
 @trixi_testset "elixir_advection_basic_gpu.jl Float32" begin
@@ -40,7 +41,8 @@ end
                         real_type=Float32)
     # Ensure that we do not have excessive memory allocations
     # (e.g., from type instabilities)
-    @test_allocations(Trixi.rhs!, semi, sol, 70_000_000)
+    semi = ode.p # `semidiscretize` adapts the semi, so we need to obtain it from the ODE problem.
+    @test_allocations(Trixi.rhs!, ode.p, sol, 1000)
 end
 end
 
@@ -55,6 +57,7 @@ end
                         linf=[0.0014537194925779984])
     # Ensure that we do not have excessive memory allocations
     # (e.g., from type instabilities)
+    semi = ode.p # `semidiscretize` adapts the semi, so we need to obtain it from the ODE problem.
     @test_allocations(Trixi.rhs!, semi, sol, 1000)
 end
 
@@ -68,6 +71,7 @@ end
                         real_type=Float32)
     # Ensure that we do not have excessive memory allocations
     # (e.g., from type instabilities)
+    semi = ode.p # `semidiscretize` adapts the semi, so we need to obtain it from the ODE problem.
     @test_allocations(Trixi.rhs!, semi, sol, 1000)
 end
 end