xref: /petsc/src/sys/objects/device/tests/ex2.c (revision 327415f76d85372a4417cf1aaa14db707d4d6c04)
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*327415f7SBarry Smith   PetscFunctionBeginUser;
119566063dSJacob Faibussowitsch   PetscCall(PetscInitialize(&argc,&argv,NULL,help));
12a4af0ceeSJacob Faibussowitsch 
13a4af0ceeSJacob Faibussowitsch   /* basic creation and destruction */
149566063dSJacob Faibussowitsch   PetscCall(PetscDeviceContextCreate(&dctx));
159566063dSJacob Faibussowitsch   PetscCall(AssertDeviceContextExists(dctx));
169566063dSJacob Faibussowitsch   PetscCall(PetscDeviceContextDestroy(&dctx));
179566063dSJacob Faibussowitsch   PetscCall(AssertDeviceContextDoesNotExist(dctx));
18a4af0ceeSJacob Faibussowitsch   /* double free is no-op */
199566063dSJacob Faibussowitsch   PetscCall(PetscDeviceContextDestroy(&dctx));
209566063dSJacob Faibussowitsch   PetscCall(AssertDeviceContextDoesNotExist(dctx));
21a4af0ceeSJacob Faibussowitsch 
22a4af0ceeSJacob Faibussowitsch   /* test global context returns a valid context */
23a4af0ceeSJacob Faibussowitsch   dctx = NULL;
249566063dSJacob Faibussowitsch   PetscCall(PetscDeviceContextGetCurrentContext(&dctx));
259566063dSJacob Faibussowitsch   PetscCall(AssertDeviceContextExists(dctx));
26a4af0ceeSJacob Faibussowitsch   /* test locally setting to null doesn't clobber the global */
27a4af0ceeSJacob Faibussowitsch   dctx = NULL;
289566063dSJacob Faibussowitsch   PetscCall(PetscDeviceContextGetCurrentContext(&dctx));
299566063dSJacob Faibussowitsch   PetscCall(AssertDeviceContextExists(dctx));
30a4af0ceeSJacob Faibussowitsch 
31a4af0ceeSJacob Faibussowitsch   /* test duplicate */
329566063dSJacob Faibussowitsch   PetscCall(PetscDeviceContextDuplicate(dctx,&ddup));
33a4af0ceeSJacob Faibussowitsch   /* both device contexts should exist */
349566063dSJacob Faibussowitsch   PetscCall(AssertDeviceContextExists(dctx));
359566063dSJacob Faibussowitsch   PetscCall(AssertDeviceContextExists(ddup));
36a4af0ceeSJacob Faibussowitsch 
37a4af0ceeSJacob Faibussowitsch   /* destroying the dup should leave the original untouched */
389566063dSJacob Faibussowitsch   PetscCall(PetscDeviceContextDestroy(&ddup));
399566063dSJacob Faibussowitsch   PetscCall(AssertDeviceContextDoesNotExist(ddup));
409566063dSJacob Faibussowitsch   PetscCall(AssertDeviceContextExists(dctx));
41a4af0ceeSJacob Faibussowitsch 
429566063dSJacob Faibussowitsch   PetscCall(PetscPrintf(PETSC_COMM_WORLD,"EXIT_SUCCESS\n"));
439566063dSJacob Faibussowitsch   PetscCall(PetscFinalize());
44b122ec5aSJacob Faibussowitsch   return 0;
45a4af0ceeSJacob Faibussowitsch }
46a4af0ceeSJacob Faibussowitsch 
47a4af0ceeSJacob Faibussowitsch /*TEST
48a4af0ceeSJacob Faibussowitsch 
49a4af0ceeSJacob Faibussowitsch  build:
50cb9b7bb0SJacob Faibussowitsch    requires: defined(PETSC_HAVE_CXX)
51a4af0ceeSJacob Faibussowitsch 
52a4af0ceeSJacob Faibussowitsch  test:
536ab378d1SJacob Faibussowitsch    TODO: broken in ci
54a4af0ceeSJacob Faibussowitsch    requires: !device
55a4af0ceeSJacob Faibussowitsch    suffix: no_device
56a4af0ceeSJacob 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]+\(\)"
57a4af0ceeSJacob Faibussowitsch 
58a4af0ceeSJacob Faibussowitsch  testset:
59a4af0ceeSJacob Faibussowitsch    output_file: ./output/ExitSuccess.out
60a4af0ceeSJacob Faibussowitsch    nsize: {{1 2 4}}
61a4af0ceeSJacob Faibussowitsch    test:
62a4af0ceeSJacob Faibussowitsch      requires: cuda
63a4af0ceeSJacob Faibussowitsch      suffix: cuda
64a4af0ceeSJacob Faibussowitsch    test:
65a4af0ceeSJacob Faibussowitsch      requires: hip
66a4af0ceeSJacob Faibussowitsch      suffix: hip
67a4af0ceeSJacob Faibussowitsch 
68a4af0ceeSJacob Faibussowitsch TEST*/
69