00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013 #ifndef __SUPERLU_dDEFS
00014 #define __SUPERLU_dDEFS
00015
00016
00017
00018
00019
00020
00021
00022 #include "superlu_defs.h"
00023
00024
00025
00026
00027
00028 typedef struct {
00029 int_t **Lrowind_bc_ptr;
00030 double **Lnzval_bc_ptr;
00031 int_t **Ufstnz_br_ptr;
00032 double **Unzval_br_ptr;
00033 #if 0
00034 int_t *Lsub_buf;
00035 double *Lval_buf;
00036 #endif
00037 int_t *Lsub_buf_2[2];
00038 double *Lval_buf_2[2];
00039 int_t *Usub_buf;
00040 double *Uval_buf;
00041 double *ujrow;
00042 int_t bufmax[NBUFFERS];
00043
00044
00045
00046
00047
00048
00049
00050
00051 int_t *ToRecv;
00052 int_t *ToSendD;
00053 int_t **ToSendR;
00054
00055
00056 int_t *fmod;
00057 int_t **fsendx_plist;
00058 int_t *frecv;
00059 int_t nfrecvx;
00060 int_t nfsendx;
00061 int_t *bmod;
00062 int_t **bsendx_plist;
00063 int_t *brecv;
00064 int_t nbrecvx;
00065 int_t nbsendx;
00066 int_t *mod_bit;
00067
00068
00069 int_t *ilsum;
00070
00071 int_t ldalsum;
00072 int_t SolveMsgSent;
00073 int_t SolveMsgVol;
00074 } LocalLU_t;
00075
00076 typedef struct {
00077 int_t *etree;
00078 Glu_persist_t *Glu_persist;
00079 LocalLU_t *Llu;
00080 } LUstruct_t;
00081
00082
00083 typedef struct {
00084 int_t lbnum;
00085 int_t indpos;
00086 } Ucb_indptr_t;
00087
00088
00089 typedef struct {
00090 int_t *extern_start;
00091 int_t *ind_tosend;
00092 int_t *ind_torecv;
00093 int_t *ptr_ind_tosend;
00094
00095 int_t *ptr_ind_torecv;
00096
00097 int *SendCounts;
00098
00099 int *RecvCounts;
00100
00101 double *val_tosend;
00102 double *val_torecv;
00103 int_t TotalIndSend;
00104
00105 int_t TotalValSend;
00106
00107 } pdgsmv_comm_t;
00108
00109
00110 typedef struct {
00111 int *B_to_X_SendCnt;
00112 int *X_to_B_SendCnt;
00113 int *ptr_to_ibuf, *ptr_to_dbuf;
00114 } pxgstrs_comm_t;
00115
00116
00117 typedef struct {
00118 int_t *row_to_proc;
00119 int_t *inv_perm_c;
00120 int_t num_diag_procs, *diag_procs, *diag_len;
00121 pdgsmv_comm_t *gsmv_comm;
00122 pxgstrs_comm_t *gstrs_comm;
00123 int_t *A_colind_gsmv;
00124
00125
00126
00127 } SOLVEstruct_t;
00128
00129
00130
00131
00132
00133
00134 #ifdef __cplusplus
00135 extern "C" {
00136 #endif
00137
00138
00139
00140 extern void
00141 dCreate_CompCol_Matrix_dist(SuperMatrix *, int_t, int_t, int_t, double *,
00142 int_t *, int_t *, Stype_t, Dtype_t, Mtype_t);
00143 extern void
00144 dCreate_CompRowLoc_Matrix_dist(SuperMatrix *, int_t, int_t, int_t, int_t,
00145 int_t, double *, int_t *, int_t *,
00146 Stype_t, Dtype_t, Mtype_t);
00147 extern void
00148 dCompRow_to_CompCol_dist(int_t, int_t, int_t, double *, int_t *, int_t *,
00149 double **, int_t **, int_t **);
00150 extern int
00151 pdCompRow_loc_to_CompCol_global(int_t, SuperMatrix *, gridinfo_t *,
00152 SuperMatrix *);
00153 extern void
00154 dCopy_CompCol_Matrix_dist(SuperMatrix *, SuperMatrix *);
00155 extern void
00156 dCreate_Dense_Matrix_dist(SuperMatrix *, int_t, int_t, double *, int_t,
00157 Stype_t, Dtype_t, Mtype_t);
00158 extern void
00159 dCreate_SuperNode_Matrix_dist(SuperMatrix *, int_t, int_t, int_t, double *,
00160 int_t *, int_t *, int_t *, int_t *, int_t *,
00161 Stype_t, Dtype_t, Mtype_t);
00162 extern void
00163 dCopy_Dense_Matrix_dist(int_t, int_t, double *, int_t,
00164 double *, int_t);
00165
00166 extern void dallocateA_dist (int_t, int_t, double **, int_t **, int_t **);
00167 extern void dGenXtrue_dist (int_t, int_t, double *, int_t);
00168 extern void dFillRHS_dist (char *, int_t, double *, int_t,
00169 SuperMatrix *, double *, int_t);
00170 extern int dcreate_matrix(SuperMatrix *, int, double **, int *,
00171 double **, int *, FILE *, gridinfo_t *);
00172
00173
00174 extern void dgsequ_dist (SuperMatrix *, double *, double *, double *,
00175 double *, double *, int_t *);
00176 extern double dlangs_dist (char *, SuperMatrix *);
00177 extern void dlaqgs_dist (SuperMatrix *, double *, double *, double,
00178 double, double, char *);
00179 extern void pdgsequ (SuperMatrix *, double *, double *, double *,
00180 double *, double *, int_t *, gridinfo_t *);
00181 extern double pdlangs (char *, SuperMatrix *, gridinfo_t *);
00182 extern void pdlaqgs (SuperMatrix *, double *, double *, double,
00183 double, double, char *);
00184 extern int pdPermute_Dense_Matrix(int_t, int_t, int_t [], int_t[],
00185 double [], int, double [], int, int,
00186 gridinfo_t *);
00187
00188 extern int sp_dtrsv_dist (char *, char *, char *, SuperMatrix *,
00189 SuperMatrix *, double *, int *);
00190 extern int sp_dgemv_dist (char *, double, SuperMatrix *, double *,
00191 int, double, double *, int);
00192 extern int sp_dgemm_dist (char *, char *, int, int, int, double,
00193 SuperMatrix *, double *, int, double,
00194 double *, int);
00195
00196 extern int_t ddistribute(fact_t, int_t, SuperMatrix *, Glu_freeable_t *,
00197 LUstruct_t *, gridinfo_t *);
00198 extern void pdgssvx_ABglobal(superlu_options_t *, SuperMatrix *,
00199 ScalePermstruct_t *, double *,
00200 int, int, gridinfo_t *, LUstruct_t *, double *,
00201 SuperLUStat_t *, int *);
00202 extern int_t pddistribute(fact_t, int_t, SuperMatrix *,
00203 ScalePermstruct_t *, Glu_freeable_t *,
00204 LUstruct_t *, gridinfo_t *);
00205 extern void pdgssvx(superlu_options_t *, SuperMatrix *,
00206 ScalePermstruct_t *, double *,
00207 int, int, gridinfo_t *, LUstruct_t *,
00208 SOLVEstruct_t *, double *, SuperLUStat_t *, int *);
00209 extern int dSolveInit(superlu_options_t *, SuperMatrix *, int_t [], int_t [],
00210 int_t, LUstruct_t *, gridinfo_t *, SOLVEstruct_t *);
00211 extern int_t pxgstrs_init(int_t, int_t, int_t, int_t,
00212 int_t [], int_t [], gridinfo_t *grid,
00213 Glu_persist_t *, SOLVEstruct_t *);
00214 extern void pxgstrs_finalize(pxgstrs_comm_t *);
00215 extern void dSolveFinalize(superlu_options_t *, SOLVEstruct_t *);
00216 extern void dldperm(int_t, int_t, int_t, int_t [], int_t [],
00217 double [], int_t *, double [], double []);
00218 extern int_t pdgstrf(superlu_options_t *, int, int, double,
00219 LUstruct_t*, gridinfo_t*, SuperLUStat_t*, int*);
00220 extern void pdgstrs_Bglobal(int_t, LUstruct_t *, gridinfo_t *,
00221 double *, int_t, int, SuperLUStat_t *, int *);
00222 extern void pdgstrs(int_t, LUstruct_t *, ScalePermstruct_t *, gridinfo_t *,
00223 double *, int_t, int_t, int_t, int, SOLVEstruct_t *,
00224 SuperLUStat_t *, int *);
00225 extern void dlsum_fmod(double *, double *, double *, double *,
00226 int, int, int_t , int_t *, int_t, int_t, int_t,
00227 int_t *, gridinfo_t *, LocalLU_t *,
00228 MPI_Request [], SuperLUStat_t *);
00229 extern void dlsum_bmod(double *, double *, double *,
00230 int, int_t, int_t *, int_t *, Ucb_indptr_t **,
00231 int_t **, int_t *, gridinfo_t *, LocalLU_t *,
00232 MPI_Request [], SuperLUStat_t *);
00233 extern void pdgsrfs(int_t, SuperMatrix *, double, LUstruct_t *,
00234 ScalePermstruct_t *, gridinfo_t *,
00235 double [], int_t, double [], int_t, int,
00236 SOLVEstruct_t *, double *, SuperLUStat_t *, int *);
00237 extern void pdgsrfs_ABXglobal(int_t, SuperMatrix *, double, LUstruct_t *,
00238 gridinfo_t *, double *, int_t, double *, int_t,
00239 int, double *, SuperLUStat_t *, int *);
00240 extern int pdgsmv_AXglobal_setup(SuperMatrix *, Glu_persist_t *,
00241 gridinfo_t *, int_t *, int_t *[],
00242 double *[], int_t *[], int_t []);
00243 extern int pdgsmv_AXglobal(int_t, int_t [], double [], int_t [],
00244 double [], double []);
00245 extern int pdgsmv_AXglobal_abs(int_t, int_t [], double [], int_t [],
00246 double [], double []);
00247 extern void pdgsmv_init(SuperMatrix *, int_t *, gridinfo_t *,
00248 pdgsmv_comm_t *);
00249 extern void pdgsmv(int_t, SuperMatrix *, gridinfo_t *, pdgsmv_comm_t *,
00250 double x[], double ax[]);
00251 extern void pdgsmv_finalize(pdgsmv_comm_t *);
00252
00253
00254 extern double *doubleMalloc_dist(int_t);
00255 extern double *doubleCalloc_dist(int_t);
00256 extern void *duser_malloc_dist (int_t, int_t);
00257 extern void duser_free_dist (int_t, int_t);
00258 extern int_t dQuerySpace_dist(int_t, LUstruct_t *, gridinfo_t *, mem_usage_t *);
00259 extern void Destroy_LU(int_t, gridinfo_t *, LUstruct_t *);
00260 extern void LUstructInit(const int_t, const int_t, LUstruct_t *);
00261 extern void LUstructFree(LUstruct_t *);
00262
00263
00264 extern void dfill_dist (double *, int_t, double);
00265 extern void dinf_norm_error_dist (int_t, int_t, double*, int_t,
00266 double*, int_t, gridinfo_t*);
00267 extern void pdinf_norm_error(int, int_t, int_t, double [], int_t,
00268 double [], int_t , gridinfo_t *);
00269 extern void dreadhb_dist (int, FILE *, int_t *, int_t *, int_t *,
00270 double **, int_t **, int_t **);
00271
00272
00273 extern int_t ddist_psymbtonum
00274 (fact_t, int_t, SuperMatrix *,
00275 ScalePermstruct_t *, Pslu_freeable_t *,
00276 LUstruct_t *, gridinfo_t *);
00277
00278
00279 extern void dPrintLblocks(int_t, int_t, gridinfo_t *, Glu_persist_t *,
00280 LocalLU_t *);
00281 extern void dPrintUblocks(int_t, int_t, gridinfo_t *, Glu_persist_t *,
00282 LocalLU_t *);
00283 extern void dPrint_CompCol_Matrix_dist(SuperMatrix *);
00284 extern void dPrint_Dense_Matrix_dist(SuperMatrix *);
00285 extern int dPrint_CompRowLoc_Matrix_dist(SuperMatrix *);
00286 extern int file_PrintDouble5(FILE *, char *, int_t, double *);
00287
00288
00289
00290 #ifdef USE_VENDOR_BLAS
00291 extern int dgemm_(char*, char*, int*, int*, int*, double*,
00292 double*, int*, double*, int*, double*,
00293 double*, int*, int, int);
00294 extern int dtrsv_(char*, char*, char*, int*, double*, int*,
00295 double*, int*, int, int, int);
00296 #else
00297 extern int dgemm_(char*, char*, int*, int*, int*, double*,
00298 double*, int*, double*, int*, double*,
00299 double*, int*);
00300 extern int dtrsv_(char*, char*, char*, int*, double*, int*,
00301 double*, int*);
00302 #endif
00303
00304 extern int dger_(int*, int*, double*, double*, int*,
00305 double*, int*, double*, int*);
00306
00307
00308 #ifdef __cplusplus
00309 }
00310 #endif
00311
00312 #endif
00313