dvbrvml.c

用于求解大型稀疏线性方程组Ax=b的数值计算库.

/*-------------------------------------------------------|
|  NIST SPARSE BLAS v. 0.9 (Sat Jul 6 14:27:21 EDT 1996) |
|                                                        |
|  Authors:                                              |
|     Karin A. Remington and Roldan Pozo                 |
|     National Institute of Standards and Technology     |
|                                                        |
|  Based on the interface standard proposed in:          | 
|   "A Revised Proposal for a Sparse BLAS Toolkit" by    |
|    S. Carney and K. Wu -- University of Minnesota      |
|    M. Heroux and G. Li -- Cray Research                |  
|    R. Pozo and K.A. Remington -- NIST                  |
|                                                        |
|  Contact:                                              |
|     Karin A. Remington, email: [e-mail]     |
--------------------------------------------------------*/


/* Created:  Sat Jul 6 14:38:12 EDT 1996 */

#include "dvbrvml.h"



void VBR_VecMult_CAB_double(
                 const int mb,  const int kb, 
                 const double *val, const int *indx,
                 const int *bindx, const int *rpntr,
                 const int *cpntr, const int *bpntrb, const int *bpntre, 
                 const double *b,   
                 double *c, 
                 const int ind_base)
{
  double t;
  const double *pb;
  double *pc=c;
  const double *pval;
  int i,j,jb,je,block;
  int cs,mm,bs,br;
  int index;
  int ii,jj;
  int m=rpntr[mb]-rpntr[0];
  int k=cpntr[kb]-cpntr[0];

  val-=ind_base;
  b-=ind_base;
  indx-=ind_base;
  bindx-=ind_base;
  cpntr-=ind_base;

    for (i=0;i!=m;i++) *pc++ = 0;                  

  c-=ind_base;
 
  for (i=0;i!=mb;i++) {      
    jb = bpntrb[i];             
    je = bpntre[i];             
    cs = rpntr[i];              
    mm = rpntr[i+1]-rpntr[i];   
    for (j=jb;j!=je;j++) {      
      index = bindx[j];
      bs=cpntr[index];             
      br=cpntr[index+1]-bs;        
      pval = &val[indx[j]];        
                                   
      pb = &b[bs];                 
      pc = &c[rpntr[i]];           
       for (jj=0;jj!=br;jj++) {       
         if( pb[jj] != 0.0 ) {           
           t =  pb[jj];            
           for (ii=0;ii!=mm;ii++) {      
             pc[ii] += t* (*pval++);        
           }
         } else {
           pval+=mm;                     
         }
       }
    }
  }
}

void VBR_VecMult_CATB_double(
                 const int mb,  const int kb, 
                 const double *val, const int *indx,
                 const int *bindx, const int *rpntr,
                 const int *cpntr, const int *bpntrb, const int *bpntre, 
                 const double *b,   
                 double *c, 
                 const int ind_base)
{
  double t;
  const double *pb;
  double *pc=c;
  const double *pval;
  int i,j,jb,je,block;
  int cs,mm,bs,cr;
  int index;
  int ii,jj;
  int m=rpntr[mb]-rpntr[0];
  int k=cpntr[kb]-cpntr[0];

  val-=ind_base;
  b-=ind_base;
  indx-=ind_base;
  bindx-=ind_base;
  cpntr-=ind_base;

    for (i=0;i!=k;i++) *pc++ = 0;                  

  c-=ind_base;
 
  for (i=0;i!=mb;i++) {      
    jb = bpntrb[i];             
    je = bpntre[i];             
    bs = rpntr[i];              
    mm = rpntr[i+1]-rpntr[i];   
    for (j=jb;j!=je;j++) {      
      index = bindx[j];
      cs=cpntr[index];             
      cr=cpntr[index+1]-cs;        
      pval = &val[indx[j]];        
                                   
      pb = &b[bs];                 
      pc = &c[cs];                 
       for (jj=0;jj!=cr;jj++) {       
           t = 0;
           for (ii=0;ii!=mm;ii++) {      
             t +=  pb[ii] * (*pval++);
           }
           pc[jj] += t;                  
       }
    }
  }
}

