[Numpy-discussion] Dot + add operation

Guillaume Bethouart guillaume.bethouart at eshard.com
Tue Mar 30 20:34:19 EDT 2021

Is it possible to add a method to perform a dot product and add the result
to an existing matrix in a single operation ?

Like C = dot_add(A, B, C) equivalent to C += A @ B.This behavior is
natively proposed by the Blas *gemm primitive.

The goal is to reduce the peak memory consumption. Indeed, during the
computation of C += A @ B, the maximum allocated memory is twice the size
of C.Using *gemm to add directly the result , the maximum memory
consumption is less than 1.5x the size of C.
This difference is significant for large matrices.

Any people interested in it ?
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <https://mail.python.org/pipermail/numpy-discussion/attachments/20210331/6eba68e5/attachment.html>

More information about the NumPy-Discussion mailing list