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

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

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

revision 2830 by artak, Tue Dec 22 01:24:40 2009 UTC revision 2831 by artak, Thu Jan 7 05:08:19 2010 UTC
# Line 116  Paso_Solver_AMG* Paso_Solver_getAMG(Paso Line 116  Paso_Solver_AMG* Paso_Solver_getAMG(Paso
116    Paso_Pattern* temp2=NULL;    Paso_Pattern* temp2=NULL;
117    */    */
118    bool_t verbose=options->verbose;    bool_t verbose=options->verbose;
119      bool_t timing=0;
120      
121    dim_t n=A_p->numRows;    dim_t n=A_p->numRows;
122    dim_t n_block=A_p->row_block_size;    dim_t n_block=A_p->row_block_size;
123    index_t* mis_marker=NULL;      index_t* mis_marker=NULL;  
# Line 207  Paso_Solver_AMG* Paso_Solver_getAMG(Paso Line 209  Paso_Solver_AMG* Paso_Solver_getAMG(Paso
209                if (options->smoother == PASO_JACOBI)                if (options->smoother == PASO_JACOBI)
210                  out->Smoother->Jacobi=Paso_Solver_getJacobi(A_p);                  out->Smoother->Jacobi=Paso_Solver_getJacobi(A_p);
211                else if (options->smoother == PASO_GS)                else if (options->smoother == PASO_GS)
212                  out->Smoother->GS=Paso_Solver_getGS(A_p,verbose));                  out->Smoother->GS=Paso_Solver_getGS(A_p,verbose);
213              #endif              #endif
214           #endif           #endif
215                    
# Line 240  Paso_Solver_AMG* Paso_Solver_getAMG(Paso Line 242  Paso_Solver_AMG* Paso_Solver_getAMG(Paso
242          }          }
243          else {          else {
244             /*Default coarseneing*/             /*Default coarseneing*/
245              Paso_Pattern_Standard(A_p,mis_marker,options->coarsening_threshold);              /*Paso_Pattern_Standard(A_p,mis_marker,options->coarsening_threshold);*/
246                /*Paso_Pattern_Read("RS.spl",n,mis_marker);*/
247              /*Paso_Pattern_YS(A_p,mis_marker,options->coarsening_threshold);*/              /*Paso_Pattern_YS(A_p,mis_marker,options->coarsening_threshold);*/
248              /*Paso_Pattern_greedy_Agg(A_p,mis_marker,options->coarsening_threshold);*/              /*Paso_Pattern_greedy_Agg(A_p,mis_marker,options->coarsening_threshold);*/
249              /*Paso_Pattern_greedy(A_p->pattern,mis_marker);*/              Paso_Pattern_greedy(A_p->pattern,mis_marker);
250              /*Paso_Pattern_Aggregiation(A_p,mis_marker,options->coarsening_threshold);*/              /*Paso_Pattern_Aggregiation(A_p,mis_marker,options->coarsening_threshold);*/
251                            
252          }          }
253                    
254          if (verbose) fprintf(stdout,"timing: Profilining for level %d:\n",level);          if (timing) fprintf(stdout,"timing: Profilining for level %d:\n",level);
255                    
256          time0=Paso_timer()-time0;          time0=Paso_timer()-time0;
257          if (verbose) fprintf(stdout,"timing: Coarsening: %e\n",time0);          if (timing) fprintf(stdout,"timing: Coarsening: %e\n",time0);
258    
259          #pragma omp parallel for private(i) schedule(static)          #pragma omp parallel for private(i) schedule(static)
260          for (i = 0; i < n; ++i) counter[i]=mis_marker[i];          for (i = 0; i < n; ++i) counter[i]=mis_marker[i];
# Line 301  Paso_Solver_AMG* Paso_Solver_getAMG(Paso Line 304  Paso_Solver_AMG* Paso_Solver_getAMG(Paso
304                                            
305                   }                   }
306                }                }
307                /*                
308                 if(level==1) {                /* if(level==1) {
309                     printf("##TOTAL: %d, ELIMINATED: %d\n",n,out->n_F);                     printf("##TOTAL: %d, ELIMINATED: %d\n",n,out->n_F);
310                     for (i = 0; i < n; ++i) {                     for (i = 0; i < n; ++i) {
311                      printf("##%d %d\n",i,mis_marker[i]);                      printf("##%d %d\n",i,!mis_marker[i]);
312                     }                     }
313                  }                  }
314                */                */
# Line 360  Paso_Solver_AMG* Paso_Solver_getAMG(Paso Line 363  Paso_Solver_AMG* Paso_Solver_getAMG(Paso
363                        time0=Paso_timer();                        time0=Paso_timer();
364                        Paso_SparseMatrix_updateWeights(A_p,out->W_FC,mis_marker);                        Paso_SparseMatrix_updateWeights(A_p,out->W_FC,mis_marker);
365                        time0=Paso_timer()-time0;                        time0=Paso_timer()-time0;
366                        if (verbose) fprintf(stdout,"timing: updateWeights: %e\n",time0);                        if (timing) fprintf(stdout,"timing: updateWeights: %e\n",time0);
367                    
368                                                
369                        /*sprintf(filename,"W_FCafter_%d",level);                        /*sprintf(filename,"W_FCafter_%d",level);
# Line 371  Paso_Solver_AMG* Paso_Solver_getAMG(Paso Line 374  Paso_Solver_AMG* Paso_Solver_getAMG(Paso
374                        time0=Paso_timer();                        time0=Paso_timer();
375                        out->P=Paso_SparseMatrix_getProlongation(out->W_FC,mis_marker);                        out->P=Paso_SparseMatrix_getProlongation(out->W_FC,mis_marker);
376                        time0=Paso_timer()-time0;                        time0=Paso_timer()-time0;
377                        if (verbose) fprintf(stdout,"timing: getProlongation: %e\n",time0);                        if (timing) fprintf(stdout,"timing: getProlongation: %e\n",time0);
378                          /*out->P=Paso_SparseMatrix_loadMM_toCSR("P1.mtx");*/
379                                                
380                                                
381                        /*                        /*sprintf(filename,"P_%d",level);
                        sprintf(filename,"P_%d",level);  
382                        Paso_SparseMatrix_saveMM(out->P,filename);                        Paso_SparseMatrix_saveMM(out->P,filename);
383                        */                        */
384                                                
385                        time0=Paso_timer();                        time0=Paso_timer();
386                        out->R=Paso_SparseMatrix_getRestriction(out->P);                        out->R=Paso_SparseMatrix_getRestriction(out->P);
387                        time0=Paso_timer()-time0;                        time0=Paso_timer()-time0;
388                        if (verbose) fprintf(stdout,"timing: getRestriction: %e\n",time0);                        if (timing) fprintf(stdout,"timing: getRestriction: %e\n",time0);
389                          /*out->R=Paso_SparseMatrix_loadMM_toCSR("R1.mtx");*/
390                                                
391                                                /*
392                        /*sprintf(filename,"R_%d",level);                        sprintf(filename,"R_%d",level);
393                        Paso_SparseMatrix_saveMM(out->R,filename);                        Paso_SparseMatrix_saveMM(out->R,filename);
394                        */                        */
395                                            
# Line 395  Paso_Solver_AMG* Paso_Solver_getAMG(Paso Line 399  Paso_Solver_AMG* Paso_Solver_getAMG(Paso
399                      time0=Paso_timer();                      time0=Paso_timer();
400                                            
401                      Atemp=Paso_SparseMatrix_MatrixMatrix(A_p,out->P);                      Atemp=Paso_SparseMatrix_MatrixMatrix(A_p,out->P);
402                        
403                      A_c=Paso_SparseMatrix_MatrixMatrix(out->R,Atemp);                      A_c=Paso_SparseMatrix_MatrixMatrix(out->R,Atemp);
404                                            
405                        /*A_c=Paso_SparseMatrix_loadMM_toCSR("A_C1.mtx");*/
406                        
407                      Paso_SparseMatrix_free(Atemp);                      Paso_SparseMatrix_free(Atemp);
408                                            
409                      /*A_c=Paso_Solver_getCoarseMatrix(A_p,out->R,out->P);*/                      /*A_c=Paso_Solver_getCoarseMatrix(A_p,out->R,out->P);*/
410                      time0=Paso_timer()-time0;                      time0=Paso_timer()-time0;
411                      if (verbose) fprintf(stdout,"timing: getCoarseMatrix: %e\n",time0);                      if (timing) fprintf(stdout,"timing: getCoarseMatrix: %e\n",time0);
412                                            
413                                                                                    
414                      /*Paso_Solver_getCoarseMatrix(A_c, A_p,out->R,out->P);*/                      /*Paso_Solver_getCoarseMatrix(A_c, A_p,out->R,out->P);*/
415                                            
416                                            /*
417                      /*sprintf(filename,"A_C_%d",level);                      sprintf(filename,"A_C_%d",level);
418                      Paso_SparseMatrix_saveMM(A_c,filename);                      Paso_SparseMatrix_saveMM(A_c,filename);
419                      */                      */
420                                            
# Line 487  void Paso_Solver_solveAMG(Paso_Solver_AM Line 494  void Paso_Solver_solveAMG(Paso_Solver_AM
494       dim_t i;       dim_t i;
495       double time0=0;       double time0=0;
496       double *r=NULL, *x0=NULL;       double *r=NULL, *x0=NULL;
497       bool_t verbose=0;       bool_t timing=0;
498            
499       dim_t post_sweeps=amg->post_sweeps;       dim_t post_sweeps=amg->post_sweeps;
500       dim_t pre_sweeps=amg->pre_sweeps;       dim_t pre_sweeps=amg->pre_sweeps;
# Line 495  void Paso_Solver_solveAMG(Paso_Solver_AM Line 502  void Paso_Solver_solveAMG(Paso_Solver_AM
502       #ifdef UMFPACK       #ifdef UMFPACK
503            Paso_UMFPACK_Handler * ptr=NULL;            Paso_UMFPACK_Handler * ptr=NULL;
504       #endif       #endif
       
