Home
last modified time | relevance | path

Searched refs:PETSC_MEMTYPE_HOST (Results 1 – 25 of 47) sorted by relevance

12

/petsc/include/
H A Dpetscdevicetypes.h46 PETSC_MEMTYPE_HOST = 0, enumerator
60 #define PETSC_MEMTYPE_KOKKOS PETSC_MEMTYPE_HOST
63 #define PetscMemTypeHost(m) (((m) & 0x1) == PETSC_MEMTYPE_HOST)
75 static_assert(PetscMemTypeHost(PETSC_MEMTYPE_HOST), "");
82 static_assert(!PetscMemTypeDevice(PETSC_MEMTYPE_HOST), "");
106 PETSC_CASE_NAME(PETSC_MEMTYPE_HOST); in PetscMemTypeToString()
202 return PETSC_MEMTYPE_HOST; in PetscOffloadMaskToMemType()
210 return PETSC_MEMTYPE_HOST; in PetscOffloadMaskToMemType()
/petsc/src/vec/is/sf/impls/basic/
H A Dsfmpi.c72 …PetscMemType rootmtype = PetscMemTypeHost(xrootmtype) ? PETSC_MEMTYPE_HOST : PETSC_MEMTYPE_DEV… in PetscSFLinkCreate_MPI()
73 …PetscMemType leafmtype = PetscMemTypeHost(xleafmtype) ? PETSC_MEMTYPE_HOST : PETSC_MEMTYPE_DEV… in PetscSFLinkCreate_MPI()
107 rootmtype_mpi = leafmtype_mpi = PETSC_MEMTYPE_HOST; in PetscSFLinkCreate_MPI()
210 …][PETSC_MEMTYPE_HOST]) PetscCall(PetscMalloc(bas->rootbuflen[PETSCSF_REMOTE] * link->unitbytes, &l… in PetscSFLinkCreate_MPI()
211 …link->rootbuf[PETSCSF_REMOTE][PETSC_MEMTYPE_HOST] = link->rootbuf_alloc[PETSCSF_REMOTE][PETSC_MEMT… in PetscSFLinkCreate_MPI()
214 …][PETSC_MEMTYPE_HOST]) PetscCall(PetscMalloc(sf->leafbuflen[PETSCSF_REMOTE] * link->unitbytes, &li… in PetscSFLinkCreate_MPI()
215 …link->leafbuf[PETSCSF_REMOTE][PETSC_MEMTYPE_HOST] = link->leafbuf_alloc[PETSCSF_REMOTE][PETSC_MEMT… in PetscSFLinkCreate_MPI()
H A Dsfpack.h307 void *h_buf = link->rootbuf[PETSCSF_REMOTE][PETSC_MEMTYPE_HOST]; in PetscSFLinkCopyRootBufferInCaseNotUseGpuAwareMPI()
311 … PetscCall((*link->Memcpy)(link, PETSC_MEMTYPE_HOST, h_buf, PETSC_MEMTYPE_DEVICE, d_buf, count)); in PetscSFLinkCopyRootBufferInCaseNotUseGpuAwareMPI()
314 … PetscCall((*link->Memcpy)(link, PETSC_MEMTYPE_DEVICE, d_buf, PETSC_MEMTYPE_HOST, h_buf, count)); in PetscSFLinkCopyRootBufferInCaseNotUseGpuAwareMPI()
325 void *h_buf = link->leafbuf[PETSCSF_REMOTE][PETSC_MEMTYPE_HOST]; in PetscSFLinkCopyLeafBufferInCaseNotUseGpuAwareMPI()
329 … PetscCall((*link->Memcpy)(link, PETSC_MEMTYPE_HOST, h_buf, PETSC_MEMTYPE_DEVICE, d_buf, count)); in PetscSFLinkCopyLeafBufferInCaseNotUseGpuAwareMPI()
332 … PetscCall((*link->Memcpy)(link, PETSC_MEMTYPE_DEVICE, d_buf, PETSC_MEMTYPE_HOST, h_buf, count)); in PetscSFLinkCopyLeafBufferInCaseNotUseGpuAwareMPI()
398 …cpy)(link, PETSC_MEMTYPE_DEVICE, bas->rootpackopt_d[scope]->array, PETSC_MEMTYPE_HOST, bas->rootpa… in PetscSFLinkGetRootPackOptAndIndices()
405 …*link->Memcpy)(link, PETSC_MEMTYPE_DEVICE, bas->irootloc_d[scope], PETSC_MEMTYPE_HOST, bas->irootl… in PetscSFLinkGetRootPackOptAndIndices()
440 …mcpy)(link, PETSC_MEMTYPE_DEVICE, sf->leafpackopt_d[scope]->array, PETSC_MEMTYPE_HOST, sf->leafpac… in PetscSFLinkGetLeafPackOptAndIndices()
447 …ll((*link->Memcpy)(link, PETSC_MEMTYPE_DEVICE, sf->rmine_d[scope], PETSC_MEMTYPE_HOST, sf->rmine +… in PetscSFLinkGetLeafPackOptAndIndices()
H A Dsfpack.c412 PetscCall(PetscFree(link->rootbuf_alloc[i][PETSC_MEMTYPE_HOST])); in PetscSFLinkDestroy()
413 PetscCall(PetscFree(link->leafbuf_alloc[i][PETSC_MEMTYPE_HOST])); in PetscSFLinkDestroy()
999 …PetscCall(PetscSFLinkGetRootPackOptAndIndices(sf, link, PETSC_MEMTYPE_HOST, scope, &count, &start,… in PetscSFLinkUnpackRootData_Private()
1022 …PetscCall(PetscSFLinkGetLeafPackOptAndIndices(sf, link, PETSC_MEMTYPE_HOST, scope, &count, &start,… in PetscSFLinkUnpackLeafData_Private()
1131 …PetscCall(PetscSFLinkGetRootPackOptAndIndices(sf, link, PETSC_MEMTYPE_HOST, PETSCSF_LOCAL, &count,… in PetscSFLinkScatterLocal()
1132 …PetscCall(PetscSFLinkGetLeafPackOptAndIndices(sf, link, PETSC_MEMTYPE_HOST, PETSCSF_LOCAL, &count,… in PetscSFLinkScatterLocal()
1361 PetscCall(PetscSFDestroyPackOpt(sf, PETSC_MEMTYPE_HOST, &sf->leafpackopt[i])); in PetscSFResetPackFields()
1362 PetscCall(PetscSFDestroyPackOpt(sf, PETSC_MEMTYPE_HOST, &bas->rootpackopt[i])); in PetscSFResetPackFields()
/petsc/src/vec/is/sf/tests/
H A Dex4k.kokkos.cxx43 PetscMemType mtype = PETSC_MEMTYPE_HOST; in main()
68 if (isHost) mtype = PETSC_MEMTYPE_HOST; in main()
73 if (mtype == PETSC_MEMTYPE_HOST) { in main()
113 if (mtype == PETSC_MEMTYPE_HOST) { in main()
148 …-pong test on %s ##\n Message(Bytes) \t\tLatency(us)\n", mtype == PETSC_MEMTYPE_HOST ? "Host" : "… in main()
154 if (mtype == PETSC_MEMTYPE_HOST) { in main()
H A Dex23.c36 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_HOST, n2, &xh)); in main()
37 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_HOST, n2, &yh)); in main()
/petsc/src/sys/objects/device/tests/
H A Dex7.c72 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_HOST, n, &tmp_ptr)); in TestAllocate()
89 PetscCall(PetscDeviceRegisterMemory(tmp_ptr, PETSC_MEMTYPE_HOST, n * sizeof(*tmp_ptr))); in TestAllocate()
113 PetscCall(PetscDeviceMalloc(sub[i], PETSC_MEMTYPE_HOST, n, &ptr)); in TestAsyncCoherence()
122 PetscCall(PetscDeviceMalloc(sub[0], PETSC_MEMTYPE_HOST, n, &ptr)); in TestAsyncCoherence()
127 PetscCall(PetscDeviceMalloc(sub[0], PETSC_MEMTYPE_HOST, n, &ptr)); in TestAsyncCoherence()
139 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_HOST, n, &tmp_ptr)); in TestAsyncCoherence()
191 PetscCall(TestAllocate(dctx, rand, PETSC_MEMTYPE_HOST)); in main()
/petsc/src/mat/tests/
H A Dex217.c33 …(HAVE_HYPRE_DEVICE) ? memtype == PETSC_MEMTYPE_DEVICE : memtype == PETSC_MEMTYPE_HOST, PETSC_COMM_… in main()
34 …else PetscCheck(memtype == PETSC_MEMTYPE_HOST, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "wrong memory typ… in main()
40 … PetscCheck(memtype == PETSC_MEMTYPE_HOST, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "wrong memory type"); in main()
/petsc/include/petsc/private/
H A Dveccupmimpl.h201 …text dctx, Vec v) noexcept PETSC_DECLTYPE_AUTO_RETURNS(VectorArray<PETSC_MEMTYPE_HOST, PETSC_MEMOR…
202 …text dctx, Vec v) noexcept PETSC_DECLTYPE_AUTO_RETURNS(VectorArray<PETSC_MEMTYPE_HOST, PETSC_MEMOR…
203 …text dctx, Vec v) noexcept PETSC_DECLTYPE_AUTO_RETURNS(VectorArray<PETSC_MEMTYPE_HOST, PETSC_MEMOR…
580 static_assert((mtype == PETSC_MEMTYPE_HOST) || (mtype == PETSC_MEMTYPE_DEVICE), ""); in GetArray()
640 static_assert((mtype == PETSC_MEMTYPE_HOST) || (mtype == PETSC_MEMTYPE_DEVICE), ""); in RestoreArray()
720 static_assert((mtype == PETSC_MEMTYPE_HOST) || (mtype == PETSC_MEMTYPE_DEVICE), ""); in PlaceArray()
753 static_assert((mtype == PETSC_MEMTYPE_HOST) || (mtype == PETSC_MEMTYPE_DEVICE), ""); in ReplaceArray()
799 static_assert((mtype == PETSC_MEMTYPE_HOST) || (mtype == PETSC_MEMTYPE_DEVICE), ""); in ResetArray()
907 PetscCall(CheckPointerMatchesMemType_(host_array, PETSC_MEMTYPE_HOST)); in Initialize_CUPMBase()
991 v->ops->replacearray = D::template ReplaceArray<PETSC_MEMTYPE_HOST>; in BindToCPU_CUPMBase()
[all …]
H A Dmatdensecupmimpl.h182 …text dctx, Mat m) noexcept PETSC_DECLTYPE_AUTO_RETURNS(MatrixArray<PETSC_MEMTYPE_HOST, PETSC_MEMOR…
183 …text dctx, Mat m) noexcept PETSC_DECLTYPE_AUTO_RETURNS(MatrixArray<PETSC_MEMTYPE_HOST, PETSC_MEMOR…
184 …text dctx, Mat m) noexcept PETSC_DECLTYPE_AUTO_RETURNS(MatrixArray<PETSC_MEMTYPE_HOST, PETSC_MEMOR…
H A Dcupmobject.hpp368 if (mtype == PETSC_MEMTYPE_HOST) { in CheckPointerMatchesMemType_()
H A Dcupmblasinterface.hpp433 auto mtype = PETSC_MEMTYPE_HOST; in PetscCUPMBlasSetPointerModeFromPointer()
/petsc/src/sys/objects/device/util/
H A Dmemory.c32 *type = PETSC_MEMTYPE_HOST; in PetscGetMemType()
/petsc/src/vec/is/sf/impls/basic/allgather/
H A Dsfallgather.c107 …py)(link, PETSC_MEMTYPE_DEVICE, leafdata, PETSC_MEMTYPE_HOST, link->leafbuf[PETSCSF_REMOTE][PETSC_… in PetscSFBcastToZero_Allgather()
/petsc/src/ksp/ksp/impls/hpddm/
H A Dhpddm.cxx217 template <PetscMemType type = PETSC_MEMTYPE_HOST>
249 …PetscCheck(type != PETSC_MEMTYPE_HOST || it == list.end(), PetscObjectComm((PetscObject)ksp), PETS… in KSPSolve_HPDDM_Private()
268 if (type == PETSC_MEMTYPE_HOST) { in KSPSolve_HPDDM_Private()
290 if (type == PETSC_MEMTYPE_HOST) { in KSPSolve_HPDDM_Private()
321 …if (type == PETSC_MEMTYPE_HOST) PetscCall(HPDDM::IterativeMethod::solve(*data->op, b, x, n, PetscO… in KSPSolve_HPDDM_Private()
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/
H A Dpetscdevice.pxi13 PETSC_MEMTYPE_HOST
H A Dpetscvec.pxi472 cdef PetscMemType mtype = PETSC_MEMTYPE_HOST
482 …devType_ = {PETSC_MEMTYPE_HOST : kDLCPU, PETSC_MEMTYPE_CUDA : kDLCUDA, PETSC_MEMTYPE_HIP : kDLROCM}
/petsc/src/vec/vec/impls/mpi/
H A Dpdvec.c1022 …ypeBegin(sf1, MPIU_INT, PETSC_MEMTYPE_HOST, &nroots2 /*rootdata*/, PETSC_MEMTYPE_HOST, nentries /*… in VecSetPreallocationCOO_MPI()
1049 …scCall(PetscSFReduceWithMemTypeBegin(sf2, MPIU_INT, PETSC_MEMTYPE_HOST, i1 + rem, PETSC_MEMTYPE_HO… in VecSetPreallocationCOO_MPI()
1132 …cSFReduceWithMemTypeBegin(vmpi->coo_sf, MPIU_SCALAR, PETSC_MEMTYPE_HOST, sendbuf, PETSC_MEMTYPE_HO… in VecSetValuesCOO_MPI()
/petsc/src/vec/vec/impls/mpi/cupm/
H A Dvecmpicupm_impl.hpp157 VecSetOp_CUPM(resetarray, VecResetArray_MPI, base_type::template ResetArray<PETSC_MEMTYPE_HOST>); in BindToCPU()
158 VecSetOp_CUPM(placearray, VecPlaceArray_MPI, base_type::template PlaceArray<PETSC_MEMTYPE_HOST>); in BindToCPU()
/petsc/src/tao/unconstrained/tutorials/
H A Drosenbrock4.h479 if (memtype_x == PETSC_MEMTYPE_HOST) { in FormObjective()
520 if (memtype_x == PETSC_MEMTYPE_HOST) { in FormGradient()
576 if (memtype_x == PETSC_MEMTYPE_HOST) { in FormObjectiveGradient()
636 if (memtype_x == PETSC_MEMTYPE_HOST) { in FormHessian()
/petsc/src/vec/is/sf/impls/basic/allgatherv/
H A Dsfallgatherv.c238 …nk->Memcpy)(link, PETSC_MEMTYPE_DEVICE, leafdata, PETSC_MEMTYPE_HOST, link->leafbuf[PETSC_MEMTYPE_… in PetscSFBcastToZero_Allgatherv()
/petsc/src/sys/objects/device/interface/
H A Dmemory.cxx48 PetscMemType mtype = PETSC_MEMTYPE_HOST; // memtype of allocation
/petsc/src/ksp/ksp/utils/lmvm/dense/
H A Ddenseqn.c351 PetscCall(PetscDeviceRegisterMemory(&stFprev, PETSC_MEMTYPE_HOST, 1 * sizeof(stFprev))); in MatUpdate_LMVMDQN()
896 PetscCall(PetscDeviceRegisterMemory(&yjtsi, PETSC_MEMTYPE_HOST, sizeof(yjtsi))); in MatMult_LMVMDBFGS()
924 PetscCall(PetscDeviceRegisterMemory(&ytx, PETSC_MEMTYPE_HOST, 1 * sizeof(ytx))); in MatMult_LMVMDBFGS()
1261 PetscCall(PetscDeviceRegisterMemory(&sjtyi, PETSC_MEMTYPE_HOST, 1 * sizeof(sjtyi))); in MatSolve_LMVMDDFP()
1289 PetscCall(PetscDeviceRegisterMemory(&stf, PETSC_MEMTYPE_HOST, sizeof(stf))); in MatSolve_LMVMDDFP()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx462 …ll(PetscSFReduceWithMemTypeBegin(reduceSF, MPIU_INT, PETSC_MEMTYPE_HOST, sendCol, PETSC_MEMTYPE_HO… in MatMPIAIJKokkosReduceBegin()
786 …all(PetscSFBcastWithMemTypeBegin(ownerSF, MPIU_INT, PETSC_MEMTYPE_HOST, E_RowLen, PETSC_MEMTYPE_HO… in MatMPIAIJKokkosBcastBegin()
857 …cCall(PetscSFBcastWithMemTypeBegin(bcastSF, MPIU_INT, PETSC_MEMTYPE_HOST, jbuf, PETSC_MEMTYPE_HOST… in MatMPIAIJKokkosBcastBegin()
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp1081 … pobj, "MatDenseGetArray_C", MatDenseGetArray_SeqDense, GetArrayC_<PETSC_MEMTYPE_HOST, PETSC_MEMOR… in BindToCPU()
1082 …j, "MatDenseGetArrayRead_C", MatDenseGetArray_SeqDense, GetArrayC_<PETSC_MEMTYPE_HOST, PETSC_MEMOR… in BindToCPU()
1083 …, "MatDenseGetArrayWrite_C", MatDenseGetArray_SeqDense, GetArrayC_<PETSC_MEMTYPE_HOST, PETSC_MEMOR… in BindToCPU()
1164 static_assert((mtype == PETSC_MEMTYPE_HOST) || (mtype == PETSC_MEMTYPE_DEVICE), ""); in GetArray()
1195 static_assert((mtype == PETSC_MEMTYPE_HOST) || (mtype == PETSC_MEMTYPE_DEVICE), ""); in RestoreArray()

12