Date: 2025-03-29T05:58:39+00:00

Actual source code: ex10.c

      1: static char help[] = "Reads a PETSc matrix and computes the 2 norm of the columns\n\n";

      3: /*
      4:   Include "petscmat.h" so that we can use matrices.
      5:   automatically includes:
      6:      petscsys.h       - base PETSc routines   petscvec.h    - vectors
      7:      petscmat.h    - matrices
      8:      petscis.h     - index sets            petscviewer.h - viewers
      9: */
     10: #include <petscmat.h>

     12: int main(int argc, char **args)
     13: {
     14:   Mat               A;                        /* matrix */
     15:   PetscViewer       fd;                       /* viewer */
     16:   char              file[PETSC_MAX_PATH_LEN]; /* input file name */
     17:   PetscReal        *norms;
     18:   PetscInt          n, cstart, cend;
     19:   PetscBool         flg;
     20:   PetscViewerFormat format;

     22:   PetscFunctionBeginUser;
     23:   PetscCall(PetscInitialize(&argc, &args, NULL, help));
     24:   /*
     25:      Determine files from which we read the matrix
     26:   */
     27:   PetscCall(PetscOptionsGetString(NULL, NULL, "-f", file, sizeof(file), &flg));
     28:   PetscCheck(flg, PETSC_COMM_WORLD, PETSC_ERR_USER, "Must indicate binary file with the -f option");

     30:   /*
     31:      Open binary file.  Note that we use FILE_MODE_READ to indicate
     32:      reading from this file.
     33:   */
     34:   PetscCall(PetscViewerCreate(PETSC_COMM_WORLD, &fd));
     35:   PetscCall(PetscViewerSetType(fd, PETSCVIEWERBINARY));
     36:   PetscCall(PetscViewerSetFromOptions(fd));
     37:   PetscCall(PetscOptionsGetEnum(NULL, NULL, "-viewer_format", PetscViewerFormats, (PetscEnum *)&format, &flg));
     38:   if (flg) PetscCall(PetscViewerPushFormat(fd, format));
     39:   PetscCall(PetscViewerFileSetMode(fd, FILE_MODE_READ));
     40:   PetscCall(PetscViewerFileSetName(fd, file));

     42:   /*
     43:     Load the matrix; then destroy the viewer.
     44:     Matrix type is set automatically but you can override it by MatSetType() prior to MatLoad().
     45:     Do that only if you really insist on the given type.
     46:   */
     47:   PetscCall(MatCreate(PETSC_COMM_WORLD, &A));
     48:   PetscCall(MatSetOptionsPrefix(A, "a_"));
     49:   PetscCall(PetscObjectSetName((PetscObject)A, "A"));
     50:   PetscCall(MatSetFromOptions(A));
     51:   PetscCall(MatLoad(A, fd));
     52:   PetscCall(PetscViewerDestroy(&fd));

     54:   PetscCall(MatGetSize(A, NULL, &n));
     55:   PetscCall(MatGetOwnershipRangeColumn(A, &cstart, &cend));
     56:   PetscCall(PetscMalloc1(n, &norms));
     57:   PetscCall(MatGetColumnNorms(A, NORM_2, norms));
     58:   PetscCall(PetscRealView(cend - cstart, norms + cstart, PETSC_VIEWER_STDOUT_WORLD));
     59:   PetscCall(PetscFree(norms));

     61:   PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)A, PETSC_VIEWER_STDOUT_WORLD));
     62:   PetscCall(MatViewFromOptions(A, NULL, "-mat_view"));

     64:   PetscCall(MatDestroy(&A));
     65:   PetscCall(PetscFinalize());
     66:   return 0;
     67: }

     69: /*TEST

     71:    test:
     72:       suffix: mpiaij
     73:       nsize: 2
     74:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES)
     75:       args: -f ${DATAFILESPATH}/matrices/small -a_mat_type mpiaij
     76:       args: -a_matload_symmetric

     78:    test:
     79:       suffix: mpiaij_hdf5
     80:       nsize: 2
     81:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES) hdf5 defined(PETSC_HDF5_HAVE_ZLIB)
     82:       args: -f ${DATAFILESPATH}/matrices/matlab/small.mat -a_mat_type mpiaij -viewer_type hdf5 -viewer_format hdf5_mat
     83:       args: -a_matload_symmetric

     85:    test:
     86:       suffix: mpiaij_rect_hdf5
     87:       nsize: 2
     88:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES) hdf5 defined(PETSC_HDF5_HAVE_ZLIB)
     89:       args: -f ${DATAFILESPATH}/matrices/matlab/small_rect.mat -a_mat_type mpiaij -viewer_type hdf5 -viewer_format hdf5_mat

     91:    test:
     92:       # test for more processes than rows
     93:       suffix: mpiaij_hdf5_tiny
     94:       nsize: 8
     95:       requires: double !complex !defined(PETSC_USE_64BIT_INDICES) hdf5 defined(PETSC_HDF5_HAVE_ZLIB)
     96:       args: -f ${wPETSC_DIR}/share/petsc/datafiles/matrices/tiny_system_with_x0.mat -a_mat_type mpiaij -viewer_type hdf5 -viewer_format hdf5_mat
     97:       args: -a_matload_symmetric

     99:    test:
    100:       # test for more processes than rows, complex
    101:       TODO: not yet implemented for MATLAB complex format
    102:       suffix: mpiaij_hdf5_tiny_complex
    103:       nsize: 8
    104:       requires: double complex !defined(PETSC_USE_64BIT_INDICES) hdf5 defined(PETSC_HDF5_HAVE_ZLIB)
    105:       args: -f ${wPETSC_DIR}/share/petsc/datafiles/matrices/tiny_system_with_x0_complex.mat -a_mat_type mpiaij -viewer_type hdf5 -viewer_format hdf5_mat
    106:       args: -a_matload_symmetric

    108:    test:
    109:       TODO: mpibaij not supported yet
    110:       suffix: mpibaij_hdf5
    111:       nsize: 2
    112:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES) hdf5 defined(PETSC_HDF5_HAVE_ZLIB)
    113:       args: -f ${DATAFILESPATH}/matrices/matlab/small.mat -a_mat_type mpibaij -a_mat_block_size 2 -viewer_type hdf5 -viewer_format hdf5_mat
    114:       args: -a_matload_symmetric

    116:    test:
    117:       suffix: mpidense
    118:       nsize: 2
    119:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES)
    120:       args: -f ${DATAFILESPATH}/matrices/small -a_mat_type mpidense
    121:       args: -a_matload_symmetric

    123:    test:
    124:       suffix: seqaij
    125:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES)
    126:       args: -f ${DATAFILESPATH}/matrices/small -a_mat_type seqaij
    127:       args: -a_matload_symmetric

    129:    test:
    130:       suffix: seqaij_hdf5
    131:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES) hdf5 defined(PETSC_HDF5_HAVE_ZLIB)
    132:       args: -f ${DATAFILESPATH}/matrices/matlab/small.mat -a_mat_type seqaij -viewer_type hdf5 -viewer_format hdf5_mat
    133:       args: -a_matload_symmetric

    135:    test:
    136:       suffix: seqaij_rect_hdf5
    137:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES) hdf5 defined(PETSC_HDF5_HAVE_ZLIB)
    138:       args: -f ${DATAFILESPATH}/matrices/matlab/small_rect.mat -a_mat_type seqaij -viewer_type hdf5 -viewer_format hdf5_mat

    140:    test:
    141:       suffix: seqdense
    142:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES)
    143:       args: -f ${DATAFILESPATH}/matrices/small -a_mat_type seqdense
    144:       args: -a_matload_symmetric

    146:    test:
    147:       suffix: seqdense_hdf5
    148:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES) hdf5 defined(PETSC_HDF5_HAVE_ZLIB)
    149:       args: -f ${DATAFILESPATH}/matrices/matlab/small_dense.mat -a_mat_type seqdense -viewer_type hdf5 -viewer_format hdf5_mat
    150:       args: -a_matload_symmetric

    152:    test:
    153:       suffix: seqdense_rect_hdf5
    154:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES) hdf5 defined(PETSC_HDF5_HAVE_ZLIB)
    155:       args: -f ${DATAFILESPATH}/matrices/matlab/small_rect_dense.mat -a_mat_type seqdense -viewer_type hdf5 -viewer_format hdf5_mat

    157:    test:
    158:       suffix: mpidense_hdf5
    159:       nsize: 2
    160:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES) hdf5 defined(PETSC_HDF5_HAVE_ZLIB)
    161:       args: -f ${DATAFILESPATH}/matrices/matlab/small_dense.mat -a_mat_type mpidense -viewer_type hdf5 -viewer_format hdf5_mat
    162:       args: -a_matload_symmetric

    164:    test:
    165:       suffix: mpidense_rect_hdf5
    166:       nsize: 2
    167:       requires: datafilespath double !complex !defined(PETSC_USE_64BIT_INDICES) hdf5 defined(PETSC_HDF5_HAVE_ZLIB)
    168:       args: -f ${DATAFILESPATH}/matrices/matlab/small_rect_dense.mat -a_mat_type mpidense -viewer_type hdf5 -viewer_format hdf5_mat
    169: TEST*/