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 #ifndef TEMPLATE_LAPACK_TRTRI_HEADER
00036 #define TEMPLATE_LAPACK_TRTRI_HEADER
00037
00038
00039 template<class Treal>
00040 int template_lapack_trtri(const char *uplo, const char *diag,
00041 const integer *n, Treal *a,
00042 const integer *lda, integer *info)
00043 {
00044
00045
00046
00047
00048
00049
00050
00051
00052
00053
00054
00055
00056
00057
00058
00059
00060
00061
00062
00063
00064
00065
00066
00067
00068
00069
00070
00071
00072
00073
00074
00075
00076
00077
00078
00079
00080
00081
00082
00083
00084
00085
00086
00087
00088
00089
00090
00091
00092
00093
00094
00095
00096
00097
00098
00099
00100
00101 integer c__1 = 1;
00102 integer c_n1 = -1;
00103 integer c__2 = 2;
00104 Treal c_b18 = 1.;
00105 Treal c_b22 = -1.;
00106
00107
00108 address a__1[2];
00109 integer a_dim1, a_offset, i__1, i__2[2], i__3, i__4, i__5;
00110 char ch__1[2];
00111
00112 integer j;
00113 logical upper;
00114 integer jb, nb, nn;
00115 logical nounit;
00116 #define a_ref(a_1,a_2) a[(a_2)*a_dim1 + a_1]
00117
00118
00119 a_dim1 = *lda;
00120 a_offset = 1 + a_dim1 * 1;
00121 a -= a_offset;
00122
00123
00124 *info = 0;
00125 upper = template_blas_lsame(uplo, "U");
00126 nounit = template_blas_lsame(diag, "N");
00127 if (! upper && ! template_blas_lsame(uplo, "L")) {
00128 *info = -1;
00129 } else if (! nounit && ! template_blas_lsame(diag, "U")) {
00130 *info = -2;
00131 } else if (*n < 0) {
00132 *info = -3;
00133 } else if (*lda < maxMACRO(1,*n)) {
00134 *info = -5;
00135 }
00136 if (*info != 0) {
00137 i__1 = -(*info);
00138 template_blas_erbla("TRTRI ", &i__1);
00139 return 0;
00140 }
00141
00142
00143
00144 if (*n == 0) {
00145 return 0;
00146 }
00147
00148
00149
00150 if (nounit) {
00151 i__1 = *n;
00152 for (*info = 1; *info <= i__1; ++(*info)) {
00153 if (a_ref(*info, *info) == 0.) {
00154 return 0;
00155 }
00156
00157 }
00158 *info = 0;
00159 }
00160
00161
00162
00163
00164 i__2[0] = 1, a__1[0] = (char*)uplo;
00165 i__2[1] = 1, a__1[1] = (char*)diag;
00166 template_blas_s_cat(ch__1, a__1, i__2, &c__2, (ftnlen)2);
00167 nb = template_lapack_ilaenv(&c__1, "DTRTRI", ch__1, n, &c_n1, &c_n1, &c_n1, (ftnlen)6, (
00168 ftnlen)2);
00169 if (nb <= 1 || nb >= *n) {
00170
00171
00172
00173 template_lapack_trti2(uplo, diag, n, &a[a_offset], lda, info);
00174 } else {
00175
00176
00177
00178 if (upper) {
00179
00180
00181
00182 i__1 = *n;
00183 i__3 = nb;
00184 for (j = 1; i__3 < 0 ? j >= i__1 : j <= i__1; j += i__3) {
00185
00186 i__4 = nb, i__5 = *n - j + 1;
00187 jb = minMACRO(i__4,i__5);
00188
00189
00190
00191 i__4 = j - 1;
00192 template_blas_trmm("Left", "Upper", "No transpose", diag, &i__4, &jb, &
00193 c_b18, &a[a_offset], lda, &a_ref(1, j), lda);
00194 i__4 = j - 1;
00195 template_blas_trsm("Right", "Upper", "No transpose", diag, &i__4, &jb, &
00196 c_b22, &a_ref(j, j), lda, &a_ref(1, j), lda);
00197
00198
00199
00200 template_lapack_trti2("Upper", diag, &jb, &a_ref(j, j), lda, info);
00201
00202 }
00203 } else {
00204
00205
00206
00207 nn = (*n - 1) / nb * nb + 1;
00208 i__3 = -nb;
00209 for (j = nn; i__3 < 0 ? j >= 1 : j <= 1; j += i__3) {
00210
00211 i__1 = nb, i__4 = *n - j + 1;
00212 jb = minMACRO(i__1,i__4);
00213 if (j + jb <= *n) {
00214
00215
00216
00217 i__1 = *n - j - jb + 1;
00218 template_blas_trmm("Left", "Lower", "No transpose", diag, &i__1, &jb,
00219 &c_b18, &a_ref(j + jb, j + jb), lda, &a_ref(j +
00220 jb, j), lda);
00221 i__1 = *n - j - jb + 1;
00222 template_blas_trsm("Right", "Lower", "No transpose", diag, &i__1, &jb,
00223 &c_b22, &a_ref(j, j), lda, &a_ref(j + jb, j),
00224 lda);
00225 }
00226
00227
00228
00229 template_lapack_trti2("Lower", diag, &jb, &a_ref(j, j), lda, info);
00230
00231 }
00232 }
00233 }
00234
00235 return 0;
00236
00237
00238
00239 }
00240
00241 #undef a_ref
00242
00243
00244 #endif