Skip to content

Commit 2a6ce58

Browse files
committed
Add comments
1 parent 9ad92bd commit 2a6ce58

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

stdlib/LinearAlgebra/src/matmul.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -935,12 +935,13 @@ Base.@constprop :aggressive generic_matmatmul!(C::AbstractVecOrMat, tA, tB, A::A
935935
if BxN != CxN
936936
throw(DimensionMismatch(lazy"matrix B has axes ($BxK,$BxN), matrix C has axes ($CxM,$CxN)"))
937937
end
938-
_add_alpha = MulAddMul{ais1,true,typeof(_add.alpha),Bool}(_add.alpha,false)
938+
_rmul_alpha = MulAddMul{ais1,true,typeof(_add.alpha),Bool}(_add.alpha,false)
939939
if isbitstype(R) && sizeof(R) 16 && !(A isa Adjoint || A isa Transpose)
940940
_rmul_or_fill!(C, _add.beta)
941941
(iszero(_add.alpha) || isempty(A) || isempty(B)) && return C
942942
@inbounds for n in BxN, k in BxK
943-
Balpha = _add_alpha(B[k,n])
943+
# Balpha = B[k,n] * alpha, but we skip the multiplication in case isone(alpha)
944+
Balpha = _rmul_alpha(B[k,n])
944945
@simd for m in AxM
945946
C[m,n] = muladd(A[m,k], Balpha, C[m,n])
946947
end

0 commit comments

Comments
 (0)