Skip to content

Commit 4839a8c

Browse files
committed
Move @stable_muladdmul within Bidiagonal-Diagonal _mul!
1 parent b8a9330 commit 4839a8c

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

src/bidiag.jl

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -890,18 +890,21 @@ function __bibimul_bulk!(C, A::Bidiagonal, B::Bidiagonal, _add)
890890
C
891891
end
892892

893-
function _mul!(C::AbstractMatrix, A::BiTriSym, B::Diagonal, _add::MulAddMul)
893+
function _mul!(C::AbstractMatrix, A::BiTriSym, B::Diagonal, alpha::Number, beta::Number)
894894
require_one_based_indexing(C)
895895
matmul_size_check(size(C), size(A), size(B))
896896
n = size(A,1)
897897
iszero(n) && return C
898-
_rmul_or_fill!(C, _add.beta) # see the same use above
899-
iszero(_add.alpha) && return C
898+
_rmul_or_fill!(C, beta) # see the same use above
899+
iszero(alpha) && return C
900900
# beta is unused in the _bidimul! call, so we set it to false
901-
_add_nonzeroalpha = _MulAddMul_nonzeroalpha(_add, Val(false))
902-
_bidimul!(C, A, B, _add_nonzeroalpha)
901+
@stable_muladdmul _mul_nonzeroalpha!(C, A, B, MulAddMul(alpha, false))
903902
C
904903
end
904+
function _mul_nonzeroalpha!(C::AbstractMatrix, A::BiTriSym, B::Diagonal, _add::MulAddMul)
905+
_add_nonzeroalpha = _MulAddMul_nonzeroalpha(_add)
906+
_bidimul!(C, A, B, _add_nonzeroalpha)
907+
end
905908
function _bidimul!(C::AbstractMatrix, A::BiTriSym, B::Diagonal, _add::MulAddMul)
906909
n = size(A,1)
907910
Al = _diag(A, -1)

0 commit comments

Comments
 (0)