00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049
00050 #include "OrthoPack.h"
00051
00052
00053 const int ORTHOPACK_MAXCGSIT = 5;
00054
00055
00056
00057
00058
00059
00060
00061
00062
00063
00064
00065 OrthoPack::OrthoPack() :
00066 callBLAS()
00067 {
00068
00069 }
00070
00071
00072 void OrthoPack::IteratedClassicalGS(Epetra_Vector *v, double *vnrm, const Epetra_MultiVector *A,
00073 double *work1, const Epetra_Operator *B) const {
00074
00075
00076
00077
00078 const double alpha = 0.5;
00079 double alpha2 = alpha*alpha;
00080
00081 double vnrm_old = 0.0;
00082 int i = 0;
00083
00084 int n = (A) ? A->MyLength() : v->MyLength();
00085 int m = (A) ? A->NumVectors() : 0;
00086
00087 Epetra_Vector w1(View, v->Map(), work1);
00088 if (B)
00089 B->Apply(*v, w1);
00090 else
00091 memcpy(work1, v->Values(), n*sizeof(double));
00092
00093 w1.Dot(*v, &vnrm_old);
00094 (*vnrm) = vnrm_old;
00095
00096 if (m > 0) {
00097
00098 double *work2 = new double[2*m];
00099
00100 for (i = 0; i < ORTHOPACK_MAXCGSIT; i ++) {
00101
00102 callBLAS.GEMV('T', n, m, 1.0, A->Values(), n, work1, 0.0, work2 + m);
00103 (A->Comm()).SumAll(work2 + m, work2, m);
00104 callBLAS.GEMV('N', n, m, -1.0, A->Values(), n, work2, 1.0, v->Values());
00105
00106 if (B)
00107 B->Apply(*v, w1);
00108 else
00109 memcpy(work1, v->Values(), n*sizeof(double));
00110
00111 w1.Dot((*v), vnrm);
00112
00113 if ((*vnrm) >= alpha2*vnrm_old)
00114 break;
00115
00116 vnrm_old = (*vnrm);
00117
00118 }
00119
00120 delete[] work2;
00121
00122 }
00123
00124 (*vnrm) = sqrt((*vnrm));
00125
00126 if (i >= ORTHOPACK_MAXCGSIT) {
00127 cout << "warning: loss of orthogonality. ";
00128 cout << "Iterated CGS not converged after " << ORTHOPACK_MAXCGSIT << " steps\n";
00129 }
00130
00131 }
00132
00133
00134
00135
00136
00137
00138
00139 void OrthoPack::ModifiedGS(Epetra_Vector *v, const Epetra_MultiVector *A) const {
00140
00141 int i;
00142 double s = 0.0;
00143
00144 int n = (A) ? A->MyLength() : v->MyLength();
00145 int m = (A) ? A->NumVectors() : 0;
00146
00147 for (i = 0; i < m; ++i) {
00148 (*A)(i)->Dot(*v, &s);
00149 callBLAS.AXPY(n, -s, A->Values() + i*n, v->Values());
00150 }
00151
00152 }
00153
00154
00155
00156
00157
00158
00159
00160 void OrthoPack::ModifiedGSB(Epetra_Vector *v, const Epetra_MultiVector *A,
00161 const Epetra_MultiVector *BA) const {
00162
00163 int i;
00164 double s = 0.0;
00165
00166 int n = (A) ? A->MyLength() : v->MyLength();
00167 int m = (A) ? A->NumVectors() : 0;
00168
00169 for (i = 0; i < m; ++i) {
00170 (*BA)(i)->Dot(*v, &s);
00171 callBLAS.AXPY(n, -s, A->Values() + i*n, v->Values());
00172 }
00173
00174 }
00175
00176