xref: /petsc/src/sys/objects/device/tests/ex2.c (revision 9dc7b89c67bc73276ac9486e039061dd02e08fe5)
1 static const char help[] = "Tests creation and destruction of PetscDeviceContext.\n\n";
2 
3 #include <petsc/private/deviceimpl.h>
4 #include "petscdevicetestcommon.h"
5 
6 int main(int argc, char *argv[])
7 {
8   PetscDeviceContext dctx = NULL,ddup = NULL;
9   PetscErrorCode     ierr;
10 
11   ierr = PetscInitialize(&argc,&argv,NULL,help);if (ierr) return ierr;
12 
13   /* basic creation and destruction */
14   ierr = PetscDeviceContextCreate(&dctx);CHKERRQ(ierr);
15   ierr = AssertDeviceContextExists(dctx);CHKERRQ(ierr);
16   ierr = PetscDeviceContextDestroy(&dctx);CHKERRQ(ierr);
17   ierr = AssertDeviceContextDoesNotExist(dctx);CHKERRQ(ierr);
18   /* double free is no-op */
19   ierr = PetscDeviceContextDestroy(&dctx);CHKERRQ(ierr);
20   ierr = AssertDeviceContextDoesNotExist(dctx);CHKERRQ(ierr);
21 
22   /* test global context returns a valid context */
23   dctx = NULL;
24   ierr = PetscDeviceContextGetCurrentContext(&dctx);CHKERRQ(ierr);
25   ierr = AssertDeviceContextExists(dctx);CHKERRQ(ierr);
26   /* test locally setting to null doesn't clobber the global */
27   dctx = NULL;
28   ierr = PetscDeviceContextGetCurrentContext(&dctx);CHKERRQ(ierr);
29   ierr = AssertDeviceContextExists(dctx);CHKERRQ(ierr);
30 
31   /* test duplicate */
32   ierr = PetscDeviceContextDuplicate(dctx,&ddup);CHKERRQ(ierr);
33   /* both device contexts should exist */
34   ierr = AssertDeviceContextExists(dctx);CHKERRQ(ierr);
35   ierr = AssertDeviceContextExists(ddup);CHKERRQ(ierr);
36 
37   /* destroying the dup should leave the original untouched */
38   ierr = PetscDeviceContextDestroy(&ddup);CHKERRQ(ierr);
39   ierr = AssertDeviceContextDoesNotExist(ddup);CHKERRQ(ierr);
40   ierr = AssertDeviceContextExists(dctx);CHKERRQ(ierr);
41 
42   ierr = PetscPrintf(PETSC_COMM_WORLD,"EXIT_SUCCESS\n");CHKERRQ(ierr);
43   ierr = PetscFinalize();
44   return ierr;
45 }
46 
47 /*TEST
48 
49  build:
50    requires: defined(PETSC_HAVE_CXX)
51 
52  test:
53    TODO: broken in ci
54    requires: !device
55    suffix: no_device
56    filter: Error: grep -E -o -e ".*No support for this operation for this object type" -e ".*PETSc is not configured with device support.*" -e "^\[0\]PETSC ERROR:.*[0-9]{1} [A-z]+\(\)"
57 
58  testset:
59    output_file: ./output/ExitSuccess.out
60    nsize: {{1 2 4}}
61    test:
62      requires: cuda
63      suffix: cuda
64    test:
65      requires: hip
66      suffix: hip
67 
68 TEST*/
69