/usr/include/superlu-dist/superlu_zdefs.h is in libsuperlu-dist-dev 5.3.0+dfsg1-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 | /*! \file
Copyright (c) 2003, The Regents of the University of California, through
Lawrence Berkeley National Laboratory (subject to receipt of any required
approvals from U.S. Dept. of Energy)
All rights reserved.
The source code is distributed under BSD license, see the file License.txt
at the top-level directory.
*/
/*! @file
* \brief Distributed SuperLU data types and function prototypes
*
* <pre>
* -- Distributed SuperLU routine (version 4.1) --
* Lawrence Berkeley National Lab, Univ. of California Berkeley.
* November 1, 2007
* April 5, 2015
* </pre>
*/
#ifndef __SUPERLU_zDEFS /* allow multiple inclusions */
#define __SUPERLU_zDEFS
/*
* File name: superlu_zdefs.h
* Purpose: Distributed SuperLU data types and function prototypes
* History:
*/
#include "superlu_defs.h"
#include "dcomplex.h"
/*-- Auxiliary data type used in PxGSTRS/PxGSTRS1. */
typedef struct {
int_t lbnum; /* Row block number (local). */
int_t indpos; /* Starting position in Uindex[]. */
} Ucb_indptr_t;
/*
* On each processor, the blocks in L are stored in compressed block
* column format, the blocks in U are stored in compressed block row format.
*/
#define MAX_LOOKAHEADS 50
typedef struct {
int_t **Lrowind_bc_ptr; /* size ceil(NSUPERS/Pc) */
doublecomplex **Lnzval_bc_ptr; /* size ceil(NSUPERS/Pc) */
int_t **Ufstnz_br_ptr; /* size ceil(NSUPERS/Pr) */
doublecomplex **Unzval_br_ptr; /* size ceil(NSUPERS/Pr) */
#if 0
int_t *Lsub_buf; /* Buffer for the remote subscripts of L */
double *Lval_buf; /* Buffer for the remote nonzeros of L */
int_t *Usub_buf; /* Buffer for the remote subscripts of U */
doublecomplex *Uval_buf; /* Buffer for the remote nonzeros of U */
#endif
int_t *Lsub_buf_2[MAX_LOOKAHEADS]; /* Buffers for the remote subscripts of L*/
doublecomplex *Lval_buf_2[MAX_LOOKAHEADS]; /* Buffers for the remote nonzeros of L */
int_t *Usub_buf_2[MAX_LOOKAHEADS]; /* Buffer for the remote subscripts of U */
doublecomplex *Uval_buf_2[MAX_LOOKAHEADS]; /* Buffer for the remote nonzeros of U */
doublecomplex *ujrow; /* used in panel factorization. */
int_t bufmax[NBUFFERS]; /* Maximum buffer size across all MPI ranks:
* 0 : maximum size of Lsub_buf[]
* 1 : maximum size of Lval_buf[]
* 2 : maximum size of Usub_buf[]
* 3 : maximum size of Uval_buf[]
* 4 : maximum size of tempv[LDA]
*/
/*-- Record communication schedule for factorization. --*/
int *ToRecv; /* Recv from no one (0), left (1), and up (2).*/
int *ToSendD; /* Whether need to send down block row. */
int **ToSendR; /* List of processes to send right block col. */
/*-- Record communication schedule for forward/back solves. --*/
int_t *fmod; /* Modification count for L-solve */
int_t **fsendx_plist; /* Column process list to send down Xk */
int_t *frecv; /* Modifications to be recv'd in proc row */
int_t nfrecvx; /* Number of Xk I will receive in L-solve */
int_t nfsendx; /* Number of Xk I will send in L-solve */
int_t *bmod; /* Modification count for U-solve */
int_t **bsendx_plist; /* Column process list to send down Xk */
int_t *brecv; /* Modifications to be recv'd in proc row */
int_t nbrecvx; /* Number of Xk I will receive in U-solve */
int_t nbsendx; /* Number of Xk I will send in U-solve */
int_t *mod_bit; /* Flag contribution from each row blocks */
/*-- Auxiliary arrays used for forward/back solves. --*/
int_t *ilsum; /* Starting position of each supernode in lsum
(local) */
int_t ldalsum; /* LDA of lsum (local) */
int_t SolveMsgSent; /* Number of actual messages sent in LU-solve */
int_t SolveMsgVol; /* Volume of messages sent in the solve phase */
/*********************/
/* The following variables are used in the hybrid solver */
/*-- Counts to be used in U^{-T} triangular solve. -- */
int_t UT_SOLVE;
int_t L_SOLVE;
int_t FRECV;
int_t ut_ldalsum; /* LDA of lsum (local) */
int_t *ut_ilsum; /* ilsum in column-wise */
int_t *utmod; /* Modification count for Ut-solve. */
int_t **ut_sendx_plist; /* Row process list to send down Xk */
int_t *utrecv; /* Modifications to be recev'd in proc column. */
int_t n_utsendx; /* Number of Xk I will receive */
int_t n_utrecvx; /* Number of Xk I will send */
int_t n_utrecvmod;
int_t nroot;
int_t *ut_modbit;
int_t *Urbs;
Ucb_indptr_t **Ucb_indptr;/* Vertical linked list pointing to Uindex[] */
int_t **Ucb_valptr; /* Vertical linked list pointing to Unzval[] */
/* some additional counters for L solve */
int_t n;
int_t nleaf;
int_t nfrecvmod;
} LocalLU_t;
typedef struct {
int_t *etree;
Glu_persist_t *Glu_persist;
LocalLU_t *Llu;
} LUstruct_t;
/*-- Data structure for communication during matrix-vector multiplication. */
typedef struct {
int_t *extern_start;
int_t *ind_tosend; /* X indeices to be sent to other processes */
int_t *ind_torecv; /* X indeices to be received from other processes */
int_t *ptr_ind_tosend;/* Printers to ind_tosend[] (Size procs)
(also point to val_torecv) */
int_t *ptr_ind_torecv;/* Printers to ind_torecv[] (Size procs)
(also point to val_tosend) */
int *SendCounts; /* Numbers of X indices to be sent
(also numbers of X values to be received) */
int *RecvCounts; /* Numbers of X indices to be received
(also numbers of X values to be sent) */
doublecomplex *val_tosend; /* X values to be sent to other processes */
doublecomplex *val_torecv; /* X values to be received from other processes */
int_t TotalIndSend; /* Total number of indices to be sent
(also total number of values to be received) */
int_t TotalValSend; /* Total number of values to be sent.
(also total number of indices to be received) */
} pzgsmv_comm_t;
/*-- Data structure holding the information for the solution phase --*/
typedef struct {
int_t *row_to_proc;
int_t *inv_perm_c;
int_t num_diag_procs, *diag_procs, *diag_len;
pzgsmv_comm_t *gsmv_comm; /* communication metadata for SpMV,
required by IterRefine. */
pxgstrs_comm_t *gstrs_comm; /* communication metadata for SpTRSV. */
int_t *A_colind_gsmv; /* After pzgsmv_init(), the global column
indices of A are translated into the relative
positions in the gathered x-vector.
This is re-used in repeated calls to pzgsmv() */
int_t *xrow_to_proc; /* used by PDSLin */
} SOLVEstruct_t;
/***********************************************************************
* Function prototypes
***********************************************************************/
#ifdef __cplusplus
extern "C" {
#endif
/* Supernodal LU factor related */
extern void
zCreate_CompCol_Matrix_dist(SuperMatrix *, int_t, int_t, int_t, doublecomplex *,
int_t *, int_t *, Stype_t, Dtype_t, Mtype_t);
extern void
zCreate_CompRowLoc_Matrix_dist(SuperMatrix *, int_t, int_t, int_t, int_t,
int_t, doublecomplex *, int_t *, int_t *,
Stype_t, Dtype_t, Mtype_t);
extern void
zCompRow_to_CompCol_dist(int_t, int_t, int_t, doublecomplex *, int_t *, int_t *,
doublecomplex **, int_t **, int_t **);
extern int
pzCompRow_loc_to_CompCol_global(int_t, SuperMatrix *, gridinfo_t *,
SuperMatrix *);
extern void
zCopy_CompCol_Matrix_dist(SuperMatrix *, SuperMatrix *);
extern void
zCreate_Dense_Matrix_dist(SuperMatrix *, int_t, int_t, doublecomplex *, int_t,
Stype_t, Dtype_t, Mtype_t);
extern void
zCreate_SuperNode_Matrix_dist(SuperMatrix *, int_t, int_t, int_t, doublecomplex *,
int_t *, int_t *, int_t *, int_t *, int_t *,
Stype_t, Dtype_t, Mtype_t);
extern void
zCopy_Dense_Matrix_dist(int_t, int_t, doublecomplex *, int_t,
doublecomplex *, int_t);
extern void zallocateA_dist (int_t, int_t, doublecomplex **, int_t **, int_t **);
extern void zGenXtrue_dist (int_t, int_t, doublecomplex *, int_t);
extern void zFillRHS_dist (char *, int_t, doublecomplex *, int_t,
SuperMatrix *, doublecomplex *, int_t);
extern int zcreate_matrix(SuperMatrix *, int, doublecomplex **, int *,
doublecomplex **, int *, FILE *, gridinfo_t *);
extern int zcreate_matrix_rb(SuperMatrix *, int, doublecomplex **, int *,
doublecomplex **, int *, FILE *, gridinfo_t *);
extern int zcreate_matrix_dat(SuperMatrix *, int, doublecomplex **, int *,
doublecomplex **, int *, FILE *, gridinfo_t *);
/* Driver related */
extern void zgsequ_dist (SuperMatrix *, double *, double *, double *,
double *, double *, int_t *);
extern double zlangs_dist (char *, SuperMatrix *);
extern void zlaqgs_dist (SuperMatrix *, double *, double *, double,
double, double, char *);
extern void pzgsequ (SuperMatrix *, double *, double *, double *,
double *, double *, int_t *, gridinfo_t *);
extern double pzlangs (char *, SuperMatrix *, gridinfo_t *);
extern void pzlaqgs (SuperMatrix *, double *, double *, double,
double, double, char *);
extern int pzPermute_Dense_Matrix(int_t, int_t, int_t [], int_t[],
doublecomplex [], int, doublecomplex [], int, int,
gridinfo_t *);
extern int sp_ztrsv_dist (char *, char *, char *, SuperMatrix *,
SuperMatrix *, doublecomplex *, int *);
extern int sp_zgemv_dist (char *, doublecomplex, SuperMatrix *, doublecomplex *,
int, doublecomplex, doublecomplex *, int);
extern int sp_zgemm_dist (char *, int, doublecomplex, SuperMatrix *,
doublecomplex *, int, doublecomplex, doublecomplex *, int);
extern float zdistribute(fact_t, int_t, SuperMatrix *, Glu_freeable_t *,
LUstruct_t *, gridinfo_t *);
extern void pzgssvx_ABglobal(superlu_dist_options_t *, SuperMatrix *,
ScalePermstruct_t *, doublecomplex *,
int, int, gridinfo_t *, LUstruct_t *, double *,
SuperLUStat_t *, int *);
extern float pzdistribute(fact_t, int_t, SuperMatrix *,
ScalePermstruct_t *, Glu_freeable_t *,
LUstruct_t *, gridinfo_t *);
extern void pzgssvx(superlu_dist_options_t *, SuperMatrix *,
ScalePermstruct_t *, doublecomplex *,
int, int, gridinfo_t *, LUstruct_t *,
SOLVEstruct_t *, double *, SuperLUStat_t *, int *);
extern int zSolveInit(superlu_dist_options_t *, SuperMatrix *, int_t [], int_t [],
int_t, LUstruct_t *, gridinfo_t *, SOLVEstruct_t *);
extern void zSolveFinalize(superlu_dist_options_t *, SOLVEstruct_t *);
extern int_t pxgstrs_init(int_t, int_t, int_t, int_t,
int_t [], int_t [], gridinfo_t *grid,
Glu_persist_t *, SOLVEstruct_t *);
extern void pxgstrs_finalize(pxgstrs_comm_t *);
extern int zldperm_dist(int_t, int_t, int_t, int_t [], int_t [],
doublecomplex [], int_t *, double [], double []);
extern int static_schedule(superlu_dist_options_t *, int, int,
LUstruct_t *, gridinfo_t *, SuperLUStat_t *,
int_t *, int_t *, int *);
extern void LUstructInit(const int_t, LUstruct_t *);
extern void LUstructFree(LUstruct_t *);
extern void Destroy_LU(int_t, gridinfo_t *, LUstruct_t *);
/* #define GPU_PROF
#define IPM_PROF */
extern int_t pzgstrf(superlu_dist_options_t *, int, int, double,
LUstruct_t*, gridinfo_t*, SuperLUStat_t*, int*);
extern void pzgstrs_Bglobal(int_t, LUstruct_t *, gridinfo_t *,
doublecomplex *, int_t, int, SuperLUStat_t *, int *);
extern void pzgstrs(int_t, LUstruct_t *, ScalePermstruct_t *, gridinfo_t *,
doublecomplex *, int_t, int_t, int_t, int, SOLVEstruct_t *,
SuperLUStat_t *, int *);
extern void zlsum_fmod(doublecomplex *, doublecomplex *, doublecomplex *, doublecomplex *,
int, int, int_t , int_t *, int_t, int_t, int_t,
int_t *, gridinfo_t *, LocalLU_t *,
MPI_Request [], SuperLUStat_t *);
extern void zlsum_bmod(doublecomplex *, doublecomplex *, doublecomplex *,
int, int_t, int_t *, int_t *, Ucb_indptr_t **,
int_t **, int_t *, gridinfo_t *, LocalLU_t *,
MPI_Request [], SuperLUStat_t *);
extern void pzgsrfs(int_t, SuperMatrix *, double, LUstruct_t *,
ScalePermstruct_t *, gridinfo_t *,
doublecomplex [], int_t, doublecomplex [], int_t, int,
SOLVEstruct_t *, double *, SuperLUStat_t *, int *);
extern void pzgsrfs_ABXglobal(int_t, SuperMatrix *, double, LUstruct_t *,
gridinfo_t *, doublecomplex *, int_t, doublecomplex *, int_t,
int, double *, SuperLUStat_t *, int *);
extern int pzgsmv_AXglobal_setup(SuperMatrix *, Glu_persist_t *,
gridinfo_t *, int_t *, int_t *[],
doublecomplex *[], int_t *[], int_t []);
extern int pzgsmv_AXglobal(int_t, int_t [], doublecomplex [], int_t [],
doublecomplex [], doublecomplex []);
extern int pzgsmv_AXglobal_abs(int_t, int_t [], doublecomplex [], int_t [],
doublecomplex [], double []);
extern void pzgsmv_init(SuperMatrix *, int_t *, gridinfo_t *,
pzgsmv_comm_t *);
extern void pzgsmv(int_t, SuperMatrix *, gridinfo_t *, pzgsmv_comm_t *,
doublecomplex x[], doublecomplex ax[]);
extern void pzgsmv_finalize(pzgsmv_comm_t *);
/* Memory-related */
extern doublecomplex *doublecomplexMalloc_dist(int_t);
extern doublecomplex *doublecomplexCalloc_dist(int_t);
extern double *doubleMalloc_dist(int_t);
extern double *doubleCalloc_dist(int_t);
extern void *duser_malloc_dist (int_t, int_t);
extern void duser_free_dist (int_t, int_t);
extern int_t zQuerySpace_dist(int_t, LUstruct_t *, gridinfo_t *,
SuperLUStat_t *, superlu_dist_mem_usage_t *);
/* Auxiliary routines */
extern void zfill_dist (doublecomplex *, int_t, doublecomplex);
extern void zinf_norm_error_dist (int_t, int_t, doublecomplex*, int_t,
doublecomplex*, int_t, gridinfo_t*);
extern void pzinf_norm_error(int, int_t, int_t, doublecomplex [], int_t,
doublecomplex [], int_t , gridinfo_t *);
extern void zreadhb_dist (int, FILE *, int_t *, int_t *, int_t *,
doublecomplex **, int_t **, int_t **);
extern void zreadtriple_dist(FILE *, int_t *, int_t *, int_t *,
doublecomplex **, int_t **, int_t **);
extern void zreadrb_dist(int, FILE *, int_t *, int_t *, int_t *,
doublecomplex **, int_t **, int_t **);
extern void zreadMM_dist(FILE *, int_t *, int_t *, int_t *,
doublecomplex **, int_t **, int_t **);
/* Distribute the data for numerical factorization */
extern float zdist_psymbtonum(fact_t, int_t, SuperMatrix *,
ScalePermstruct_t *, Pslu_freeable_t *,
LUstruct_t *, gridinfo_t *);
extern void pzGetDiagU(int_t, LUstruct_t *, gridinfo_t *, doublecomplex *);
/* Routines for debugging */
extern void zPrintLblocks(int, int_t, gridinfo_t *, Glu_persist_t *,
LocalLU_t *);
extern void zPrintUblocks(int, int_t, gridinfo_t *, Glu_persist_t *,
LocalLU_t *);
extern void zPrint_CompCol_Matrix_dist(SuperMatrix *);
extern void zPrint_Dense_Matrix_dist(SuperMatrix *);
extern int zPrint_CompRowLoc_Matrix_dist(SuperMatrix *);
extern void PrintDoublecomplex(char *, int_t, doublecomplex *);
extern int file_PrintDoublecomplex(FILE *fp, char *, int_t, doublecomplex *);
/* BLAS */
#ifdef USE_VENDOR_BLAS
extern void zgemm_(const char*, const char*, const int*, const int*, const int*,
const doublecomplex*, const doublecomplex*, const int*, const doublecomplex*,
const int*, const doublecomplex*, doublecomplex*, const int*, int, int);
extern void ztrsv_(char*, char*, char*, int*, doublecomplex*, int*,
doublecomplex*, int*, int, int, int);
extern void ztrsm_(char*, char*, char*, char*, int*, int*,
doublecomplex*, doublecomplex*, int*, doublecomplex*,
int*, int, int, int, int);
extern void zgemv_(char *, int *, int *, doublecomplex *, doublecomplex *a, int *,
doublecomplex *, int *, doublecomplex *, doublecomplex *, int *, int);
extern void zgeru_(int*, int*, doublecomplex*, doublecomplex*, int*,
doublecomplex*, int*, doublecomplex*, int*);
#else
extern int zgemm_(const char*, const char*, const int*, const int*, const int*,
const doublecomplex*, const doublecomplex*, const int*, const doublecomplex*,
const int*, const doublecomplex*, doublecomplex*, const int*);
extern int ztrsv_(char*, char*, char*, int*, doublecomplex*, int*,
doublecomplex*, int*);
extern int ztrsm_(char*, char*, char*, char*, int*, int*,
doublecomplex*, doublecomplex*, int*, doublecomplex*, int*);
extern int zgemv_(char *, int *, int *, doublecomplex *, doublecomplex *a, int *,
doublecomplex *, int *, doublecomplex *, doublecomplex *, int *);
extern int zgeru_(int*, int*, doublecomplex*, doublecomplex*, int*,
doublecomplex*, int*, doublecomplex*, int*);
#endif
#ifdef __cplusplus
}
#endif
#endif /* __SUPERLU_dDEFS */
|