Skip to content

Commit 069a216

Browse files
committed
Tweak
1 parent 97b67b0 commit 069a216

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lib/mpsgraphs/matmul.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,14 +31,14 @@ function _matmul!(c::MtlArray{Tc}, a::MtlArray{Tab}, b::MtlArray{Tab}, alpha::Nu
3131

3232
matmul = matrixMultiplicationWithPrimaryTensor(graph, transB, transA)
3333

34-
afteralpha = if alpha == 1
34+
afteralpha = if isone(alpha)
3535
matmul
3636
else
3737
alphatensor = constantWithScalar(graph, alpha, Tc)
3838
multiplicationWithPrimaryTensor(graph, alphatensor, matmul)
3939
end
4040

41-
afterbeta = if beta == 0
41+
afterbeta = if iszero(beta)
4242
afteralpha
4343
else
4444
placeC = placeholderTensor(graph, size(c), Tc)

0 commit comments

Comments
 (0)