| /petsc/include/ |
| H A D | petscdevice.h | 48 PETSC_EXTERN PetscErrorCode PetscDeviceContextCreate(PetscDeviceContext *); 49 PETSC_EXTERN PetscErrorCode PetscDeviceContextDestroy(PetscDeviceContext *); 50 PETSC_EXTERN PetscErrorCode PetscDeviceContextSetStreamType(PetscDeviceContext, PetscStreamType); 51 PETSC_EXTERN PetscErrorCode PetscDeviceContextGetStreamType(PetscDeviceContext, PetscStreamType *); 52 PETSC_EXTERN PetscErrorCode PetscDeviceContextSetDevice(PetscDeviceContext, PetscDevice); 53 PETSC_EXTERN PetscErrorCode PetscDeviceContextGetDevice(PetscDeviceContext, PetscDevice *); 54 PETSC_EXTERN PetscErrorCode PetscDeviceContextGetDeviceType(PetscDeviceContext, PetscDeviceType *); 55 PETSC_EXTERN PetscErrorCode PetscDeviceContextSetUp(PetscDeviceContext); 56 PETSC_EXTERN PetscErrorCode PetscDeviceContextDuplicate(PetscDeviceContext, PetscDeviceContext *); 57 PETSC_EXTERN PetscErrorCode PetscDeviceContextQueryIdle(PetscDeviceContext, PetscBool *); [all …]
|
| /petsc/src/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | petscdevice.pxi | 46 PetscErrorCode PetscDeviceContextCreate(PetscDeviceContext *) 47 PetscErrorCode PetscDeviceContextDestroy(PetscDeviceContext *) 48 PetscErrorCode PetscDeviceContextSetStreamType(PetscDeviceContext, PetscStreamType) 49 PetscErrorCode PetscDeviceContextGetStreamType(PetscDeviceContext, PetscStreamType *) 50 PetscErrorCode PetscDeviceContextSetDevice(PetscDeviceContext, PetscDevice) 51 PetscErrorCode PetscDeviceContextGetDevice(PetscDeviceContext, PetscDevice *) 52 PetscErrorCode PetscDeviceContextGetDeviceType(PetscDeviceContext, PetscDeviceType *) 53 PetscErrorCode PetscDeviceContextSetUp(PetscDeviceContext) 54 PetscErrorCode PetscDeviceContextDuplicate(PetscDeviceContext, PetscDeviceContext *) 55 PetscErrorCode PetscDeviceContextQueryIdle(PetscDeviceContext, PetscBool *) [all …]
|
| /petsc/src/sys/objects/device/c/ |
| H A D | device.c | 74 PetscErrorCode PetscDeviceContextCreate(PetscDeviceContext *dctx) in PetscDeviceContextCreate() 82 PetscErrorCode PetscDeviceContextDestroy(PetscDeviceContext *dctx) in PetscDeviceContextDestroy() 90 PetscErrorCode PetscDeviceContextSetStreamType(PETSC_UNUSED PetscDeviceContext dctx, PETSC_UNUSED P… in PetscDeviceContextSetStreamType() 96 PetscErrorCode PetscDeviceContextGetStreamType(PETSC_UNUSED PetscDeviceContext dctx, PetscStreamTyp… in PetscDeviceContextGetStreamType() 104 PetscErrorCode PetscDeviceContextSetDevice(PETSC_UNUSED PetscDeviceContext dctx, PETSC_UNUSED Petsc… in PetscDeviceContextSetDevice() 110 PetscErrorCode PetscDeviceContextGetDevice(PETSC_UNUSED PetscDeviceContext dctx, PetscDevice *devic… in PetscDeviceContextGetDevice() 118 PetscErrorCode PetscDeviceContextGetDeviceType(PETSC_UNUSED PetscDeviceContext dctx, PetscDeviceTyp… in PetscDeviceContextGetDeviceType() 126 PetscErrorCode PetscDeviceContextSetUp(PETSC_UNUSED PetscDeviceContext dctx) in PetscDeviceContextSetUp() 132 PetscErrorCode PetscDeviceContextDuplicate(PETSC_UNUSED PetscDeviceContext dctx, PetscDeviceContext… in PetscDeviceContextDuplicate() 140 PetscErrorCode PetscDeviceContextQueryIdle(PETSC_UNUSED PetscDeviceContext dctx, PetscBool *idle) in PetscDeviceContextQueryIdle() [all …]
|
| /petsc/include/petsc/private/ |
| H A D | deviceimpl.h | 122 PetscDeviceContext pvdc_dctx_ = dctx; \ 141 PetscDeviceContext pccdc_dctx1_ = (dctx1), pccdc_dctx2_ = (dctx2); \ 185 PetscErrorCode (*createcontext)(PetscDeviceContext); 212 PetscErrorCode (*destroy)(PetscDeviceContext); 213 PetscErrorCode (*changestreamtype)(PetscDeviceContext, PetscStreamType); 214 PetscErrorCode (*setup)(PetscDeviceContext); 215 PetscErrorCode (*query)(PetscDeviceContext, PetscBool *); 216 PetscErrorCode (*waitforcontext)(PetscDeviceContext, PetscDeviceContext); 217 PetscErrorCode (*synchronize)(PetscDeviceContext); 218 PetscErrorCode (*getblashandle)(PetscDeviceContext, void *); [all …]
|
| H A D | cupmobject.hpp | 139 constexpr explicit RestoreableArray(PetscDeviceContext) noexcept; 142 PetscDeviceContext dctx_ = nullptr; 160 constexpr inline RestoreableArray<T, MT, MA>::RestoreableArray(PetscDeviceContext dctx) noexcept : … in RestoreableArray() 199 …static PetscErrorCode GetFromHandleDispatch_(PetscDeviceContext, cupmBlasHandle_t *, cupmSolverHan… 200 …static PetscErrorCode GetHandleDispatch_(PetscDeviceContext *, cupmBlasHandle_t *, cupmSolverHandl… 209 …static PetscErrorCode GetHandles_(PetscDeviceContext *, cupmBlasHandle_t * = nullptr, cupmSolverHa… 212 …static PetscErrorCode GetHandles_(PetscDeviceContext *, cupmBlasHandle_t *, cupmStream_t *) noexce… 213 …static PetscErrorCode GetHandles_(PetscDeviceContext *, cupmSolverHandle_t *, cupmStream_t *) noex… 216 static PetscErrorCode GetHandles_(PetscDeviceContext *, cupmSolverHandle_t *) noexcept; 217 static PetscErrorCode GetHandles_(PetscDeviceContext *, cupmStream_t *) noexcept; [all …]
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm.hpp | 47 …static PetscErrorCode PointwiseBinary_(BinaryFuncT &&, Vec, Vec, Vec, PetscDeviceContext = nullptr… 49 …(PetscErrorCode (*)(Vec, Vec, Vec), BinaryFuncT &&, Vec, Vec, Vec, PetscDeviceContext = nullptr) n… 51 …static PetscErrorCode PointwiseUnary_(UnaryFuncT &&, Vec, Vec, PetscDeviceContext = nullptr) noexc… 53 …mplex = */ std::true_type, Vec, PetscInt, const Vec[], PetscScalar *, PetscDeviceContext) noexcept; 54 …plex = */ std::false_type, Vec, PetscInt, const Vec[], PetscScalar *, PetscDeviceContext) noexcept; 56 …static PetscErrorCode MDot_kernel_dispatch_(PetscDeviceContext, cupmStream_t, const PetscScalar *,… 58 …static PetscErrorCode MDot_kernel_dispatch_(PetscDeviceContext, cupmStream_t, const PetscScalar *,… 60 …static PetscErrorCode MAXPY_kernel_dispatch_(PetscDeviceContext, cupmStream_t, PetscScalar *, cons… 62 …static PetscErrorCode MAXPY_kernel_dispatch_(PetscDeviceContext, cupmStream_t, PetscScalar *, cons… 64 …static PetscErrorCode CreateSeqCUPM_(Vec, PetscDeviceContext, PetscScalar * /*host_ptr*/ = nullptr… [all …]
|
| /petsc/src/sys/objects/device/impls/cupm/ |
| H A D | cupmcontext.hpp | 81 …PETSC_NODISCARD static constexpr PetscDeviceContext_IMPLS *impls_cast_(PetscDeviceContext ptr) noe… in impls_cast_() 91 …static PetscErrorCode initialize_handle_(stream_tag, PetscDeviceContext) noexcept { return PETSC_S… in initialize_handle_() argument 93 static PetscErrorCode initialize_handle_(blas_tag, PetscDeviceContext dctx) noexcept in initialize_handle_() 120 static PetscErrorCode initialize_handle_(solver_tag, PetscDeviceContext dctx) noexcept in initialize_handle_() 147 …static PetscErrorCode check_current_device_(PetscDeviceContext dctxl, PetscDeviceContext dctxr) no… in check_current_device_() 160 …static PetscErrorCode check_current_device_(PetscDeviceContext dctx) noexcept { return check_curre… in check_current_device_() 198 static PetscErrorCode destroy(PetscDeviceContext) noexcept; 199 static PetscErrorCode changeStreamType(PetscDeviceContext, PetscStreamType) noexcept; 200 static PetscErrorCode setUp(PetscDeviceContext) noexcept; 201 static PetscErrorCode query(PetscDeviceContext, PetscBool *) noexcept; [all …]
|
| /petsc/src/sys/objects/device/impls/sycl/ |
| H A D | syclcontext.sycl.cxx | 41 static PetscErrorCode initialize_(PetscInt id, PetscDeviceContext dctx) noexcept in initialize_() 60 static PetscErrorCode destroy(PetscDeviceContext dctx) noexcept in destroy() 68 static PetscErrorCode setUp(PetscDeviceContext dctx) noexcept in setUp() 85 static PetscErrorCode query(PetscDeviceContext dctx, PetscBool *idle) noexcept in query() 94 static PetscErrorCode synchronize(PetscDeviceContext dctx) noexcept in synchronize() 105 static PetscErrorCode getStreamHandle(PetscDeviceContext dctx, void **handle) noexcept in getStreamHandle() 112 static PetscErrorCode beginTimer(PetscDeviceContext dctx) noexcept in beginTimer() 140 static PetscErrorCode endTimer(PetscDeviceContext dctx, PetscLogDouble *elapsed) noexcept in endTimer() 156 …static PetscErrorCode changeStreamType(PetscDeviceContext, PetscStreamType) noexcept { SETERRQ(PET… in changeStreamType() argument 157 …static PetscErrorCode waitForContext(PetscDeviceContext, PetscDeviceContext) noexcept { SETERRQ(PE… in waitForContext() argument [all …]
|
| /petsc/src/sys/objects/device/impls/host/ |
| H A D | hostcontext.cxx | 19 static PetscErrorCode destroy(PetscDeviceContext) noexcept { return PETSC_SUCCESS; } in destroy() argument 20 …static PetscErrorCode changeStreamType(PetscDeviceContext, PetscStreamType) noexcept { return PETS… in changeStreamType() argument 21 static PetscErrorCode setUp(PetscDeviceContext) noexcept { return PETSC_SUCCESS; } in setUp() argument 22 static PetscErrorCode query(PetscDeviceContext, PetscBool *idle) noexcept in query() argument 28 …static PetscErrorCode waitForContext(PetscDeviceContext, PetscDeviceContext) noexcept { return PET… in waitForContext() argument 29 static PetscErrorCode synchronize(PetscDeviceContext) noexcept { return PETSC_SUCCESS; } in synchronize() argument 66 PetscErrorCode PetscDeviceContextCreate_HOST(PetscDeviceContext dctx) in PetscDeviceContextCreate_HOST()
|
| /petsc/src/sys/objects/device/interface/ |
| H A D | dcontext.cxx | 14 PetscErrorCode construct_(PetscDeviceContext dctx) const noexcept in construct_() 26 static PetscErrorCode destroy_(PetscDeviceContext dctx) noexcept in destroy_() 38 static PetscErrorCode reset_(PetscDeviceContext dctx, bool zero = true) noexcept in reset_() 61 static PetscErrorCode invalidate_(PetscDeviceContext dctx) noexcept in invalidate_() 101 PetscErrorCode PetscDeviceContextCreate(PetscDeviceContext *dctx) in PetscDeviceContextCreate() 146 PetscErrorCode PetscDeviceContextDestroy(PetscDeviceContext *dctx) in PetscDeviceContextDestroy() 181 PetscErrorCode PetscDeviceContextSetStreamType(PetscDeviceContext dctx, PetscStreamType type) in PetscDeviceContextSetStreamType() 220 PetscErrorCode PetscDeviceContextGetStreamType(PetscDeviceContext dctx, PetscStreamType *type) in PetscDeviceContextGetStreamType() 242 static PetscErrorCode PetscDeviceContextSetDevice_Private(PetscDeviceContext dctx, PetscDevice devi… in PetscDeviceContextSetDevice_Private() 263 PetscErrorCode PetscDeviceContextSetDefaultDeviceForType_Internal(PetscDeviceContext dctx, PetscDev… in PetscDeviceContextSetDefaultDeviceForType_Internal() [all …]
|
| /petsc/src/sys/objects/device/tests/ |
| H A D | ex4.c | 5 static PetscErrorCode DoFork(PetscDeviceContext parent, PetscInt n, PetscDeviceContext **sub) in DoFork() 28 …c PetscErrorCode TestNestedPetscDeviceContextForkJoin(PetscDeviceContext parCtx, PetscDeviceContex… in TestNestedPetscDeviceContextForkJoin() 31 PetscDeviceContext *subsub; in TestNestedPetscDeviceContextForkJoin() 47 static PetscErrorCode TestPetscDeviceContextForkJoin(PetscDeviceContext dctx) in TestPetscDeviceContextForkJoin() 49 PetscDeviceContext *sub; in TestPetscDeviceContextForkJoin() 74 PetscDeviceContext dctx; in main()
|
| /petsc/src/sys/objects/device/tests/output/ |
| H A D | ex6_root_device_context_stream_type-default_with_barrier_hip.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-default_with_barrier_cuda.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-nonblocking_host_with_device.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-nonblocking_hip.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-default_with_barrier_host_with_device.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-default_hip.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-null_sycl.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-default_host_with_device.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-default_sycl.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-null_host_with_device.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-nonblocking_cuda.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-null_host_no_device.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-default_host_no_device.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|
| H A D | ex6_root_device_context_stream_type-null_hip.out | 1 PetscDeviceContext Object: global root (root_) 1 MPI process 8 PetscDeviceContext Object: global root (root_) 1 MPI process 15 PetscDeviceContext Object: 1 MPI process 22 PetscDeviceContext Object: 1 MPI process 29 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process 36 PetscDeviceContext Object: null context 0 (null_context_0_) 1 MPI process
|