1: static char help[] = "Test VecSetValuesCOO\n\n";

  3: #include <petscmat.h>
  4: int main(int argc, char **args)
  5: {
  6:   Vec            x, y;
  7:   const PetscInt M = 18;
  8:   PetscMPIInt    rank, size;
  9:   PetscBool      equal;
 10:   PetscScalar   *vals;
 11:   PetscBool      ignoreRemote = PETSC_FALSE;

 13:   PetscInt i0[] = {3, 4, 1, 10, 0, 1, 1, 2, 1, 1, 2, 2, 3, 3, 4, 4, 1, 2, 5, 5, 6, 4, 17, 0, 1, 1, 8, 5, 5, 6, 4, 7, 8, 5};
 14:   PetscInt i1[] = {8, 5, 15, 16, 6, 13, 4, 17, 8, 9, 9, 10, 6, 12, 7, 3, 4, 1, 1, 2, 5, 5, 6, 14, 17, 8, 9, 9, 10, 4, 5, 10, 11, 1, 2};
 15:   PetscInt i2[] = {7, 7, 8, 8, 9, 16, 17, 9, 10, 1, 1, -2, 2, 3, 3, 14, 4, 5, 10, 13, 9, 9, 10, 1, 0, 0, 5, 5, 6, 6, 13, 13, 14, -14, 4, 4, 5, 11, 11, 12, 15, 15, 16};

 17:   struct {
 18:     PetscInt  *i;
 19:     PetscCount n;
 20:   } coo[3] = {
 21:     {i0, PETSC_STATIC_ARRAY_LENGTH(i0)},
 22:     {i1, PETSC_STATIC_ARRAY_LENGTH(i1)},
 23:     {i2, PETSC_STATIC_ARRAY_LENGTH(i2)}
 24:   };

 26:   PetscFunctionBeginUser;
 27:   PetscCall(PetscInitialize(&argc, &args, NULL, help));
 28:   PetscCall(PetscOptionsGetBool(NULL, NULL, "-ignore_remote", &ignoreRemote, NULL));
 29:   PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
 30:   PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size));

 32:   PetscCheck(size <= 3, PETSC_COMM_WORLD, PETSC_ERR_WRONG_MPI_SIZE, "This test requires at most 3 processes");

 34:   PetscCall(VecCreate(PETSC_COMM_WORLD, &x));
 35:   PetscCall(VecSetSizes(x, PETSC_DECIDE, M));
 36:   PetscCall(VecSetType(x, VECSTANDARD));
 37:   PetscCall(VecSetOption(x, VEC_IGNORE_OFF_PROC_ENTRIES, ignoreRemote));
 38:   PetscCall(VecSetOption(x, VEC_IGNORE_NEGATIVE_INDICES, PETSC_TRUE));

 40:   for (PetscInt k = 0; k < coo[rank].n; k++) {
 41:     PetscScalar val = (PetscScalar)coo[rank].i[k];
 42:     PetscCall(VecSetValues(x, 1, &coo[rank].i[k], &val, ADD_VALUES));
 43:   }
 44:   PetscCall(VecAssemblyBegin(x));
 45:   PetscCall(VecAssemblyEnd(x));

 47:   PetscCall(VecCreate(PETSC_COMM_WORLD, &y));
 48:   PetscCall(VecSetSizes(y, PETSC_DECIDE, M));
 49:   PetscCall(VecSetFromOptions(y));
 50:   PetscCall(VecSetOption(y, VEC_IGNORE_OFF_PROC_ENTRIES, ignoreRemote));
 51:   PetscCall(VecSetOption(y, VEC_IGNORE_NEGATIVE_INDICES, PETSC_TRUE));
 52:   PetscCall(VecSetPreallocationCOO(y, coo[rank].n, coo[rank].i));

 54:   PetscCall(PetscMalloc1(coo[rank].n, &vals));
 55:   for (PetscInt k = 0; k < coo[rank].n; k++) vals[k] = (PetscScalar)coo[rank].i[k];
 56:   PetscCall(VecSetValuesCOO(y, vals, ADD_VALUES));

 58:   PetscCall(VecEqual(x, y, &equal));

 60:   if (!equal) {
 61:     PetscCall(VecView(x, PETSC_VIEWER_STDOUT_WORLD));
 62:     PetscCall(VecView(y, PETSC_VIEWER_STDOUT_WORLD));
 63:     SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_PLIB, "VecSetValuesCOO() failed");
 64:   }

 66:   PetscCall(PetscFree(vals));
 67:   PetscCall(VecDestroy(&x));
 68:   PetscCall(VecDestroy(&y));
 69:   PetscCall(PetscFinalize());
 70:   return 0;
 71: }

 73: /*TEST

 75:   testset:
 76:     output_file: output/empty.out
 77:     nsize: {{1 2 3}}
 78:     args: -ignore_remote {{0 1}}

 80:     test:
 81:       suffix: kokkos
 82:       requires: kokkos_kernels
 83:       args: -vec_type kokkos

 85:     test:
 86:       suffix: cuda
 87:       requires: cuda
 88:       args: -vec_type cuda

 90:     test:
 91:       suffix: std
 92:       args: -vec_type standard

 94: TEST*/