Skip to content

Commit 1cc02f5

Browse files
committed
add overwrite checking
1 parent e43d543 commit 1cc02f5

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

pygpu/blas.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ def gemmBatch_3d(double alpha, GpuArray A, GpuArray B,
204204
if beta != 0.0:
205205
raise ValueError("C not provided and beta != 0")
206206
C = pygpu_empty(3, Cshp, A.ga.typecode, GA_ANY_ORDER, A.context, None)
207-
else:
207+
elif not overwrite_c:
208208
C = pygpu_copy(C, GA_ANY_ORDER)
209209
pygpu_blas_rgemmBatch_3d(transA, transB, alpha, A, B, beta, C, 0)
210210

0 commit comments

Comments
 (0)