void VBRsymm_VecMult_CAB_double(
                 const int mb,  const int kb, 
                 const double *val, const int *indx,
                 const int *bindx, const int *rpntr,
                 const int *cpntr, const int *bpntrb, const int *bpntre, 
                 const double *b,   
                 double *c, 
                 const int ind_base)
{
  double t;
  const double *pb;
  double *pc=c;
  const double *pval;
  const double *ptmp; 
  int i,j,jb,je,block;
  int cs,css,cr,mm,bs,br;
  int index;
  int ii,jj;
  int m=rpntr[mb]-rpntr[0];
  int k=cpntr[kb]-cpntr[0];

  val-=ind_base;
  b-=ind_base;
  indx-=ind_base;
  bindx-=ind_base;
  cpntr-=ind_base;

    for (i=0;i!=m;i++) *pc++ = 0;                  

  c-=ind_base;
 
  for (i=0;i!=mb;i++) {      
    jb = bpntrb[i];             
    je = bpntre[i];             
    cs = rpntr[i];              
    mm = rpntr[i+1]-rpntr[i];   
    for (j=jb;j!=je;j++) {      
      index = bindx[j];
      bs=cpntr[index];             
      br=cpntr[index+1]-bs;        
      pval = &val[indx[j]];        
      ptmp = pval;                 
                                   
      pb = &b[bs];                 
      pc = &c[rpntr[i]];           
       for (jj=0;jj!=br;jj++) {       
         if( pb[jj] != 0.0 ) {           
           t =  pb[jj];            
           for (ii=0;ii!=mm;ii++) {      
             pc[ii] += t* (*pval++);        
           }
         } else {
           pval+=mm;                     
         }
       }
 
      if ( index == i + ind_base ) {
        continue;                  
      }
 
      cr= br;                      
      pval = ptmp;                 
                                   
      pb = &b[cs];                 
      pc = &c[bs];                 
       for (jj=0;jj!=cr;jj++) {       
           t = 0;
           for (ii=0;ii!=mm;ii++) {      
             t +=  pb[ii] * (*pval++);
           }
           pc[jj] += t;                  
       }
    }
  }
}

void VBRskew_VecMult_CAB_double(
                 const int mb,  const int kb, 
                 const double *val, const int *indx,
                 const int *bindx, const int *rpntr,
                 const int *cpntr, const int *bpntrb, const int *bpntre, 
                 const double *b,   
                 double *c, 
                 const int ind_base)
{
  double t;
  const double *pb;
  double *pc=c;
  const double *pval;
  const double *ptmp;
  int i,j,jb,je,block;
  int cs,css,cr,mm,bs,br;
  int index;
  int ii,jj;
  int m=rpntr[mb]-rpntr[0];
  int k=cpntr[kb]-cpntr[0];

  val-=ind_base;
  b-=ind_base;
  indx-=ind_base;
  bindx-=ind_base;
  cpntr-=ind_base;

    for (i=0;i!=m;i++) *pc++ = 0;                  

  c-=ind_base;
 
  for (i=0;i!=mb;i++) {      
    jb = bpntrb[i];             
    je = bpntre[i];             
    cs = rpntr[i];              
    mm = rpntr[i+1]-rpntr[i];   
    for (j=jb;j!=je;j++) {      
      index = bindx[j];
 
      if ( index == i + ind_base ) {
        continue;                  
      }
      bs=cpntr[index];             
      br=cpntr[index+1]-bs;        
      pval = &val[indx[j]];        
      ptmp = pval;                 
                                   
      pb = &b[bs];                 
      pc = &c[rpntr[i]];           
       for (jj=0;jj!=br;jj++) {       
         if( pb[jj] != 0.0 ) {           
           t =  pb[jj];            
           for (ii=0;ii!=mm;ii++) {      
             pc[ii] += t* (*pval++);        
           }
         } else {
           pval+=mm;                     
         }
       }
 
      cr= br;                      
      pval = ptmp;                 
                                   
      pb = &b[cs];                 
      pc = &c[bs];                 
       for (jj=0;jj!=cr;jj++) {       
           t = 0;
           for (ii=0;ii!=mm;ii++) {      
             t -=  pb[ii] * (*pval++);
           }
           pc[jj] += t;                  
       }
    }
  }
}

void VBRskew_VecMult_CATB_double(
                 const int mb,  const int kb, 
                 const double *val, const int *indx,
                 const int *bindx, const int *rpntr,
                 const int *cpntr, const int *bpntrb, const int *bpntre, 
                 const double *b,   
                 double *c, 
                 const int ind_base)
{
  double t;
  const double *pb;
  double *pc=c;
  const double *pval;
  int i,j,jb,je,block;
  int cs,css,cr,mm,bs,br;
  int index;
  int ii,jj;
  int m=rpntr[mb]-rpntr[0];
  int k=cpntr[kb]-cpntr[0];
  const double *ptmp;

  val-=ind_base;
  b-=ind_base;
  indx-=ind_base;
  bindx-=ind_base;
  cpntr-=ind_base;

    for (i=0;i!=k;i++) *pc++ = 0;                  

  c-=ind_base;
 
  for (i=0;i!=mb;i++) {      
    jb = bpntrb[i];             
    je = bpntre[i];             
    cs = rpntr[i];              
    mm = rpntr[i+1]-rpntr[i];   
    for (j=jb;j!=je;j++) {      
      index = bindx[j];
 
      if ( index == i + ind_base ) {
        continue;                  
      }
      bs=cpntr[index];             
      br=cpntr[index+1]-bs;        
      pval = &val[indx[j]];        
      ptmp = pval;                 
                                   
      pb = &b[bs];                 
      pc = &c[rpntr[i]];           
       for (jj=0;jj!=br;jj++) {       
         if( pb[jj] != 0.0 ) {           
           t =  pb[jj];            
           for (ii=0;ii!=mm;ii++) {      
             pc[ii] -= t* (*pval++);        
           }
         } else {
           pval+=mm;                     
         }
       }
 
      cr= br;                      
      pval = ptmp;                 
                                   
      pb = &b[cs];                 
      pc = &c[bs];                 
       for (jj=0;jj!=cr;jj++) {       
           t = 0;
           for (ii=0;ii!=mm;ii++) {      
             t +=  pb[ii] * (*pval++);
           }
           pc[jj] += t;                  
       }
    }
  }
}




