10e6b6b59SJacob Faibussowitsch #include <petsc/private/deviceimpl.h> 20e6b6b59SJacob Faibussowitsch 30e6b6b59SJacob Faibussowitsch #include <petsc/private/cpp/macros.hpp> 40e6b6b59SJacob Faibussowitsch #include <petsc/private/cpp/utility.hpp> 50e6b6b59SJacob Faibussowitsch 60e6b6b59SJacob Faibussowitsch namespace Petsc { 70e6b6b59SJacob Faibussowitsch 80e6b6b59SJacob Faibussowitsch namespace device { 90e6b6b59SJacob Faibussowitsch 100e6b6b59SJacob Faibussowitsch namespace host { 110e6b6b59SJacob Faibussowitsch 120e6b6b59SJacob Faibussowitsch namespace impl { 130e6b6b59SJacob Faibussowitsch 140e6b6b59SJacob Faibussowitsch class DeviceContext { 150e6b6b59SJacob Faibussowitsch public: 160e6b6b59SJacob Faibussowitsch PETSC_CXX_COMPAT_DECL(PetscErrorCode destroy(PetscDeviceContext)) { return 0; } 170e6b6b59SJacob Faibussowitsch PETSC_CXX_COMPAT_DECL(PetscErrorCode changeStreamType(PetscDeviceContext, PetscStreamType)) { return 0; } 180e6b6b59SJacob Faibussowitsch PETSC_CXX_COMPAT_DECL(PetscErrorCode setUp(PetscDeviceContext)) { return 0; } 190e6b6b59SJacob Faibussowitsch PETSC_CXX_COMPAT_DECL(PetscErrorCode query(PetscDeviceContext, PetscBool *idle)) { 200e6b6b59SJacob Faibussowitsch PetscFunctionBegin; 210e6b6b59SJacob Faibussowitsch *idle = PETSC_TRUE; // the host is always idle 220e6b6b59SJacob Faibussowitsch PetscFunctionReturn(0); 230e6b6b59SJacob Faibussowitsch } 240e6b6b59SJacob Faibussowitsch PETSC_CXX_COMPAT_DECL(PetscErrorCode waitForContext(PetscDeviceContext, PetscDeviceContext)) { return 0; } 250e6b6b59SJacob Faibussowitsch PETSC_CXX_COMPAT_DECL(PetscErrorCode synchronize(PetscDeviceContext)) { return 0; } 260e6b6b59SJacob Faibussowitsch PETSC_CXX_COMPAT_DECL(PetscErrorCode getBlasHandle(PetscDeviceContext, void *)) { SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Not implemented"); } 270e6b6b59SJacob Faibussowitsch PETSC_CXX_COMPAT_DECL(PetscErrorCode getSolverHandle(PetscDeviceContext, void *)) { SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Not implemented"); } 280e6b6b59SJacob Faibussowitsch PETSC_CXX_COMPAT_DECL(PetscErrorCode getStreamHandle(PetscDeviceContext, void *)) { SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Not implemented"); } 290e6b6b59SJacob Faibussowitsch PETSC_CXX_COMPAT_DECL(PetscErrorCode beginTimer(PetscDeviceContext)) { SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Not implemented"); } 300e6b6b59SJacob Faibussowitsch PETSC_CXX_COMPAT_DECL(PetscErrorCode endTimer(PetscDeviceContext, PetscLogDouble *)) { SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Not implemented"); } 310e6b6b59SJacob Faibussowitsch 32*3f675fcfSPierre Jolivet const _DeviceContextOps ops = {destroy, changeStreamType, setUp, query, waitForContext, synchronize, getBlasHandle, getSolverHandle, getStreamHandle, beginTimer, endTimer, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr}; 330e6b6b59SJacob Faibussowitsch }; 340e6b6b59SJacob Faibussowitsch 350e6b6b59SJacob Faibussowitsch } // namespace impl 360e6b6b59SJacob Faibussowitsch 370e6b6b59SJacob Faibussowitsch } // namespace host 380e6b6b59SJacob Faibussowitsch 390e6b6b59SJacob Faibussowitsch } // namespace device 400e6b6b59SJacob Faibussowitsch 410e6b6b59SJacob Faibussowitsch } // namespace Petsc 420e6b6b59SJacob Faibussowitsch 430e6b6b59SJacob Faibussowitsch PetscErrorCode PetscDeviceContextCreate_HOST(PetscDeviceContext dctx) { 440e6b6b59SJacob Faibussowitsch static constexpr auto hostctx = ::Petsc::device::host::impl::DeviceContext{}; 450e6b6b59SJacob Faibussowitsch 460e6b6b59SJacob Faibussowitsch PetscFunctionBegin; 470e6b6b59SJacob Faibussowitsch PetscAssert(!dctx->data, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "PetscDeviceContext %" PetscInt64_FMT " is of type host, but still has data member %p", PetscObjectCast(dctx)->id, dctx->data); 480e6b6b59SJacob Faibussowitsch PetscCall(PetscArraycpy(dctx->ops, &hostctx.ops, 1)); 490e6b6b59SJacob Faibussowitsch PetscFunctionReturn(0); 500e6b6b59SJacob Faibussowitsch } 51