/[escript]/trunk/paso/src/ILU.c
ViewVC logotype

Diff of /trunk/paso/src/ILU.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

trunk/paso/src/Solver_ILU.c revision 1628 by phornby, Fri Jul 11 13:12:46 2008 UTC trunk/paso/src/ILU.c revision 3120 by gross, Mon Aug 30 10:48:00 2010 UTC
# Line 1  Line 1 
1    
 /* $Id$ */  
   
2  /*******************************************************  /*******************************************************
3   *  *
4   *           Copyright 2003-2007 by ACceSS MNRF  * Copyright (c) 2003-2010 by University of Queensland
5   *       Copyright 2007 by University of Queensland  * Earth Systems Science Computational Center (ESSCC)
6   *  * http://www.uq.edu.au/esscc
7   *                http://esscc.uq.edu.au  *
8   *        Primary Business: Queensland, Australia  * Primary Business: Queensland, Australia
9   *  Licensed under the Open Software License version 3.0  * Licensed under the Open Software License version 3.0
10   *     http://www.opensource.org/licenses/osl-3.0.php  * http://www.opensource.org/licenses/osl-3.0.php
11   *  *
12   *******************************************************/  *******************************************************/
13    
14    
15  /**************************************************************/  /**************************************************************/
16    
# Line 20  Line 19 
19  /**************************************************************/  /**************************************************************/
20    
21  /* Copyrights by ACcESS Australia 2003,2004,2005              */  /* Copyrights by ACcESS Australia 2003,2004,2005              */
22  /* Author: gross@access.edu.au                                */  /* Author: Lutz Gross, l.gross@uq.edu.au                                */
23    
24  /**************************************************************/  /**************************************************************/
25    
26  #include "Paso.h"  #include "Paso.h"
27  #include "Solver.h"  #include "Preconditioner.h"
28  #include "PasoUtil.h"  #include "PasoUtil.h"
29    
30  /**************************************************************/  /**************************************************************/
# Line 34  Line 33 
33    
34  void Paso_Solver_ILU_free(Paso_Solver_ILU * in) {  void Paso_Solver_ILU_free(Paso_Solver_ILU * in) {
35       if (in!=NULL) {       if (in!=NULL) {
         MEMFREE(in->colorOf);  
36          MEMFREE(in->factors);          MEMFREE(in->factors);
         MEMFREE(in->main_iptr);    
         Paso_Pattern_free(in->pattern);    
37          MEMFREE(in);          MEMFREE(in);
38       }       }
39  }  }
# Line 48  void Paso_Solver_ILU_free(Paso_Solver_IL Line 44  void Paso_Solver_ILU_free(Paso_Solver_IL
44    
45  */  */
46  Paso_Solver_ILU* Paso_Solver_getILU(Paso_SparseMatrix * A,bool_t verbose) {  Paso_Solver_ILU* Paso_Solver_getILU(Paso_SparseMatrix * A,bool_t verbose) {
47    dim_t n=A->numRows;    const dim_t n=A->numRows;
48    dim_t n_block=A->row_block_size;    const dim_t n_block=A->row_block_size;
49    index_t num_colors=0, *mis_marker=NULL;    const index_t* colorOf = Paso_Pattern_borrowColoringPointer(A->pattern);
50      const dim_t num_colors = Paso_Pattern_getNumColors(A->pattern);
51      const index_t *ptr_main = Paso_SparseMatrix_borrowMainDiagonalPointer(A);
52    register double A11,A12,A13,A21,A22,A23,A31,A32,A33,D;    register double A11,A12,A13,A21,A22,A23,A31,A32,A33,D;
53    register double S11,S12,S13,S21,S22,S23,S31,S32,S33;    register double S11,S12,S13,S21,S22,S23,S31,S32,S33;
54    register index_t i,iptr_main,iptr,iptr_ik,k,iptr_kj,j,iptr_ij,color,color2;    register index_t i,iptr_main,iptr_ik,k,iptr_kj,j,iptr_ij,color,color2, iptr;
55    double time0,time_color,time_fac;    double time0=0,time_fac=0;
56    /* allocations: */      /* allocations: */  
57    Paso_Solver_ILU* out=MEMALLOC(1,Paso_Solver_ILU);    Paso_Solver_ILU* out=MEMALLOC(1,Paso_Solver_ILU);
58    if (Paso_checkPtr(out)) return NULL;    if (Paso_checkPtr(out)) return NULL;
   out->colorOf=MEMALLOC(n,index_t);  
59    out->factors=MEMALLOC(A->len,double);    out->factors=MEMALLOC(A->len,double);
60    out->main_iptr=MEMALLOC(n,index_t);    
61    out->pattern=Paso_Pattern_getReference(A->pattern);    if ( ! Paso_checkPtr(out->factors)  ) {
   out->n_block=n_block;  
   out->n=n;  
   if ( !(Paso_checkPtr(out->colorOf) || Paso_checkPtr(out->main_iptr) || Paso_checkPtr(out->factors)) ) {  
     time0=Paso_timer();  
     Paso_Pattern_color(A->pattern,&out->num_colors,out->colorOf);  
     time_color=Paso_timer()-time0;  
62    
     if (Paso_noError()) {  
63         time0=Paso_timer();         time0=Paso_timer();
64         /* find main diagonal and copy matrix values */  
65         #pragma omp parallel for schedule(static) private(i,iptr,iptr_main,k)         #pragma omp parallel for schedule(static) private(i,iptr,k)
66         for (i = 0; i < n; ++i) {         for (i = 0; i < n; ++i) {
67             for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; ++iptr) {                 for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; iptr++) {
68                 iptr_main=A->pattern->ptr[0]-1;                       for (k=0;k<n_block*n_block;++k) out->factors[n_block*n_block*iptr+k]=A->val[n_block*n_block*iptr+k];
                 for (iptr=A->pattern->ptr[i];iptr<A->pattern->ptr[i+1]; iptr++) {  
                    if (A->pattern->index[iptr]==i) iptr_main=iptr;  
                    for (k=0;k<n_block*n_block;++k) out->factors[n_block*n_block*iptr+k]=A->val[n_block*n_block*iptr+k];  
69                 }                 }
70                 out->main_iptr[i]=iptr_main;         }    
                if (iptr_main==A->pattern->ptr[0]-1)  
                   Paso_setError(VALUE_ERROR, "Paso_Solver_getILU: no main diagonal");  
            }  
        }  
71         /* start factorization */         /* start factorization */
72             for (color=0;color<num_colors && Paso_noError();++color) {
        #pragma omp barrier  
        for (color=0;color<out->num_colors && Paso_noError();++color) {  
73                if (n_block==1) {                if (n_block==1) {
74                   #pragma omp parallel for schedule(static) private(i,color2,iptr_ik,k,iptr_kj,S11,j,iptr_ij,A11,iptr_main,D)                   #pragma omp parallel for schedule(static) private(i,color2,iptr_ik,k,iptr_kj,S11,j,iptr_ij,A11,iptr_main,D)
75                   for (i = 0; i < n; ++i) {                   for (i = 0; i < n; ++i) {
76                      if (out->colorOf[i]==color) {                      if (colorOf[i]==color) {
77                         for (color2=0;color2<color;++color2) {                         for (color2=0;color2<color;++color2) {
78                            for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {                            for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {
79                               k=A->pattern->index[iptr_ik];                                                         k=A->pattern->index[iptr_ik];                          
80                               if (out->colorOf[k]==color2) {                               if (colorOf[k]==color2) {
81                                  A11=out->factors[iptr_ik];                                  A11=out->factors[iptr_ik];
82                                  /* a_ij=a_ij-a_ik*a_kj */                                  /* a_ij=a_ij-a_ik*a_kj */
83                                  for (iptr_kj=A->pattern->ptr[k];iptr_kj<A->pattern->ptr[k+1]; iptr_kj++) {                                  for (iptr_kj=A->pattern->ptr[k];iptr_kj<A->pattern->ptr[k+1]; iptr_kj++) {
84                                     j=A->pattern->index[iptr_kj];                                     j=A->pattern->index[iptr_kj];
85                                     if (out->colorOf[j]>color2) {                                     if (colorOf[j]>color2) {
86                                        S11=out->factors[iptr_kj];                                        S11=out->factors[iptr_kj];
87                                        for (iptr_ij=A->pattern->ptr[i];iptr_ij<A->pattern->ptr[i+1]; iptr_ij++) {                                        for (iptr_ij=A->pattern->ptr[i];iptr_ij<A->pattern->ptr[i+1]; iptr_ij++) {
88                                           if (j==A->pattern->index[iptr_ij]) {                                           if (j==A->pattern->index[iptr_ij]) {
# Line 114  Paso_Solver_ILU* Paso_Solver_getILU(Paso Line 95  Paso_Solver_ILU* Paso_Solver_getILU(Paso
95                               }                               }
96                            }                            }
97                         }                         }
98                         iptr_main=out->main_iptr[i];                         iptr_main=ptr_main[i];
99                         D=out->factors[iptr_main];                         D=out->factors[iptr_main];
100                         if (ABS(D)>0.) {                         if (ABS(D)>0.) {
101                            D=1./D;                            D=1./D;
# Line 122  Paso_Solver_ILU* Paso_Solver_getILU(Paso Line 103  Paso_Solver_ILU* Paso_Solver_getILU(Paso
103                            /* a_ik=a_ii^{-1}*a_ik */                            /* a_ik=a_ii^{-1}*a_ik */
104                            for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {                            for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {
105                               k=A->pattern->index[iptr_ik];                               k=A->pattern->index[iptr_ik];
106                               if (out->colorOf[k]>color) {                               if (colorOf[k]>color) {
107                                  A11=out->factors[iptr_ik];                                  A11=out->factors[iptr_ik];
108                                  out->factors[iptr_ik]=A11*D;                                  out->factors[iptr_ik]=A11*D;
109                               }                                                             }                              
# Line 135  Paso_Solver_ILU* Paso_Solver_getILU(Paso Line 116  Paso_Solver_ILU* Paso_Solver_getILU(Paso
116                } else if (n_block==2) {                } else if (n_block==2) {
117                   #pragma omp parallel for schedule(static) private(i,color2,iptr_ik,k,iptr_kj,S11,S21,S12,S22,j,iptr_ij,A11,A21,A12,A22,iptr_main,D)                   #pragma omp parallel for schedule(static) private(i,color2,iptr_ik,k,iptr_kj,S11,S21,S12,S22,j,iptr_ij,A11,A21,A12,A22,iptr_main,D)
118                   for (i = 0; i < n; ++i) {                   for (i = 0; i < n; ++i) {
119                      if (out->colorOf[i]==color) {                      if (colorOf[i]==color) {
120                         for (color2=0;color2<color;++color2) {                         for (color2=0;color2<color;++color2) {
121                            for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {                            for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {
122                               k=A->pattern->index[iptr_ik];                                                         k=A->pattern->index[iptr_ik];                          
123                               if (out->colorOf[k]==color2) {                               if (colorOf[k]==color2) {
124                                  A11=out->factors[iptr_ik*4  ];                                  A11=out->factors[iptr_ik*4  ];
125                                  A21=out->factors[iptr_ik*4+1];                                  A21=out->factors[iptr_ik*4+1];
126                                  A12=out->factors[iptr_ik*4+2];                                  A12=out->factors[iptr_ik*4+2];
# Line 147  Paso_Solver_ILU* Paso_Solver_getILU(Paso Line 128  Paso_Solver_ILU* Paso_Solver_getILU(Paso
128                                  /* a_ij=a_ij-a_ik*a_kj */                                  /* a_ij=a_ij-a_ik*a_kj */
129                                  for (iptr_kj=A->pattern->ptr[k];iptr_kj<A->pattern->ptr[k+1]; iptr_kj++) {                                  for (iptr_kj=A->pattern->ptr[k];iptr_kj<A->pattern->ptr[k+1]; iptr_kj++) {
130                                     j=A->pattern->index[iptr_kj];                                     j=A->pattern->index[iptr_kj];
131                                     if (out->colorOf[j]>color2) {                                     if (colorOf[j]>color2) {
132                                        S11=out->factors[iptr_kj*4];                                        S11=out->factors[iptr_kj*4];
133                                        S21=out->factors[iptr_kj*4+1];                                        S21=out->factors[iptr_kj*4+1];
134                                        S12=out->factors[iptr_kj*4+2];                                        S12=out->factors[iptr_kj*4+2];
# Line 166  Paso_Solver_ILU* Paso_Solver_getILU(Paso Line 147  Paso_Solver_ILU* Paso_Solver_getILU(Paso
147                               }                               }
148                            }                            }
149                         }                         }
150                         iptr_main=out->main_iptr[i];                         iptr_main=ptr_main[i];
151                         A11=out->factors[iptr_main*4];                         A11=out->factors[iptr_main*4];
152                         A21=out->factors[iptr_main*4+1];                         A21=out->factors[iptr_main*4+1];
153                         A12=out->factors[iptr_main*4+2];                         A12=out->factors[iptr_main*4+2];
# Line 185  Paso_Solver_ILU* Paso_Solver_getILU(Paso Line 166  Paso_Solver_ILU* Paso_Solver_getILU(Paso
166                            /* a_ik=a_ii^{-1}*a_ik */                            /* a_ik=a_ii^{-1}*a_ik */
167                            for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {                            for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {
168                               k=A->pattern->index[iptr_ik];                               k=A->pattern->index[iptr_ik];
169                               if (out->colorOf[k]>color) {                               if (colorOf[k]>color) {
170                                  A11=out->factors[iptr_ik*4  ];                                  A11=out->factors[iptr_ik*4  ];
171                                  A21=out->factors[iptr_ik*4+1];                                  A21=out->factors[iptr_ik*4+1];
172                                  A12=out->factors[iptr_ik*4+2];                                  A12=out->factors[iptr_ik*4+2];
# Line 204  Paso_Solver_ILU* Paso_Solver_getILU(Paso Line 185  Paso_Solver_ILU* Paso_Solver_getILU(Paso
185                } else if (n_block==3) {                } else if (n_block==3) {
186                   #pragma omp parallel for schedule(static) private(i,color2,iptr_ik,k,iptr_kj,S11,S21,S31,S12,S22,S32,S13,S23,S33,j,iptr_ij,A11,A21,A31,A12,A22,A32,A13,A23,A33,iptr_main,D)                   #pragma omp parallel for schedule(static) private(i,color2,iptr_ik,k,iptr_kj,S11,S21,S31,S12,S22,S32,S13,S23,S33,j,iptr_ij,A11,A21,A31,A12,A22,A32,A13,A23,A33,iptr_main,D)
187                   for (i = 0; i < n; ++i) {                   for (i = 0; i < n; ++i) {
188                      if (out->colorOf[i]==color) {                      if (colorOf[i]==color) {
189                         for (color2=0;color2<color;++color2) {                         for (color2=0;color2<color;++color2) {
190                            for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {                            for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {
191                               k=A->pattern->index[iptr_ik];                                                         k=A->pattern->index[iptr_ik];                          
192                               if (out->colorOf[k]==color2) {                               if (colorOf[k]==color2) {
193                                  A11=out->factors[iptr_ik*9  ];                                  A11=out->factors[iptr_ik*9  ];
194                                  A21=out->factors[iptr_ik*9+1];                                  A21=out->factors[iptr_ik*9+1];
195                                  A31=out->factors[iptr_ik*9+2];                                  A31=out->factors[iptr_ik*9+2];
# Line 221  Paso_Solver_ILU* Paso_Solver_getILU(Paso Line 202  Paso_Solver_ILU* Paso_Solver_getILU(Paso
202                                  /* a_ij=a_ij-a_ik*a_kj */                                  /* a_ij=a_ij-a_ik*a_kj */
203                                  for (iptr_kj=A->pattern->ptr[k];iptr_kj<A->pattern->ptr[k+1]; iptr_kj++) {                                  for (iptr_kj=A->pattern->ptr[k];iptr_kj<A->pattern->ptr[k+1]; iptr_kj++) {
204                                     j=A->pattern->index[iptr_kj];                                     j=A->pattern->index[iptr_kj];
205                                     if (out->colorOf[j]>color2) {                                     if (colorOf[j]>color2) {
206                                        S11=out->factors[iptr_kj*9  ];                                        S11=out->factors[iptr_kj*9  ];
207                                        S21=out->factors[iptr_kj*9+1];                                        S21=out->factors[iptr_kj*9+1];
208                                        S31=out->factors[iptr_kj*9+2];                                        S31=out->factors[iptr_kj*9+2];
# Line 250  Paso_Solver_ILU* Paso_Solver_getILU(Paso Line 231  Paso_Solver_ILU* Paso_Solver_getILU(Paso
231                               }                               }
232                            }                            }
233                         }                         }
234                         iptr_main=out->main_iptr[i];                         iptr_main=ptr_main[i];
235                         A11=out->factors[iptr_main*9  ];                         A11=out->factors[iptr_main*9  ];
236                         A21=out->factors[iptr_main*9+1];                         A21=out->factors[iptr_main*9+1];
237                         A31=out->factors[iptr_main*9+2];                         A31=out->factors[iptr_main*9+2];
# Line 286  Paso_Solver_ILU* Paso_Solver_getILU(Paso Line 267  Paso_Solver_ILU* Paso_Solver_getILU(Paso
267                            /* a_ik=a_ii^{-1}*a_ik */                            /* a_ik=a_ii^{-1}*a_ik */
268                            for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {                            for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {
269                               k=A->pattern->index[iptr_ik];                               k=A->pattern->index[iptr_ik];
270                               if (out->colorOf[k]>color) {                               if (colorOf[k]>color) {
271                                  A11=out->factors[iptr_ik*9  ];                                  A11=out->factors[iptr_ik*9  ];
272                                  A21=out->factors[iptr_ik*9+1];                                  A21=out->factors[iptr_ik*9+1];
273                                  A31=out->factors[iptr_ik*9+2];                                  A31=out->factors[iptr_ik*9+2];
# Line 318  Paso_Solver_ILU* Paso_Solver_getILU(Paso Line 299  Paso_Solver_ILU* Paso_Solver_getILU(Paso
299                #pragma omp barrier                #pragma omp barrier
300         }         }
301         time_fac=Paso_timer()-time0;         time_fac=Paso_timer()-time0;
      }  
302    }    }
303    if (Paso_noError()) {    if (Paso_noError()) {
304        if (verbose) {        if (verbose) printf("timing: ILU: coloring/elemination : %e sec\n",time_fac);
          printf("ILU: %d color used \n",out->num_colors);  
          printf("timing: ILU: coloring/elemination : %e/%e\n",time_color,time_fac);  
      }  
305       return out;       return out;
306    } else  {    } else  {
307       Paso_Solver_ILU_free(out);       Paso_Solver_ILU_free(out);
# Line 343  Paso_Solver_ILU* Paso_Solver_getILU(Paso Line 320  Paso_Solver_ILU* Paso_Solver_getILU(Paso
320    
321  */  */
322    
323  void Paso_Solver_solveILU(Paso_Solver_ILU * ilu, double * x, double * b) {  void Paso_Solver_solveILU(Paso_SparseMatrix * A, Paso_Solver_ILU * ilu, double * x, const double * b) {
324       register dim_t i,k;       register dim_t i,k;
325       register index_t color,iptr_ik,iptr_main;       register index_t color,iptr_ik,iptr_main;
326       register double S1,S2,S3,R1,R2,R3;       register double S1,S2,S3,R1,R2,R3;
327       dim_t n_block=ilu->n_block;       const dim_t n=A->numRows;
328       dim_t n=ilu->n;       const dim_t n_block=A->row_block_size;
329         const index_t* colorOf = Paso_Pattern_borrowColoringPointer(A->pattern);
330         const dim_t num_colors = Paso_Pattern_getNumColors(A->pattern);
331         const index_t *ptr_main = Paso_SparseMatrix_borrowMainDiagonalPointer(A);
332            
333            
334       /* copy x into b*/       /* copy x into b*/
335       #pragma omp parallel for private(i) schedule(static)       #pragma omp parallel for private(i) schedule(static)
336       for (i=0;i<n*n_block;++i) x[i]=b[i];       for (i=0;i<n*n_block;++i) x[i]=b[i];
337       /* forward substitution */       /* forward substitution */
338       for (color=0;color<ilu->num_colors;++color) {       for (color=0;color<num_colors;++color) {
339             if (n_block==1) {             if (n_block==1) {
340                #pragma omp parallel for schedule(static) private(i,iptr_ik,k,S1,R1,iptr_main)                #pragma omp parallel for schedule(static) private(i,iptr_ik,k,S1,R1,iptr_main)
341                for (i = 0; i < n; ++i) {                for (i = 0; i < n; ++i) {
342                     if (ilu->colorOf[i]==color) {                     if (colorOf[i]==color) {
343                       /* x_i=x_i-a_ik*x_k */                                           /* x_i=x_i-a_ik*x_k */                    
344                       S1=x[i];                       S1=x[i];
345                       for (iptr_ik=ilu->pattern->ptr[i];iptr_ik<ilu->pattern->ptr[i+1]; ++iptr_ik) {                       for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {
346                            k=ilu->pattern->index[iptr_ik];                                                      k=A->pattern->index[iptr_ik];                          
347                            if (ilu->colorOf[k]<color) {                            if (colorOf[k]<color) {
348                               R1=x[k];                                                             R1=x[k];                              
349                               S1-=ilu->factors[iptr_ik]*R1;                               S1-=ilu->factors[iptr_ik]*R1;
350                            }                            }
351                       }                       }
352                       iptr_main=ilu->main_iptr[i];                       iptr_main=ptr_main[i];
353                       x[i]=ilu->factors[iptr_main]*S1;                       x[i]=ilu->factors[iptr_main]*S1;
354                     }                     }
355                }                }
356             } else if (n_block==2) {             } else if (n_block==2) {
357                #pragma omp parallel for schedule(static) private(i,iptr_ik,k,iptr_main,S1,S2,R1,R2)                #pragma omp parallel for schedule(static) private(i,iptr_ik,k,iptr_main,S1,S2,R1,R2)
358                for (i = 0; i < n; ++i) {                for (i = 0; i < n; ++i) {
359                     if (ilu->colorOf[i]==color) {                     if (colorOf[i]==color) {
360                       /* x_i=x_i-a_ik*x_k */                       /* x_i=x_i-a_ik*x_k */
361                       S1=x[2*i];                       S1=x[2*i];
362                       S2=x[2*i+1];                       S2=x[2*i+1];
363                       for (iptr_ik=ilu->pattern->ptr[i];iptr_ik<ilu->pattern->ptr[i+1]; ++iptr_ik) {                       for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {
364                            k=ilu->pattern->index[iptr_ik];                                                      k=A->pattern->index[iptr_ik];                          
365                            if (ilu->colorOf[k]<color) {                            if (colorOf[k]<color) {
366                               R1=x[2*k];                               R1=x[2*k];
367                               R2=x[2*k+1];                               R2=x[2*k+1];
368                               S1-=ilu->factors[4*iptr_ik  ]*R1+ilu->factors[4*iptr_ik+2]*R2;                               S1-=ilu->factors[4*iptr_ik  ]*R1+ilu->factors[4*iptr_ik+2]*R2;
369                               S2-=ilu->factors[4*iptr_ik+1]*R1+ilu->factors[4*iptr_ik+3]*R2;                               S2-=ilu->factors[4*iptr_ik+1]*R1+ilu->factors[4*iptr_ik+3]*R2;
370                            }                            }
371                       }                       }
372                       iptr_main=ilu->main_iptr[i];                       iptr_main=ptr_main[i];
373                       x[2*i  ]=ilu->factors[4*iptr_main  ]*S1+ilu->factors[4*iptr_main+2]*S2;                       x[2*i  ]=ilu->factors[4*iptr_main  ]*S1+ilu->factors[4*iptr_main+2]*S2;
374                       x[2*i+1]=ilu->factors[4*iptr_main+1]*S1+ilu->factors[4*iptr_main+3]*S2;                       x[2*i+1]=ilu->factors[4*iptr_main+1]*S1+ilu->factors[4*iptr_main+3]*S2;
375                     }                     }
# Line 398  void Paso_Solver_solveILU(Paso_Solver_IL Line 378  void Paso_Solver_solveILU(Paso_Solver_IL
378             } else if (n_block==3) {             } else if (n_block==3) {
379                #pragma omp parallel for schedule(static) private(i,iptr_ik,iptr_main,k,S1,S2,S3,R1,R2,R3)                #pragma omp parallel for schedule(static) private(i,iptr_ik,iptr_main,k,S1,S2,S3,R1,R2,R3)
380                for (i = 0; i < n; ++i) {                for (i = 0; i < n; ++i) {
381                     if (ilu->colorOf[i]==color) {                     if (colorOf[i]==color) {
382                       /* x_i=x_i-a_ik*x_k */                       /* x_i=x_i-a_ik*x_k */
383                       S1=x[3*i];                       S1=x[3*i];
384                       S2=x[3*i+1];                       S2=x[3*i+1];
385                       S3=x[3*i+2];                       S3=x[3*i+2];
386                       for (iptr_ik=ilu->pattern->ptr[i];iptr_ik<ilu->pattern->ptr[i+1]; ++iptr_ik) {                       for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {
387                            k=ilu->pattern->index[iptr_ik];                                                      k=A->pattern->index[iptr_ik];                          
388                            if (ilu->colorOf[k]<color) {                            if (colorOf[k]<color) {
389                               R1=x[3*k];                               R1=x[3*k];
390                               R2=x[3*k+1];                               R2=x[3*k+1];
391                               R3=x[3*k+2];                               R3=x[3*k+2];
# Line 414  void Paso_Solver_solveILU(Paso_Solver_IL Line 394  void Paso_Solver_solveILU(Paso_Solver_IL
394                               S3-=ilu->factors[9*iptr_ik+2]*R1+ilu->factors[9*iptr_ik+5]*R2+ilu->factors[9*iptr_ik+8]*R3;                               S3-=ilu->factors[9*iptr_ik+2]*R1+ilu->factors[9*iptr_ik+5]*R2+ilu->factors[9*iptr_ik+8]*R3;
395                            }                            }
396                       }                       }
397                       iptr_main=ilu->main_iptr[i];                       iptr_main=ptr_main[i];
398                       x[3*i  ]=ilu->factors[9*iptr_main  ]*S1+ilu->factors[9*iptr_main+3]*S2+ilu->factors[9*iptr_main+6]*S3;                       x[3*i  ]=ilu->factors[9*iptr_main  ]*S1+ilu->factors[9*iptr_main+3]*S2+ilu->factors[9*iptr_main+6]*S3;
399                       x[3*i+1]=ilu->factors[9*iptr_main+1]*S1+ilu->factors[9*iptr_main+4]*S2+ilu->factors[9*iptr_main+7]*S3;                       x[3*i+1]=ilu->factors[9*iptr_main+1]*S1+ilu->factors[9*iptr_main+4]*S2+ilu->factors[9*iptr_main+7]*S3;
400                       x[3*i+2]=ilu->factors[9*iptr_main+2]*S1+ilu->factors[9*iptr_main+5]*S2+ilu->factors[9*iptr_main+8]*S3;                       x[3*i+2]=ilu->factors[9*iptr_main+2]*S1+ilu->factors[9*iptr_main+5]*S2+ilu->factors[9*iptr_main+8]*S3;
# Line 423  void Paso_Solver_solveILU(Paso_Solver_IL Line 403  void Paso_Solver_solveILU(Paso_Solver_IL
403             }             }
404       }       }
405       /* backward substitution */       /* backward substitution */
406       for (color=(ilu->num_colors)-1;color>-1;--color) {       for (color=(num_colors)-1;color>-1;--color) {
407             if (n_block==1) {             if (n_block==1) {
408                #pragma omp parallel for schedule(static) private(i,iptr_ik,k,S1,R1)                #pragma omp parallel for schedule(static) private(i,iptr_ik,k,S1,R1)
409                for (i = 0; i < n; ++i) {                for (i = 0; i < n; ++i) {
410                     if (ilu->colorOf[i]==color) {                     if (colorOf[i]==color) {
411                       /* x_i=x_i-a_ik*x_k */                       /* x_i=x_i-a_ik*x_k */
412                       S1=x[i];                       S1=x[i];
413                       for (iptr_ik=ilu->pattern->ptr[i];iptr_ik<ilu->pattern->ptr[i+1]; ++iptr_ik) {                       for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {
414                            k=ilu->pattern->index[iptr_ik];                                                      k=A->pattern->index[iptr_ik];                          
415                            if (ilu->colorOf[k]>color) {                            if (colorOf[k]>color) {
416                               R1=x[k];                               R1=x[k];
417                               S1-=ilu->factors[iptr_ik]*R1;                               S1-=ilu->factors[iptr_ik]*R1;
418                            }                            }
# Line 443  void Paso_Solver_solveILU(Paso_Solver_IL Line 423  void Paso_Solver_solveILU(Paso_Solver_IL
423             } else if (n_block==2) {             } else if (n_block==2) {
424                #pragma omp parallel for schedule(static) private(i,iptr_ik,k,S1,S2,R1,R2)                #pragma omp parallel for schedule(static) private(i,iptr_ik,k,S1,S2,R1,R2)
425                for (i = 0; i < n; ++i) {                for (i = 0; i < n; ++i) {
426                     if (ilu->colorOf[i]==color) {                     if (colorOf[i]==color) {
427                       /* x_i=x_i-a_ik*x_k */                       /* x_i=x_i-a_ik*x_k */
428                       S1=x[2*i];                       S1=x[2*i];
429                       S2=x[2*i+1];                       S2=x[2*i+1];
430                       for (iptr_ik=ilu->pattern->ptr[i];iptr_ik<ilu->pattern->ptr[i+1]; ++iptr_ik) {                       for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {
431                            k=ilu->pattern->index[iptr_ik];                                                      k=A->pattern->index[iptr_ik];                          
432                            if (ilu->colorOf[k]>color) {                            if (colorOf[k]>color) {
433                               R1=x[2*k];                               R1=x[2*k];
434                               R2=x[2*k+1];                               R2=x[2*k+1];
435                               S1-=ilu->factors[4*iptr_ik  ]*R1+ilu->factors[4*iptr_ik+2]*R2;                               S1-=ilu->factors[4*iptr_ik  ]*R1+ilu->factors[4*iptr_ik+2]*R2;
# Line 463  void Paso_Solver_solveILU(Paso_Solver_IL Line 443  void Paso_Solver_solveILU(Paso_Solver_IL
443             } else if (n_block==3) {             } else if (n_block==3) {
444                #pragma omp parallel for schedule(static) private(i,iptr_ik,k,S1,S2,S3,R1,R2,R3)                #pragma omp parallel for schedule(static) private(i,iptr_ik,k,S1,S2,S3,R1,R2,R3)
445                for (i = 0; i < n; ++i) {                for (i = 0; i < n; ++i) {
446                     if (ilu->colorOf[i]==color) {                     if (colorOf[i]==color) {
447                       /* x_i=x_i-a_ik*x_k */                       /* x_i=x_i-a_ik*x_k */
448                       S1=x[3*i  ];                       S1=x[3*i  ];
449                       S2=x[3*i+1];                       S2=x[3*i+1];
450                       S3=x[3*i+2];                       S3=x[3*i+2];
451                       for (iptr_ik=ilu->pattern->ptr[i];iptr_ik<ilu->pattern->ptr[i+1]; ++iptr_ik) {                       for (iptr_ik=A->pattern->ptr[i];iptr_ik<A->pattern->ptr[i+1]; ++iptr_ik) {
452                            k=ilu->pattern->index[iptr_ik];                                                      k=A->pattern->index[iptr_ik];                          
453                            if (ilu->colorOf[k]>color) {                            if (colorOf[k]>color) {
454                               R1=x[3*k];                               R1=x[3*k];
455                               R2=x[3*k+1];                               R2=x[3*k+1];
456                               R3=x[3*k+2];                               R3=x[3*k+2];
# Line 489  void Paso_Solver_solveILU(Paso_Solver_IL Line 469  void Paso_Solver_solveILU(Paso_Solver_IL
469       }       }
470       return;       return;
471  }  }
472  /*  
  * $Log$  
  * Revision 1.2  2005/09/15 03:44:40  jgs  
  * Merge of development branch dev-02 back to main trunk on 2005-09-15  
  *  
  * Revision 1.1.2.1  2005/09/05 06:29:50  gross  
  * These files have been extracted from finley to define a stand alone libray for iterative  
  * linear solvers on the ALTIX. main entry through Paso_solve. this version compiles but  
  * has not been tested yet.  
  *  
  *  
  */  

Legend:
Removed from v.1628  
changed lines
  Added in v.3120

  ViewVC Help
Powered by ViewVC 1.1.26