vsum invalid index

[jariji]

Am I doing something wrong here?

using LoopVectorization
let n = 10000, k = 3, A = rand(n, k)
    sum(1:n) do i
        A[i,1]
    end # ok
    vsum(1:n) do i
        A[i,1]
    end
    #= 
ERROR: ArgumentError: invalid index: 4 x VectorizationBase.Vec{4, Int64}
VectorizationBase.Vec{4, Int64}<1, 2, 3, 4>
VectorizationBase.Vec{4, Int64}<5, 6, 7, 8>
VectorizationBase.Vec{4, Int64}<9, 10, 11, 12>
VectorizationBase.Vec{4, Int64}<13, 14, 15, 16> of type VectorizationBase.VecUnroll{3, 4, Int64, VectorizationBase.Vec{4, Int64}}
     =#
end

Julia 1.10.3
LoopVectorization v0.12.171

[chriselrod]

I don't know if this is applicable to your actual use case, but for your given example, you could:

julia> let n = 10000, k = 3, A = rand(n, k)
           sum(1:n) do i
               A[i,1]
           end ≈ vsum(@view(A[:,1]))
       end
true

Indexing isn't supported in vsum's function.