505                        
506       r=MEMALLOC(amg->n,double);       r=MEMALLOC(amg->n,double);
507       x0=MEMALLOC(amg->n,double);       x0=MEMALLOC(amg->n,double);
# Line 504  void Paso_Solver_solveAMG(Paso_Solver_AM Line 510  void Paso_Solver_solveAMG(Paso_Solver_AM
510                
511        time0=Paso_timer();        time0=Paso_timer();
512        /*If all unknown are eliminated then Jacobi is the best preconditioner*/        /*If all unknown are eliminated then Jacobi is the best preconditioner*/
513        /*Paso_Solver_solveJacobi(amg->GS,x,b);*/  
         
514        if (amg->n_F==0 || amg->n_F==amg->n) {        if (amg->n_F==0 || amg->n_F==amg->n) {
515          if(amg->Smoother->ID==PASO_JACOBI)          if(amg->Smoother->ID==PASO_JACOBI)
516              Paso_Solver_solveJacobi(amg->Smoother->Jacobi,x,b);              Paso_Solver_solveJacobi(amg->Smoother->Jacobi,x,b);
# Line 518  void Paso_Solver_solveAMG(Paso_Solver_AM Line 523  void Paso_Solver_solveAMG(Paso_Solver_AM
523         #else         #else
524            #ifdef UMFPACK            #ifdef UMFPACK
525               ptr=(Paso_UMFPACK_Handler *)(amg->solver);               ptr=(Paso_UMFPACK_Handler *)(amg->solver);
526               Paso_UMFPACK1(&ptr,amg->A,x,b,verbose);               Paso_UMFPACK1(&ptr,amg->A,x,b,timing);
527               amg->solver=(void*) ptr;               amg->solver=(void*) ptr;
528            #else                  #else      
529             if(amg->Smoother->ID==PASO_JACOBI)             if(amg->Smoother->ID==PASO_JACOBI)
# Line 530  void Paso_Solver_solveAMG(Paso_Solver_AM Line 535  void Paso_Solver_solveAMG(Paso_Solver_AM
535         }         }
536                
537         time0=Paso_timer()-time0;         time0=Paso_timer()-time0;
538         if (verbose) fprintf(stdout,"timing: DIRECT SOLVER: %e\n",time0);         if (timing) fprintf(stdout,"timing: DIRECT SOLVER: %e\n",time0);
539                
540       } else {       } else {
541          /* presmoothing */          /* presmoothing */
# Line 564  void Paso_Solver_solveAMG(Paso_Solver_AM Line 569  void Paso_Solver_solveAMG(Paso_Solver_AM
569           /***********/           /***********/
570                    
571           time0=Paso_timer()-time0;           time0=Paso_timer()-time0;
572           if (verbose) fprintf(stdout,"timing: Presmooting: %e\n",time0);           if (timing) fprintf(stdout,"timing: Presmooting: %e\n",time0);
573           /* end of presmoothing */           /* end of presmoothing */
574                    
575           time0=Paso_timer();           time0=Paso_timer();
# Line 574  void Paso_Solver_solveAMG(Paso_Solver_AM Line 579  void Paso_Solver_solveAMG(Paso_Solver_AM
579           /*r=b-Ax*/           /*r=b-Ax*/
580           Paso_SparseMatrix_MatrixVector_CSR_OFFSET0(-1.,amg->A,x,1.,r);           Paso_SparseMatrix_MatrixVector_CSR_OFFSET0(-1.,amg->A,x,1.,r);
581                    
582          /* b_c <- R*r  */          /* b_c = R*r  */
583           Paso_SparseMatrix_MatrixVector_CSR_OFFSET0(1.,amg->R,r,0.,amg->b_C);           Paso_SparseMatrix_MatrixVector_CSR_OFFSET0(1.,amg->R,r,0.,amg->b_C);
584                    
585          time0=Paso_timer()-time0;          time0=Paso_timer()-time0;
586          if (verbose) fprintf(stdout,"timing: Before next level: %e\n",time0);          if (timing) fprintf(stdout,"timing: Before next level: %e\n",time0);
587                    
588          /* x_C=AMG(b_C)     */          /* x_C=AMG(b_C)     */
589          Paso_Solver_solveAMG(amg->AMG_of_Coarse,amg->x_C,amg->b_C);          Paso_Solver_solveAMG(amg->AMG_of_Coarse,amg->x_C,amg->b_C);
590                    
591          time0=Paso_timer();          time0=Paso_timer();
592                    
593          /* x_0 <- P*x_c */          /* x_0 = P*x_c */
594          Paso_SparseMatrix_MatrixVector_CSR_OFFSET0(1.,amg->P,amg->x_C,0.,x0);          Paso_SparseMatrix_MatrixVector_CSR_OFFSET0(1.,amg->P,amg->x_C,0.,x0);
595                    
596          /* x=x+x0 */          /* x=x+x0 */
# Line 605  void Paso_Solver_solveAMG(Paso_Solver_AM Line 610  void Paso_Solver_solveAMG(Paso_Solver_AM
610        else if (amg->Smoother->ID==PASO_GS)            else if (amg->Smoother->ID==PASO_GS)    
611              Paso_Solver_solveGS(amg->Smoother->GS,x0,r);              Paso_Solver_solveGS(amg->Smoother->GS,x0,r);
612                
         
613        #pragma omp parallel for private(i) schedule(static)        #pragma omp parallel for private(i) schedule(static)
614        for (i=0;i<amg->n;++i)  {        for (i=0;i<amg->n;++i)  {
615         x[i]+=x0[i];         x[i]+=x0[i];
616        }        }
617          
618          /***************/          /***************/
619          while(post_sweeps>1) {          while(post_sweeps>1) {
620                        
# Line 632  void Paso_Solver_solveAMG(Paso_Solver_AM Line 637  void Paso_Solver_solveAMG(Paso_Solver_AM
637          /**************/          /**************/
638                
639        time0=Paso_timer()-time0;        time0=Paso_timer()-time0;
640        if (verbose) fprintf(stdout,"timing: Postsmoothing: %e\n",time0);        if (timing) fprintf(stdout,"timing: Postsmoothing: %e\n",time0);
641    
642        /*end of postsmoothing*/        /*end of postsmoothing*/
643            

Legend:
Removed from v.2830  
changed lines
  Added in v.2831

  ViewVC Help
Powered by ViewVC 1.1.26