xref: /petsc/src/sys/objects/device/tests/ex2.c (revision 84ff8c8b54fd7c9cb88641c01dfe6357ec5f72d0)
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_DIALECT_CXX11)
51 
52  test:
53    requires: !device
54    suffix: no_device
55    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]+\(\)"
56 
57  testset:
58    output_file: ./output/ExitSuccess.out
59    nsize: {{1 2 4}}
60    test:
61      requires: cuda
62      suffix: cuda
63    test:
64      requires: hip
65      suffix: hip
66 
67 TEST*/
68