Skip to content

Commit b9d02fa

Browse files
committed
Extend mul! to include alfa and beta
1 parent 486a534 commit b9d02fa

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/tensors/diagonal.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -207,9 +207,10 @@ end
207207

208208
function LinearAlgebra.mul!(dC::DiagonalTensorMap,
209209
dA::DiagonalTensorMap,
210-
dB::DiagonalTensorMap)
210+
dB::DiagonalTensorMap,
211+
α::Number, β::Number)
211212
dC.domain == dA.domain == dB.domain || throw(SpaceMismatch())
212-
dC.data .= dA.data .* dB.data
213+
mul!(Diagonal(dC.data), Diagonal(dA.data), Diagonal(dB.data), α, β)
213214
return dC
214215
end
215216

0 commit comments

Comments
 (0)