Page 1 of 1

axpy_batched

Posted: Wed Sep 18, 2019 8:40 am
by genestoltz
Am I correct to assume that axpy_batched has not yet been implemented?

What would be the preferred way to execute a batched operation for a scalar times a vector (Don't care about the addition)?

Thanks.

Re: axpy_batched

Posted: Wed Sep 18, 2019 12:01 pm
by mgates3
So you really want scal_batched:

y_i = alpha * y_i for i = 1, ..., batch_size

Is that right?

Are your vectors scattered in memory, or packed into a single array? If in a single array, would a matrix scale function meet your needs, i.e.,

A = alpha * A

for an m-by-batch_size matrix A.

-mark