Skip to content

Commit 49aa252

Browse files
committed
Use DensePointerWrapper
1 parent 500ae18 commit 49aa252

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/condense_loopset.jl

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -445,8 +445,11 @@ function check_shouldindbyind(ls::LoopSet, ind::Symbol, shouldindbyind::Vector{B
445445
end
446446

447447

448-
@inline dummy_ptrarray(sp, A) = sp
449-
@inline dummy_ptrarray(sp::AbstractStridedPointer{T,N}, A::AbstractArray{T,N}) where {T,N} = PtrArray(sp, VectorizationBase.zerotuple(Val{N}()), VectorizationBase.val_dense_dims(A))
448+
@inline densewrapper(sp, A) = sp
449+
# @inline dummy_ptrarray(sp::AbstractStridedPointer{T,N}, A::AbstractArray{T,N}) where {T,N} = PtrArray(sp, VectorizationBase.zerotuple(Val{N}()), VectorizationBase.val_dense_dims(A))
450+
@inline densewrapper(sp::AbstractStridedPointer{T,N}, A::AbstractArray{T,N}) where {T,N} = _densewrapper(sp, VectorizationBase.val_dense_dims(A))
451+
@inline _densewrapper(sp, ::Nothing) = sp
452+
@inline _densewrapper(sp::AbstractStridedPointer, ::Val{D}) where {D} = VectorizationBase.DensePointerWrapper{D}(sp)
450453

451454
# write a "check_loops_safe_to_zerorangestart
452455
# that will be used to
@@ -496,7 +499,7 @@ function add_grouped_strided_pointer!(extra_args::Expr, ls::LoopSet)
496499
for (k,gespindsummary) gespsummaries
497500
ref = allarrayrefs[k]
498501
gespinds = calcgespinds(ls, ref, gespindsummary, shouldindbyind)
499-
push!(tgarrays.args, Expr(:call, lv(:dummy_ptrarray), Expr(:call, lv(:gespf1), vptr(ref), gespinds), name(ref)))
502+
push!(tgarrays.args, Expr(:call, lv(:densewrapper), Expr(:call, lv(:gespf1), vptr(ref), gespinds), name(ref)))
500503
end
501504
push!(gsp.args, tgarrays)
502505
matcheddims = Expr(:tuple)

0 commit comments

Comments
 (0)