|
Packit |
67cb25 |
/* spdgemv.c
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* Copyright (C) 2012-2014 Patrick Alken
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* This program is free software; you can redistribute it and/or modify
|
|
Packit |
67cb25 |
* it under the terms of the GNU General Public License as published by
|
|
Packit |
67cb25 |
* the Free Software Foundation; either version 3 of the License, or (at
|
|
Packit |
67cb25 |
* your option) any later version.
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* This program is distributed in the hope that it will be useful, but
|
|
Packit |
67cb25 |
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit |
67cb25 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Packit |
67cb25 |
* General Public License for more details.
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* You should have received a copy of the GNU General Public License
|
|
Packit |
67cb25 |
* along with this program; if not, write to the Free Software
|
|
Packit |
67cb25 |
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
Packit |
67cb25 |
*/
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
#include <config.h>
|
|
Packit |
67cb25 |
#include <stdlib.h>
|
|
Packit |
67cb25 |
#include <math.h>
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
#include <gsl/gsl_math.h>
|
|
Packit |
67cb25 |
#include <gsl/gsl_vector.h>
|
|
Packit |
67cb25 |
#include <gsl/gsl_errno.h>
|
|
Packit |
67cb25 |
#include <gsl/gsl_spmatrix.h>
|
|
Packit |
67cb25 |
#include <gsl/gsl_spblas.h>
|
|
Packit |
67cb25 |
#include <gsl/gsl_blas.h>
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/*
|
|
Packit |
67cb25 |
gsl_spblas_dgemv()
|
|
Packit |
67cb25 |
Multiply a sparse matrix and a vector
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
Inputs: alpha - scalar factor
|
|
Packit |
67cb25 |
A - sparse matrix
|
|
Packit |
67cb25 |
x - dense vector
|
|
Packit |
67cb25 |
beta - scalar factor
|
|
Packit |
67cb25 |
y - (input/output) dense vector
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
Return: y = alpha*op(A)*x + beta*y
|
|
Packit |
67cb25 |
*/
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
int
|
|
Packit |
67cb25 |
gsl_spblas_dgemv(const CBLAS_TRANSPOSE_t TransA, const double alpha,
|
|
Packit |
67cb25 |
const gsl_spmatrix *A, const gsl_vector *x,
|
|
Packit |
67cb25 |
const double beta, gsl_vector *y)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
const size_t M = A->size1;
|
|
Packit |
67cb25 |
const size_t N = A->size2;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if ((TransA == CblasNoTrans && N != x->size) ||
|
|
Packit |
67cb25 |
(TransA == CblasTrans && M != x->size))
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
GSL_ERROR("invalid length of x vector", GSL_EBADLEN);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
else if ((TransA == CblasNoTrans && M != y->size) ||
|
|
Packit |
67cb25 |
(TransA == CblasTrans && N != y->size))
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
GSL_ERROR("invalid length of y vector", GSL_EBADLEN);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
else
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
size_t j, p;
|
|
Packit |
67cb25 |
size_t incX, incY;
|
|
Packit |
67cb25 |
size_t lenX, lenY;
|
|
Packit |
67cb25 |
double *X, *Y;
|
|
Packit |
67cb25 |
double *Ad;
|
|
Packit |
67cb25 |
size_t *Ap, *Ai, *Aj;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if (TransA == CblasNoTrans)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
lenX = N;
|
|
Packit |
67cb25 |
lenY = M;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
else
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
lenX = M;
|
|
Packit |
67cb25 |
lenY = N;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* form y := beta*y */
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
Y = y->data;
|
|
Packit |
67cb25 |
incY = y->stride;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if (beta == 0.0)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
size_t jy = 0;
|
|
Packit |
67cb25 |
for (j = 0; j < lenY; ++j)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
Y[jy] = 0.0;
|
|
Packit |
67cb25 |
jy += incY;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
else if (beta != 1.0)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
size_t jy = 0;
|
|
Packit |
67cb25 |
for (j = 0; j < lenY; ++j)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
Y[jy] *= beta;
|
|
Packit |
67cb25 |
jy += incY;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if (alpha == 0.0)
|
|
Packit |
67cb25 |
return GSL_SUCCESS;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* form y := alpha*op(A)*x + y */
|
|
Packit |
67cb25 |
Ap = A->p;
|
|
Packit |
67cb25 |
Ad = A->data;
|
|
Packit |
67cb25 |
X = x->data;
|
|
Packit |
67cb25 |
incX = x->stride;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if ((GSL_SPMATRIX_ISCCS(A) && (TransA == CblasNoTrans)) ||
|
|
Packit |
67cb25 |
(GSL_SPMATRIX_ISCRS(A) && (TransA == CblasTrans)))
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
Ai = A->i;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
for (j = 0; j < lenX; ++j)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
for (p = Ap[j]; p < Ap[j + 1]; ++p)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
Y[Ai[p] * incY] += alpha * Ad[p] * X[j * incX];
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
else if ((GSL_SPMATRIX_ISCCS(A) && (TransA == CblasTrans)) ||
|
|
Packit |
67cb25 |
(GSL_SPMATRIX_ISCRS(A) && (TransA == CblasNoTrans)))
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
Ai = A->i;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
for (j = 0; j < lenY; ++j)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
for (p = Ap[j]; p < Ap[j + 1]; ++p)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
Y[j * incY] += alpha * Ad[p] * X[Ai[p] * incX];
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
else if (GSL_SPMATRIX_ISTRIPLET(A))
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
if (TransA == CblasNoTrans)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
Ai = A->i;
|
|
Packit |
67cb25 |
Aj = A->p;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
else
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
Ai = A->p;
|
|
Packit |
67cb25 |
Aj = A->i;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
for (p = 0; p < A->nz; ++p)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
Y[Ai[p] * incY] += alpha * Ad[p] * X[Aj[p] * incX];
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
else
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
GSL_ERROR("unsupported matrix type", GSL_EINVAL);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
return GSL_SUCCESS;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
} /* gsl_spblas_dgemv() */
|