Skip to content

Commit 3baf6e7

Browse files
authored
rowsupport in MatMulVecAdd materialize (#353)
1 parent 6e2a3a1 commit 3baf6e7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/generic/matmul.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ end
9999
checkdimensions(M)
100100
α,A,B,β,C = M.α,M.A,M.B,M.β,M.C
101101
_fill_lmul!(β, C)
102-
@inbounds for j = 1:size(A,2), k = colrange(A,j)
102+
@inbounds for j = rowsupport(A), k = colrange(A,j)
103103
C[k] += α*inbands_getindex(A,k,j)*B[j]
104104
end
105105
C
@@ -111,7 +111,7 @@ end
111111
A = transpose(At)
112112
_fill_lmul!(β, C)
113113

114-
@inbounds for j = 1:size(A,2), k = colrange(A,j)
114+
@inbounds for j = rowsupport(A), k = colrange(A,j)
115115
C[j] += α*transpose(inbands_getindex(A,k,j))*B[k]
116116
end
117117
C
@@ -122,7 +122,7 @@ end
122122
α,Ac,B,β,C = M.α,M.A,M.B,M.β,M.C
123123
A = Ac'
124124
_fill_lmul!(β, C)
125-
@inbounds for j = 1:size(A,2), k = colrange(A,j)
125+
@inbounds for j = rowsupport(A), k = colrange(A,j)
126126
C[j] += α*inbands_getindex(A,k,j)'*B[k]
127127
end
128128
C

0 commit comments

Comments
 (0)