devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add BLAS function axpby.

Browse Source 
libeigen/eigen!2083
This commit is contained in:
Antonio Sánchez
2025-11-25 23:13:02 +00:00
committed by Rasmus Munk Larsen
parent a1eeb02204
commit 48eb5227c8
3 changed files with 52 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
blas/level1_impl.h
View File

@@ -22,11 +22,42 @@ EIGEN_BLAS_FUNC(axpy)
else if (*incx > 0 && *incy > 0)
make_vector(y, *n, *incy) += alpha * make_vector(x, *n, *incx);
else if (*incx > 0 && *incy < 0)
make_vector(y, *n, -*incy).reverse() += alpha * make_vector(x, *n, *incx);
make_vector(y, *n, -*incy) += alpha * make_vector(x, *n, *incx).reverse();
else if (*incx < 0 && *incy > 0)
make_vector(y, *n, *incy) += alpha * make_vector(x, *n, -*incx).reverse();
else if (*incx < 0 && *incy < 0)
make_vector(y, *n, -*incy).reverse() += alpha * make_vector(x, *n, -*incx).reverse();
make_vector(y, *n, -*incy) += alpha * make_vector(x, *n, -*incx);
}
EIGEN_BLAS_FUNC(axpby)
(const int *pn, const RealScalar *palpha, const RealScalar *px, const int *pincx, const RealScalar *pbeta,
RealScalar *py, const int *pincy) {
const Scalar *x = reinterpret_cast<const Scalar *>(px);
Scalar *y = reinterpret_cast<Scalar *>(py);
const Scalar alpha = *reinterpret_cast<const Scalar *>(palpha);
const Scalar beta = *reinterpret_cast<const Scalar *>(pbeta);
const int n = *pn;
if (n <= 0) return;
if (Eigen::numext::equal_strict(beta, Scalar(1))) {
EIGEN_BLAS_FUNC_NAME(axpy)(pn, palpha, px, pincx, py, pincy);
return;
}
const int incx = *pincx;
const int incy = *pincy;
if (incx == 1 && incy == 1)
make_vector(y, n) = alpha * make_vector(x, n) + beta * make_vector(y, n);
else if (incx > 0 && incy > 0)
make_vector(y, n, incy) = alpha * make_vector(x, n, incx) + beta * make_vector(y, n, incy);
else if (incx > 0 && incy < 0)
make_vector(y, n, -incy) = alpha * make_vector(x, n, incx).reverse() + beta * make_vector(y, n, -incy);
else if (incx < 0 && incy > 0)
make_vector(y, n, incy) = alpha * make_vector(x, n, -incx).reverse() + beta * make_vector(y, n, incy);
else if (incx < 0 && incy < 0)
make_vector(y, n, -incy) = alpha * make_vector(x, n, -incx) + beta * make_vector(y, n, -incy);
}
EIGEN_BLAS_FUNC(copy)(int *n, RealScalar *px, int *incx, RealScalar *py, int *incy) {