Mercurial > repos > public > sbplib_julia

comparison src/SbpOperators/stencil.jl @ 1454:c10c6c3e9247 bugfix/sbp_operators/stencil_return_type

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fix broken test using Base.promote_op. Needs to be evaluated for performance
author Jonatan Werpers <jonatan@werpers.com>
date Fri, 24 Nov 2023 22:58:37 +0100
parents 14cb97284373
children 6e9ac7925f20
comparison
equal deleted inserted replaced
1446:d744b01be520 1454:c10c6c3e9247
67 end 67 end
68 return s.weights[1 + i - s.range[1]] 68 return s.weights[1 + i - s.range[1]]
69 end 69 end
70 70
71 Base.@propagate_inbounds @inline function apply_stencil(s::Stencil, v::AbstractVector, i::Int) 71 Base.@propagate_inbounds @inline function apply_stencil(s::Stencil, v::AbstractVector, i::Int)
72 w = zero(promote_type(eltype(s),eltype(v))) 72 T = Base.promote_op(*, eltype(s), eltype(v))
73 w = zero(T)
73 @simd for k ∈ 1:length(s) 74 @simd for k ∈ 1:length(s)
74 w += s.weights[k]*v[i + s.range[k]] 75 w += s.weights[k]*v[i + s.range[k]]
75 end 76 end
76 77
77 return w 78 return w
78 end 79 end
79 80
80 Base.@propagate_inbounds @inline function apply_stencil_backwards(s::Stencil, v::AbstractVector, i::Int) 81 Base.@propagate_inbounds @inline function apply_stencil_backwards(s::Stencil, v::AbstractVector, i::Int)
81 w = zero(promote_type(eltype(s),eltype(v))) 82 T = Base.promote_op(*, eltype(s), eltype(v))
83 w = zero(T)
82 @simd for k ∈ length(s):-1:1 84 @simd for k ∈ length(s):-1:1
83 w += s.weights[k]*v[i - s.range[k]] 85 w += s.weights[k]*v[i - s.range[k]]
84 end 86 end
85 return w 87 return w
86 end 88 end