Reenable GroupedStridedPointer to save strides

Author: chriselrod

Project.toml

@@ -1,7 +1,7 @@
 name = "LoopVectorization"
 uuid = "bdcacae8-1622-11e9-2a5c-532679323890"
 authors = ["Chris Elrod <[email protected]>"]
-version = "0.12.15"
+version = "0.12.16"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

src/LoopVectorization.jl

@@ -40,7 +40,7 @@ using SLEEFPirates: log_fast, log2_fast, log10_fast
 
 using ArrayInterface
 using ArrayInterface: OptionallyStaticUnitRange, OptionallyStaticRange, Zero, One, StaticBool, True, False, reduce_tup, indices, UpTri, LoTri
-using StrideArraysCore: CloseOpen
+using StrideArraysCore: CloseOpen, PtrArray
 # @static if VERSION ≥ v"1.6.0-rc1" #TODO: delete `else` when dropping 1.5 support
     # using ArrayInterface: static_step
 # else # Julia 1.5 did not define `step` on CartesianIndices

src/condense_loopset.jl

@@ -350,6 +350,11 @@ function check_shouldindbyind(ls::LoopSet, ind::Symbol, shouldindbyind::Vector{B
   end
   true
 end
+
+
+@inline dummy_ptrarray(sp, A) = sp
+@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))
+
 # write a "check_loops_safe_to_zerorangestart
 # that will be used to
 # 1) decide whether to zerorangestart
@@ -398,7 +403,7 @@ function add_grouped_strided_pointer!(extra_args::Expr, ls::LoopSet)
   for (k,gespindsummary) ∈ gespsummaries
     ref = allarrayrefs[k]
     gespinds = calcgespinds(ls, ref, gespindsummary, shouldindbyind)
-    push!(tgarrays.args, Expr(:call, lv(:gespf1), vptr(ref), gespinds))
+    push!(tgarrays.args, Expr(:call, lv(:dummy_ptrarray), Expr(:call, lv(:gespf1), vptr(ref), gespinds), name(ref)))
   end
   push!(gsp.args, tgarrays)
   matcheddims = Expr(:tuple)