Actual source code: tfs.h
petsc-3.7.7 2017-09-25
5: /**********************************const.h*************************************
7: Author: Henry M. Tufo III
9: e-mail: hmt@cs.brown.edu
11: snail-mail:
12: Division of Applied Mathematics
13: Brown University
14: Providence, RI 02912
16: Last Modification:
17: 6.21.97
18: ***********************************const.h************************************/
20: /**********************************const.h*************************************
21: File Description:
22: -----------------
24: ***********************************const.h************************************/
25: #include <petscsys.h>
26: #include <petscblaslapack.h>
28: #define X 0
29: #define Y 1
30: #define Z 2
31: #define XY 3
32: #define XZ 4
33: #define YZ 5
36: #define THRESH 0.2
37: #define N_HALF 4096
38: #define PRIV_BUF_SZ 45
40: /*4096 8192 32768 65536 1048576 */
41: #define MAX_MSG_BUF 32768
43: #define FULL 2
44: #define PARTIAL 1
45: #define NONE 0
47: #define BYTE 8
48: #define BIT_0 0x1
49: #define BIT_1 0x2
50: #define BIT_2 0x4
51: #define BIT_3 0x8
52: #define BIT_4 0x10
53: #define BIT_5 0x20
54: #define BIT_6 0x40
55: #define BIT_7 0x80
56: #define TOP_BIT PETSC_MIN_INT
58: #define C 0
61: #define MAX_VEC 1674
62: #define FORMAT 30
63: #define MAX_COL_LEN 100
64: #define MAX_LINE FORMAT*MAX_COL_LEN
65: #define DELIM " \n \t"
66: #define LINE 12
67: #define C_LINE 80
69: #define UT 5 /* dump upper 1/2 */
70: #define LT 6 /* dump lower 1/2 */
71: #define SYMM 8 /* we assume symm and dump upper 1/2 */
72: #define NON_SYMM 9
74: #define ROW 10
75: #define COL 11
77: #define EPS 1.0e-14
78: #define EPS2 1.0e-07
81: #define MPI 1
82: #define NX 2
84: #define LOG2(x) (PetscScalar)log((double)x)/log(2)
85: #define SWAP(a,b) temp=(a); (a)=(b); (b)=temp;
86: #define P_SWAP(a,b) ptr =(a); (a)=(b); (b)=ptr;
88: #define MAX_FABS(x,y) (PetscAbsScalar(x)>PetscAbsScalar(y)) ? ((PetscScalar)x) : ((PetscScalar)y)
89: #define MIN_FABS(x,y) (PetscAbsScalar(x)<PetscAbsScalar(y)) ? ((PetscScalar)x) : ((PetscScalar)y)
91: /* specer's existence ... can be done w/MAX_ABS */
92: #define EXISTS(x,y) ((x)==0.0) ? (y) : (x)
94: #define MULT_NEG_ONE(a) (a) *= -1;
95: #define NEG(a) (a) |= BIT_31;
96: #define POS(a) (a) &= INT_MAX;
101: /**********************************types.h*************************************
103: Author: Henry M. Tufo III
105: e-mail: hmt@cs.brown.edu
107: snail-mail:
108: Division of Applied Mathematics
109: Brown University
110: Providence, RI 02912
112: Last Modification:
113: 6.21.97
114: ***********************************types.h************************************/
116: typedef PetscErrorCode (*vfp)(void*,void*,PetscInt,...);
117: typedef PetscErrorCode (*rbfp)(PetscScalar*, PetscScalar*, PetscInt len);
118: typedef PetscInt (*bfp)(void*, void*, PetscInt *len, MPI_Datatype *dt);
120: /***********************************comm.h*************************************
122: Author: Henry M. Tufo III
124: e-mail: hmt@cs.brown.edu
126: snail-mail:
127: Division of Applied Mathematics
128: Brown University
129: Providence, RI 02912
131: Last Modification:
132: 6.21.97
133: ***********************************comm.h*************************************/
134: PETSC_INTERN PetscMPIInt PCTFS_my_id;
135: PETSC_INTERN PetscMPIInt PCTFS_num_nodes;
136: PETSC_INTERN PetscMPIInt PCTFS_floor_num_nodes;
137: PETSC_INTERN PetscMPIInt PCTFS_i_log2_num_nodes;
139: PETSC_INTERN PetscErrorCode PCTFS_giop(PetscInt *vals, PetscInt *work, PetscInt n, PetscInt *oprs);
140: PETSC_INTERN PetscErrorCode PCTFS_grop(PetscScalar *vals, PetscScalar *work, PetscInt n, PetscInt *oprs);
141: PETSC_INTERN PetscErrorCode PCTFS_comm_init(void);
142: PETSC_INTERN PetscErrorCode PCTFS_giop_hc(PetscInt *vals, PetscInt *work, PetscInt n, PetscInt *oprs, PetscInt dim);
143: PETSC_INTERN PetscErrorCode PCTFS_grop_hc(PetscScalar *vals, PetscScalar *work, PetscInt n, PetscInt *oprs, PetscInt dim);
144: PETSC_INTERN PetscErrorCode PCTFS_ssgl_radd(PetscScalar *vals, PetscScalar *work, PetscInt level, PetscInt *segs);
146: #define MSGTAG0 101
147: #define MSGTAG1 1001
148: #define MSGTAG2 76207
149: #define MSGTAG3 100001
150: #define MSGTAG4 163841
151: #define MSGTAG5 249439
152: #define MSGTAG6 10000001
154: #define NON_UNIFORM 0
155: #define GL_MAX 1
156: #define GL_MIN 2
157: #define GL_MULT 3
158: #define GL_ADD 4
159: #define GL_B_XOR 5
160: #define GL_B_OR 6
161: #define GL_B_AND 7
162: #define GL_L_XOR 8
163: #define GL_L_OR 9
164: #define GL_L_AND 10
165: #define GL_MAX_ABS 11
166: #define GL_MIN_ABS 12
167: #define GL_EXISTS 13
169: PETSC_INTERN PetscInt *PCTFS_ivec_copy(PetscInt *arg1, PetscInt *arg2, PetscInt n);
171: PETSC_INTERN PetscErrorCode PCTFS_ivec_zero(PetscInt *arg1, PetscInt n);
172: PETSC_INTERN PetscErrorCode PCTFS_ivec_set(PetscInt *arg1, PetscInt arg2, PetscInt n);
174: PETSC_INTERN PetscInt PCTFS_ivec_lb(PetscInt *work, PetscInt n);
175: PETSC_INTERN PetscInt PCTFS_ivec_ub(PetscInt *work, PetscInt n);
176: PETSC_INTERN PetscInt PCTFS_ivec_sum(PetscInt *arg1, PetscInt n);
178: PETSC_INTERN vfp PCTFS_ivec_fct_addr(PetscInt type);
180: PETSC_INTERN PetscErrorCode PCTFS_ivec_non_uniform(PetscInt *arg1, PetscInt *arg2, PetscInt n, PetscInt *arg3);
181: PETSC_INTERN PetscErrorCode PCTFS_ivec_max(PetscInt *arg1, PetscInt *arg2, PetscInt n);
182: PETSC_INTERN PetscErrorCode PCTFS_ivec_min(PetscInt *arg1, PetscInt *arg2, PetscInt n);
183: PETSC_INTERN PetscErrorCode PCTFS_ivec_mult(PetscInt *arg1, PetscInt *arg2, PetscInt n);
184: PETSC_INTERN PetscErrorCode PCTFS_ivec_add(PetscInt *arg1, PetscInt *arg2, PetscInt n);
185: PETSC_INTERN PetscErrorCode PCTFS_ivec_xor(PetscInt *arg1, PetscInt *arg2, PetscInt n);
186: PETSC_INTERN PetscErrorCode PCTFS_ivec_or(PetscInt *arg1, PetscInt *arg2, PetscInt len);
187: PETSC_INTERN PetscErrorCode PCTFS_ivec_and(PetscInt *arg1, PetscInt *arg2, PetscInt len);
188: PETSC_INTERN PetscErrorCode PCTFS_ivec_lxor(PetscInt *arg1, PetscInt *arg2, PetscInt n);
189: PETSC_INTERN PetscErrorCode PCTFS_ivec_lor(PetscInt *arg1, PetscInt *arg2, PetscInt len);
190: PETSC_INTERN PetscErrorCode PCTFS_ivec_land(PetscInt *arg1, PetscInt *arg2, PetscInt len);
191: PETSC_INTERN PetscErrorCode PCTFS_ivec_and3(PetscInt *arg1, PetscInt *arg2, PetscInt *arg3, PetscInt n);
193: PETSC_INTERN PetscErrorCode PCTFS_ivec_sort_companion(PetscInt *ar, PetscInt *ar2, PetscInt size);
194: PETSC_INTERN PetscErrorCode PCTFS_ivec_sort(PetscInt *ar, PetscInt size);
195: PETSC_INTERN PetscErrorCode PCTFS_SMI_sort(void *ar1, void *ar2, PetscInt size, PetscInt type);
196: PETSC_INTERN PetscInt PCTFS_ivec_binary_search(PetscInt item, PetscInt *list, PetscInt n);
197: PETSC_INTERN PetscInt PCTFS_ivec_linear_search(PetscInt item, PetscInt *list, PetscInt n);
199: PETSC_INTERN PetscErrorCode PCTFS_ivec_sort_companion_hack(PetscInt *ar, PetscInt **ar2, PetscInt size);
201: #define SORT_INTEGER 1
202: #define SORT_INT_PTR 2
204: PETSC_INTERN PetscErrorCode PCTFS_rvec_zero(PetscScalar *arg1, PetscInt n);
205: PETSC_INTERN PetscErrorCode PCTFS_rvec_one(PetscScalar *arg1, PetscInt n);
206: PETSC_INTERN PetscErrorCode PCTFS_rvec_set(PetscScalar *arg1, PetscScalar arg2, PetscInt n);
207: PETSC_INTERN PetscErrorCode PCTFS_rvec_copy(PetscScalar *arg1, PetscScalar *arg2, PetscInt n);
208: PETSC_INTERN PetscErrorCode PCTFS_rvec_scale(PetscScalar *arg1, PetscScalar arg2, PetscInt n);
210: PETSC_INTERN vfp PCTFS_rvec_fct_addr(PetscInt type);
211: PETSC_INTERN PetscErrorCode PCTFS_rvec_add(PetscScalar *arg1, PetscScalar *arg2, PetscInt n);
212: PETSC_INTERN PetscErrorCode PCTFS_rvec_mult(PetscScalar *arg1, PetscScalar *arg2, PetscInt n);
213: PETSC_INTERN PetscErrorCode PCTFS_rvec_max(PetscScalar *arg1, PetscScalar *arg2, PetscInt n);
214: PETSC_INTERN PetscErrorCode PCTFS_rvec_max_abs(PetscScalar *arg1, PetscScalar *arg2, PetscInt n);
215: PETSC_INTERN PetscErrorCode PCTFS_rvec_min(PetscScalar *arg1, PetscScalar *arg2, PetscInt n);
216: PETSC_INTERN PetscErrorCode PCTFS_rvec_min_abs(PetscScalar *arg1, PetscScalar *arg2, PetscInt n);
217: PETSC_INTERN PetscErrorCode PCTFS_vec_exists(PetscScalar *arg1, PetscScalar *arg2, PetscInt n);
219: /***********************************gs.h***************************************
221: Author: Henry M. Tufo III
223: e-mail: hmt@cs.brown.edu
225: snail-mail:
226: Division of Applied Mathematics
227: Brown University
228: Providence, RI 02912
230: Last Modification:
231: 6.21.97
232: ************************************gs.h**************************************/
234: typedef struct gather_scatter_id *PCTFS_gs_ADT;
235: /*typedef PetscErrorCode (*Rbfp)(PetscScalar *, PetscScalar *, PetscInt len);*/
237: PETSC_INTERN PCTFS_gs_ADT PCTFS_gs_init(PetscInt *elms, PetscInt nel, PetscInt level);
238: PETSC_INTERN PetscErrorCode PCTFS_gs_gop_vec(PCTFS_gs_ADT PCTFS_gs_handle, PetscScalar *vals, const char *op, PetscInt step);
239: /* PETSC_INTERN PetscErrorCode PCTFS_gs_gop_binary(PCTFS_gs_ADT gs, PetscScalar *vals, Rbfp fct); */
240: PETSC_INTERN PetscErrorCode PCTFS_gs_gop_hc(PCTFS_gs_ADT PCTFS_gs_handle, PetscScalar *vals, const char *op, PetscInt dim);
241: PETSC_INTERN PetscErrorCode PCTFS_gs_free(PCTFS_gs_ADT PCTFS_gs_handle);
242: PETSC_INTERN PetscErrorCode PCTFS_gs_init_msg_buf_sz(PetscInt buf_size);
243: PETSC_INTERN PetscErrorCode PCTFS_gs_init_vec_sz(PetscInt size);
245: /*************************************xxt.h************************************
246: Module Name: xxt
247: Module Info: need xxt.{c,h} gs.{c,h} comm.{c,h} ivec.{c,h} error.{c,h}
249: author: Henry M. Tufo III
250: e-mail: hmt@asci.uchicago.edu
251: contact:
252: +--------------------------------+--------------------------------+
253: |MCS Division - Building 221 |Department of Computer Science |
254: |Argonne National Laboratory |Ryerson 152 |
255: |9700 S. Cass Avenue |The University of Chicago |
256: |Argonne, IL 60439 |Chicago, IL 60637 |
257: |(630) 252-5354/5986 ph/fx |(773) 702-6019/8487 ph/fx |
258: +--------------------------------+--------------------------------+
260: Last Modification: 3.20.01
261: **************************************xxt.h***********************************/
263: typedef struct xxt_CDT *xxt_ADT;
266: /*************************************xxt.h************************************
267: Function: XXT_new()
269: Return: ADT ptr or NULL upon failure.
270: Description: This function allocates and returns an xxt handle
271: Usage: xxt_handle = xxt_new();
272: **************************************xxt.h***********************************/
273: PETSC_INTERN xxt_ADT XXT_new(void);
276: /*************************************xxt.h************************************
277: Function: XXT_free()
279: Input : pointer to ADT.
281: Description: This function frees the storage associated with an xxt handle
282: Usage: XXT_free(xxt_handle);
283: **************************************xxt.h***********************************/
284: PETSC_INTERN PetscInt XXT_free(xxt_ADT xxt_handle);
287: /*************************************xxt.h************************************
288: Function: XXT_factor
290: Input : ADT ptr, and pointer to object
291: Return: 0 on failure, 1 on success
292: Description: This function sets the xxt solver
294: xxt assumptions: given n rows of global coarse matrix (E_loc) where
295: o global dofs N = sum_p(n), p=0,P-1
296: (i.e. row dist. with no dof replication)
297: (5.21.00 will handle dif replication case)
298: o m is the number of columns in E_loc (m>=n)
299: o local2global holds global number of column i (i=0,...,m-1)
300: o local2global holds global number of row i (i=0,...,n-1)
301: o mylocmatvec performs E_loc . x_loc where x_loc is an vector of
302: length m in 1-1 correspondence with local2global
303: (note that gs package takes care of communication).
304: (note do not zero out upper m-n entries!)
305: o mylocmatvec(void *grid_data, double *in, double *out)
307: ML beliefs/usage: move this to to ML_XXT_factor routine
308: o my_ml holds address of ML struct associated w/E_loc, grid_data, grid_tag
309: o grid_tag, grid_data, my_ml used in
310: ML_Set_CSolve(my_ml, grid_tag, grid_data, ML_Do_CoarseDirect);
311: o grid_data used in
312: A_matvec(grid_data,v,u);
314: Usage:
315: **************************************xxt.h***********************************/
316: PETSC_INTERN PetscInt XXT_factor(xxt_ADT xxt_handle, /* prev. allocated xxt handle */
317: PetscInt *local2global, /* global column mapping */
318: PetscInt n, /* local num rows */
319: PetscInt m, /* local num cols */
320: PetscErrorCode (*matvec)(void*,PetscScalar*,PetscScalar*), /* b_loc=A_local.x_loc */
321: void *grid_data); /* grid data for matvec */
324: /*************************************xxt.h************************************
325: Function: XXT_solve
327: Input : ADT ptr, b (rhs)
328: Output: x (soln)
329: Return:
330: Description: This function performs x = E^-1.b
331: Usage:
332: XXT_solve(xxt_handle, double *x, double *b)
333: XXT_solve(xxt_handle, double *x, NULL)
334: assumes x has been initialized to be b
335: **************************************xxt.h***********************************/
336: PETSC_INTERN PetscInt XXT_solve(xxt_ADT xxt_handle, PetscScalar *x, PetscScalar *b);
338: /*************************************xxt.h************************************
339: Function: XXT_stats
341: Input : handle
342: **************************************xxt.h***********************************/
343: PETSC_INTERN PetscInt XXT_stats(xxt_ADT xxt_handle);
346: /*************************************xxt.h************************************
347: Function: XXT_sp_1()
349: Input : pointer to ADT
350: Output:
351: Return:
352: Description: sets xxt parameter 1 in xxt_handle
353: Usage: implement later
355: void XXT_sp_1(xxt_handle,parameter 1 value)
356: **************************************xxt.h***********************************/
359: /*************************************xyt.h************************************
360: Module Name: xyt
361: Module Info: need xyt.{c,h} gs.{c,h} comm.{c,h} ivec.{c,h} error.{c,h}
363: author: Henry M. Tufo III
364: e-mail: hmt@asci.uchicago.edu
365: contact:
366: +--------------------------------+--------------------------------+
367: |MCS Division - Building 221 |Department of Computer Science |
368: |Argonne National Laboratory |Ryerson 152 |
369: |9700 S. Cass Avenue |The University of Chicago |
370: |Argonne, IL 60439 |Chicago, IL 60637 |
371: |(630) 252-5354/5986 ph/fx |(773) 702-6019/8487 ph/fx |
372: +--------------------------------+--------------------------------+
374: Last Modification: 3.20.01
375: **************************************xyt.h***********************************/
377: typedef struct xyt_CDT *xyt_ADT;
380: /*************************************xyt.h************************************
381: Function: XYT_new()
383: Return: ADT ptr or NULL upon failure.
384: Description: This function allocates and returns an xyt handle
385: Usage: xyt_handle = xyt_new();
386: **************************************xyt.h***********************************/
387: PETSC_INTERN xyt_ADT XYT_new(void);
390: /*************************************xyt.h************************************
391: Function: XYT_free()
393: Input : pointer to ADT.
394: Description: This function frees the storage associated with an xyt handle
395: Usage: XYT_free(xyt_handle);
396: **************************************xyt.h***********************************/
397: PETSC_INTERN PetscInt XYT_free(xyt_ADT xyt_handle);
400: /*************************************xyt.h************************************
401: Function: XYT_factor
403: Input : ADT ptr, and pointer to object
404: Output:
405: Return: 0 on failure, 1 on success
406: Description: This function sets the xyt solver
408: xyt assumptions: given n rows of global coarse matrix (E_loc) where
409: o global dofs N = sum_p(n), p=0,P-1
410: (i.e. row dist. with no dof replication)
411: (5.21.00 will handle dif replication case)
412: o m is the number of columns in E_loc (m>=n)
413: o local2global holds global number of column i (i=0,...,m-1)
414: o local2global holds global number of row i (i=0,...,n-1)
415: o mylocmatvec performs E_loc . x_loc where x_loc is an vector of
416: length m in 1-1 correspondence with local2global
417: (note that gs package takes care of communication).
418: (note do not zero out upper m-n entries!)
419: o mylocmatvec(void *grid_data, double *in, double *out)
421: ML beliefs/usage: move this to to ML_XYT_factor routine
422: o my_ml holds address of ML struct associated w/E_loc, grid_data, grid_tag
423: o grid_tag, grid_data, my_ml used in
424: ML_Set_CSolve(my_ml, grid_tag, grid_data, ML_Do_CoarseDirect);
425: o grid_data used in
426: A_matvec(grid_data,v,u);
428: Usage:
429: **************************************xyt.h***********************************/
430: PETSC_INTERN PetscInt XYT_factor(xyt_ADT xyt_handle, /* prev. allocated xyt handle */
431: PetscInt *local2global, /* global column mapping */
432: PetscInt n, /* local num rows */
433: PetscInt m, /* local num cols */
434: PetscErrorCode (*matvec)(void*,PetscScalar*,PetscScalar*), /* b_loc=A_local.x_loc */
435: void *grid_data); /* grid data for matvec */
438: /*************************************xyt.h************************************
439: Function: XYT_solve
441: Input : ADT ptr, b (rhs)
442: Output: x (soln)
443: Return:
444: Description: This function performs x = E^-1.b
445: Usage: XYT_solve(xyt_handle, double *x, double *b)
446: **************************************xyt.h***********************************/
447: PETSC_INTERN PetscInt XYT_solve(xyt_ADT xyt_handle, PetscScalar *x, PetscScalar *b);
450: /*************************************xyt.h************************************
451: Function: XYT_stats
453: Input : handle
454: **************************************xyt.h***********************************/
455: PETSC_INTERN PetscInt XYT_stats(xyt_ADT xyt_handle);
458: /********************************bit_mask.h************************************
460: Author: Henry M. Tufo III
462: e-mail: hmt@cs.brown.edu
464: snail-mail:
465: Division of Applied Mathematics
466: Brown University
467: Providence, RI 02912
469: Last Modification:
470: 11.21.97
471: *********************************bit_mask.h***********************************/
472: PETSC_INTERN PetscInt PCTFS_div_ceil(PetscInt numin, PetscInt denom);
473: PETSC_INTERN PetscErrorCode PCTFS_set_bit_mask(PetscInt *bm, PetscInt len, PetscInt val);
474: PETSC_INTERN PetscInt PCTFS_len_bit_mask(PetscInt num_items);
475: PETSC_INTERN PetscInt PCTFS_ct_bits(char *ptr, PetscInt n);
476: PETSC_INTERN PetscErrorCode PCTFS_bm_to_proc(char *ptr, PetscInt p_mask, PetscInt *msg_list);
477: PETSC_INTERN PetscInt PCTFS_len_buf(PetscInt item_size, PetscInt num_items);
479: #endif