Mercurial > repos > public > sbplib_julia

diff src/SbpOperators/stencil.jl @ 1452:4f79ab676ebc bugfix/sbp_operators/stencil_return_type

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fix broken tests using sum(f,...). Needs to be evaluated for performance
author Jonatan Werpers <jonatan@werpers.com>
date Fri, 24 Nov 2023 21:56:40 +0100
parents 14cb97284373
children 18cb103e499c
line wrap: on
line diff
--- a/src/SbpOperators/stencil.jl	Fri Nov 24 21:49:07 2023 +0100
+++ b/src/SbpOperators/stencil.jl	Fri Nov 24 21:56:40 2023 +0100
@@ -69,20 +69,15 @@
 end
 
 Base.@propagate_inbounds @inline function apply_stencil(s::Stencil, v::AbstractVector, i::Int)
-    w = zero(promote_type(eltype(s),eltype(v)))
-    @simd for k ∈ 1:length(s)
-        w += s.weights[k]*v[i + s.range[k]]
+    return sum(1:length(s)) do k
+        s.weights[k]*v[i + s.range[k]]
     end
-
-    return w
 end
 
 Base.@propagate_inbounds @inline function apply_stencil_backwards(s::Stencil, v::AbstractVector, i::Int)
-    w = zero(promote_type(eltype(s),eltype(v)))
-    @simd for k ∈ length(s):-1:1
-        w += s.weights[k]*v[i - s.range[k]]
+    return sum(length(s):-1:1) do k
+        s.weights[k]*v[i - s.range[k]]
     end
-    return w
 end
 
 function left_pad(s::Stencil, N)