spmv_ubcsr.h

/*
 *  file :  spmv_ubcsr.h
 *  desc :  Row-unaligned block compressed sparse row format.
 */

#if !defined(INC_SPMV_UBCSR_H)
#define INC_SPMV_UBCSR_H

#if !defined(INC_SPMV_H)
typedef struct tagSparseMatrixCSR SparseMatrixCSR;
#endif

#if !defined(INC_SPMV_VBR_H)
typedef struct tagSparseMatrixVBR SparseMatrixVBR;
#endif

/*  
 *  Sparse matrix in row-unaligned blocked CSR storage.
 *  
 *  Note: Value blocks are stored in row-major order.
 *
 *  Data structure invariants:
 *     nnz == 0  ==>  b_row_ptr[0] = 0, all other pointer fields are NULL.
 */
typedef struct tagSparseMatrixUBCSR
{
  int m, n;         /* true dimensions */
  int nnz;          /* true # of non-zeros (not incl. explicit zeros) */
  int r, c;         /* register block size */
  int num_blocks;       /* number of stored blocks */

  int bm;           /* # of block rows */
  int *b_row_ptr;       /* block row starts; ceil(m/r)+1==bm+1 entries */
  int *b_row_ind;       /* block row indices; ceil(m/r)==bm entries */
  int *b_col_ind;       /* block column indices; num_blocks entries */
  double *b_val;        /* block values; nnz entries */

  int is_symm;          /* 1 <==> matrix is symmetric */
} SparseMatrixUBCSR;


typedef void (*SPMV_UBCSR_FP) (int bm, const int *ptr,
                   const int *row_ind, const int *col_ind,
                   const double *val, const double *x, double *y);

#define UBCSR_MAX_ROWS  10
#define UBCSR_MAX_COLS  10

extern SPMV_UBCSR_FP spmvUBlkCSRRoutines[];

extern void spmv_convert_vbr2ubcsr (const SparseMatrixVBR * A,
                    int r, int c, SparseMatrixUBCSR * B1,
                    SparseMatrixCSR * B2);
    /*
     *  spmv_convert_vbr2ubcsr( A, B1, B2 );
     *
     *  Compute the decomposition A = B1 + B2, where A is a given
     *  VBR matrix, B1 is a row-unaligned blocked matrix in r x c
     *  UBCSR format, and B2 is a CSR matrix.
     */

extern void spmv_alloc_ubcsr (SparseMatrixUBCSR * A, int r, int c, int m,
                  int n, int mb, int nblocks);
extern void spmv_destroy_ubcsr (SparseMatrixUBCSR * A);

extern void spmv_ubcsr (const SparseMatrixUBCSR * A,
            const double *x, double *y);

#endif

/* eof */

---