Skip to content

Commit c822206

Browse files
committed
DArray: Small matmul bugfix
1 parent 9443ffb commit c822206

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

Diff for: src/array/mul.jl

+5-5
Original file line numberDiff line numberDiff line change
@@ -109,8 +109,8 @@ function gemm_dagger!(
109109
Bmt, Bnt = size(Bc)
110110
Cmt, Cnt = size(Cc)
111111

112-
alpha = _add.alpha
113-
beta = _add.beta
112+
alpha = T(_add.alpha)
113+
beta = T(_add.beta)
114114

115115
if Ant != Bmt
116116
throw(DimensionMismatch(lazy"A has number of blocks ($Amt,$Ant) but B has number of blocks ($Bmt,$Bnt)"))
@@ -212,8 +212,8 @@ function syrk_dagger!(
212212
Amt, Ant = size(Ac)
213213
Cmt, Cnt = size(Cc)
214214

215-
alpha = _add.alpha
216-
beta = _add.beta
215+
alpha = T(_add.alpha)
216+
beta = T(_add.beta)
217217

218218
uplo = 'U'
219219
if Ant != Cmt
@@ -233,7 +233,7 @@ function syrk_dagger!(
233233
Dagger.@spawn BLAS.herk!(
234234
uplo,
235235
trans,
236-
alpha,
236+
real(alpha),
237237
In(Ac[n, k]),
238238
mzone,
239239
InOut(Cc[n, n]),

0 commit comments

Comments
 (0)