void VBR_VecMult_CaAB_double(
                 const int mb,  const int kb, const double alpha,
                 const double *val, const int *indx,
                 const int *bindx, const int *rpntr,
                 const int *cpntr, const int *bpntrb, const int *bpntre, 
                 const double *b,   
                 double *c, 
                 const int ind_base)
{
  double t;
  const double *pb;
  double *pc=c;
  const double *pval;
  int i,j,jb,je,block;
  int cs,mm,bs,br;
  int index;
  int ii,jj;
  int m=rpntr[mb]-rpntr[0];
  int k=cpntr[kb]-cpntr[0];

  val-=ind_base;
  b-=ind_base;
  indx-=ind_base;
  bindx-=ind_base;
  cpntr-=ind_base;

    for (i=0;i!=m;i++) *pc++ = 0;                  

  c-=ind_base;
 
  for (i=0;i!=mb;i++) {      
    jb = bpntrb[i];             
    je = bpntre[i];             
    cs = rpntr[i];              
    mm = rpntr[i+1]-rpntr[i];   
    for (j=jb;j!=je;j++) {      
      index = bindx[j];
      bs=cpntr[index];             
      br=cpntr[index+1]-bs;        
      pval = &val[indx[j]];        
                                   
      pb = &b[bs];                 
      pc = &c[rpntr[i]];           
       for (jj=0;jj!=br;jj++) {       
         if( pb[jj] != 0.0 ) {           
           t = alpha * pb[jj];            
           for (ii=0;ii!=mm;ii++) {      
             pc[ii] += t* (*pval++);        
           }
         } else {
           pval+=mm;                     
         }
       }
    }
  }
}

void VBR_VecMult_CaATB_double(
                 const int mb,  const int kb, const double alpha,
                 const double *val, const int *indx,
                 const int *bindx, const int *rpntr,
                 const int *cpntr, const int *bpntrb, const int *bpntre, 
                 const double *b,   
                 double *c, 
                 const int ind_base)
{
  double t;
  const double *pb;
  double *pc=c;
  const double *pval;
  int i,j,jb,je,block;
  int cs,mm,bs,cr;
  int index;
  int ii,jj;
  int m=rpntr[mb]-rpntr[0];
  int k=cpntr[kb]-cpntr[0];

  val-=ind_base;
  b-=ind_base;
  indx-=ind_base;
  bindx-=ind_base;
  cpntr-=ind_base;

    for (i=0;i!=k;i++) *pc++ = 0;                  

  c-=ind_base;
 
  for (i=0;i!=mb;i++) {      
    jb = bpntrb[i];             
    je = bpntre[i];             
    bs = rpntr[i];              
    mm = rpntr[i+1]-rpntr[i];   
    for (j=jb;j!=je;j++) {      
      index = bindx[j];
      cs=cpntr[index];             
      cr=cpntr[index+1]-cs;        
      pval = &val[indx[j]];        
                                   
      pb = &b[bs];                 
      pc = &c[cs];                 
       for (jj=0;jj!=cr;jj++) {       
           t = 0;
           for (ii=0;ii!=mm;ii++) {      
             t += alpha * pb[ii] * (*pval++);
           }
           pc[jj] += t;                  
       }
    }
  }
}

void VBRsymm_VecMult_CaAB_double(
                 const int mb,  const int kb, const double alpha,
                 const double *val, const int *indx,
                 const int *bindx, const int *rpntr,
                 const int *cpntr, const int *bpntrb, const int *bpntre, 
                 const double *b,   
                 double *c, 
                 const int ind_base)
{
  double t;
  const double *pb;
  double *pc=c;
  const double *pval;
  const double *ptmp; 
  int i,j,jb,je,block;
  int cs,css,cr,mm,bs,br;
  int index;
  int ii,jj;
  int m=rpntr[mb]-rpntr[0];
  int k=cpntr[kb]-cpntr[0];

  val-=ind_base;
  b-=ind_base;
  indx-=ind_base;
  bindx-=ind_base;
  cpntr-=ind_base;

    for (i=0;i!=m;i++) *pc++ = 0;                  

  c-=ind_base;
 
  for (i=0;i!=mb;i++) {      
    jb = bpntrb[i];             
    je = bpntre[i];             
    cs = rpntr[i];              
    mm = rpntr[i+1]-rpntr[i];   
    for (j=jb;j!=je;j++) {      
      index = bindx[j];
      bs=cpntr[index];