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