Date: 2013-07-02T14:47:02+00:00

Actual source code: tfs.h
petsc-3.4.2 2013-07-02

      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