diff --git a/src/Spaces/dss.jl b/src/Spaces/dss.jl index abb955dabe..b5df5db645 100644 --- a/src/Spaces/dss.jl +++ b/src/Spaces/dss.jl @@ -836,7 +836,7 @@ function dss_local_vertices!( sum_data = mapreduce( ⊞, vertex; - init = RecursiveApply.rmap(zero, slab(perimeter_data, 1, 1)[1]), + init = RecursiveApply.rzero(eltype(slab(perimeter_data, 1, 1))), ) do (lidx, vert) ip = Topologies.perimeter_vertex_node_index(vert) perimeter_slab = slab(perimeter_data, level, lidx) @@ -906,9 +906,8 @@ function dss_local_ghost!( sum_data = mapreduce( ⊞, vertex; - init = RecursiveApply.rmap( - zero, - slab(perimeter_data, 1, 1)[1], + init = RecursiveApply.rzero( + eltype(slab(perimeter_data, 1, 1)), ), ) do (isghost, idx, vert) ip = Topologies.perimeter_vertex_node_index(vert) diff --git a/test/Operators/spectralelement/benchmark_ops.jl b/test/Operators/spectralelement/benchmark_ops.jl index 2dd94ab424..f11469dd22 100644 --- a/test/Operators/spectralelement/benchmark_ops.jl +++ b/test/Operators/spectralelement/benchmark_ops.jl @@ -138,7 +138,7 @@ using JET p = @allocated kernel_complicated_field_dss!(kernel_args) @test p == 0 p = @allocated kernel_complicated_field2_dss!(kernel_args) - @test_broken p == 0 + @test p == 0 # Inference tests JET.@test_opt kernel_scalar_dss!(kernel_args) JET.@test_opt kernel_vector_dss!(kernel_args) @@ -146,6 +146,6 @@ using JET JET.@test_opt kernel_ntuple_field_dss!(kernel_args) JET.@test_opt kernel_ntuple_floats_dss!(kernel_args) JET.@test_opt kernel_complicated_field_dss!(kernel_args) - # JET.@test_opt kernel_complicated_field2_dss!(kernel_args) # fails + JET.@test_opt kernel_complicated_field2_dss!(kernel_args) end end