Skip to content

Commit

Permalink
formating
Browse files Browse the repository at this point in the history
  • Loading branch information
hofmannmartin authored Feb 19, 2021
1 parent 3065d65 commit 1d70e27
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/Kaczmarz.jl
Original file line number Diff line number Diff line change
Expand Up @@ -280,13 +280,13 @@ for (T,W, WS,shufflevectorMask,vσ) in [(Float32,:WF32,:WF32S,:shufflevectorMask
vptrA = stridedpointer(A)
vptrb = stridedpointer(b)
for _ = 1:Nrep
Base.Cartesian.@nexprs 4 i -> vb_i = vload(vptrb, ($W*(i-1) + idx,))
Base.Cartesian.@nexprs 4 i -> va_i = vload(vptrA, ($W*(i-1) + idx,k))
Base.Cartesian.@nexprs 4 i -> vb_i = vload(vptrb, ($W*(i-1) + idx,))
Base.Cartesian.@nexprs 4 i -> va_i = vload(vptrA, ($W*(i-1) + idx,k))
Base.Cartesian.@nexprs 4 i -> begin
vb_i = muladd(va_i, vβr, vb_i)
va_i = shufflevector(va_i, $shufflevectorMask)
vb_i = muladd(va_i, vβi, vb_i)
vstore!(vptrb, vb_i, ($W*(i-1) + idx,))
vstore!(vptrb, vb_i, ($W*(i-1) + idx,))
end
idx += iOffset
end
Expand All @@ -308,7 +308,7 @@ for (T,W, WS,shufflevectorMask,vσ) in [(Float32,:WF32,:WF32S,:shufflevectorMask
vb = muladd(va, vβr, vb)
va = shufflevector(va, $shufflevectorMask)
vb = muladd(va, vβi, vb)
vstore!(vptrb, vb, (idx,), vloadMask)
vstore!(vptrb, vb, (idx,), vloadMask)
end
end # GC.@preserve
end
Expand Down

0 comments on commit 1d70e27

Please sign in to comment.