xref: /petsc/src/mat/tests/ex239.c (revision 609caa7c8c030312b00807b4f015fd827bb80932)

static char help[] = "Test device/host memory allocation in MatDenseSeqCUDA()\n\n";

/* Contributed by: Victor Eijkhout <eijkhout@tacc.utexas.edu> */

#include <petscmat.h>
int main(int argc, char **argv)
{
  PetscInt  global_size = 100;
  Mat       cuda_matrix;
  Vec       input, output;
  MPI_Comm  comm = PETSC_COMM_SELF;
  PetscReal nrm  = 1;

  PetscFunctionBeginUser;
  PetscCall(PetscInitialize(&argc, &argv, NULL, help));
  PetscCall(MatCreateDenseCUDA(comm, global_size, global_size, global_size, global_size, NULL, &cuda_matrix));

  PetscCall(VecCreateSeqCUDA(comm, global_size, &input));
  PetscCall(VecDuplicate(input, &output));
  PetscCall(VecSet(input, 1.));
  PetscCall(VecSet(output, 2.));
  PetscCall(MatMult(cuda_matrix, input, output));
  PetscCall(VecNorm(output, NORM_2, &nrm));
  PetscCheck(nrm <= PETSC_SMALL, PETSC_COMM_SELF, PETSC_ERR_PLIB, "PETSc generated wrong result. Should be 0, but is %g", (double)nrm);
  PetscCall(VecDestroy(&input));
  PetscCall(VecDestroy(&output));
  PetscCall(MatDestroy(&cuda_matrix));
  PetscCall(PetscFinalize());
  return 0;
}

/*TEST
   build:
     requires: cuda

   test:
     nsize: 1
     output_file: output/empty.out

TEST*/