fix tests

JuliaGPU · Nov 7, 2023 · cf04fa2 · cf04fa2
1 parent 402187d
commit cf04fa2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/layout.jl b/src/layout.jl
@@ -44,7 +44,7 @@ end
 end
 
 @inline @generated function vstorea!(::Type{Vec{N, T}}, ptr::Core.LLVMPtr{T, AS},
-                                     x::NTuple{M,<:Any}) where {N, T, AS, M}
+                                     x::NTuple{M,<:Any}, i::Integer = 1) where {N, T, AS, M}
     alignment = sizeof(T) * N
 
     ex = quote end
@@ -56,7 +56,7 @@ end
             y = @ntuple $N j -> VecElement{T}(x[j+$offset].value)
             vec_ptr = Base.bitcast(Core.LLVMPtr{NTuple{N, VecElement{T}}, AS}, ptr)
             @boundscheck checkalignment(vec_ptr)
-            unsafe_store!(vec_ptr, y, $offset ÷ N + 1, Val($alignment))
+            unsafe_store!(vec_ptr, y, $offset ÷ N + (i - 1) ÷ N + 1, Val($alignment))
         end).args)
     end