Lines Matching refs:dctx
16 PetscDeviceContext dctx; in main() local
32 PetscCall(PetscDeviceContextGetCurrentContext(&dctx)); in main()
36 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_HOST, n2, &xh)); in main()
37 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_HOST, n2, &yh)); in main()
38 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_DEVICE, n2, &xd)); in main()
39 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_DEVICE, n2, &yd)); in main()
45 PetscCall(PetscDeviceMemcpy(dctx, xd, xh, sizeof(PetscInt) * n2)); in main()
46 PetscCall(PetscDeviceMemcpy(dctx, yd, yh, sizeof(PetscInt) * n2)); in main()
50 PetscCall(PetscDeviceMemcpy(dctx, yh, yd, sizeof(PetscInt) * n)); in main()
51 PetscCall(PetscDeviceContextSynchronize(dctx)); // finish the async memcpy in main()
56 PetscCall(PetscDeviceMemcpy(dctx, xh, xd, sizeof(PetscInt) * n2)); in main()
57 PetscCall(PetscDeviceContextSynchronize(dctx)); // finish the async memcpy in main()
60 PetscCall(PetscDeviceFree(dctx, xh)); in main()
61 PetscCall(PetscDeviceFree(dctx, yh)); in main()
62 PetscCall(PetscDeviceFree(dctx, xd)); in main()
63 PetscCall(PetscDeviceFree(dctx, yd)); in main()