Home
last modified time | relevance | path

Searched refs:memtype (Results 1 – 21 of 21) sorted by relevance

/petsc/src/mat/tests/
H A Dex217.c9 PetscMemType memtype; in main() local
29 PetscCall(MatGetCurrentMemType(A, &memtype)); in main()
30 …if (iscuda) PetscCheck(memtype == PETSC_MEMTYPE_CUDA, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "wrong mem… in main()
31 …else if (iship) PetscCheck(memtype == PETSC_MEMTYPE_HIP, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "wrong … in main()
32 …else if (iskokkos) PetscCheck(memtype == PETSC_MEMTYPE_KOKKOS, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "… in main()
33 … (ishypre) PetscCheck(PetscDefined(HAVE_HYPRE_DEVICE) ? memtype == PETSC_MEMTYPE_DEVICE : memtype … in main()
34 …else PetscCheck(memtype == PETSC_MEMTYPE_HOST, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "wrong memory typ… in main()
39 PetscCall(MatGetCurrentMemType(A, &memtype)); in main()
40 … PetscCheck(memtype == PETSC_MEMTYPE_HOST, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "wrong memory type"); in main()
/petsc/src/vec/vec/tests/
H A Dbenchmark_veccreate.c13 PetscMemType memtype; in main() local
27 PetscCall(VecGetArrayAndMemType(x, &array, &memtype)); in main()
H A Dex65.c9 PetscMemType memtype; in main() local
25 PetscCall(VecGetArrayAndMemType(global, &array, &memtype)); in main()
/petsc/src/ksp/ksp/utils/lmvm/dense/
H A Dcd_utils.c88 …UpperTriangularSolveInPlace_Internal(MatLMVMDenseType type, PetscMemType memtype, PetscBool hermit… in MatUpperTriangularSolveInPlace_Internal() argument
99 if (PetscMemTypeHost(memtype)) { in MatUpperTriangularSolveInPlace_Internal()
106 } else if (PetscMemTypeDevice(memtype)) { in MatUpperTriangularSolveInPlace_Internal()
112 if (PetscMemTypeHost(memtype)) { in MatUpperTriangularSolveInPlace_Internal()
132 } else if (PetscMemTypeDevice(memtype)) { in MatUpperTriangularSolveInPlace_Internal()
H A Ddenseqn.c340 PetscMemType memtype; in MatUpdate_LMVMDQN() local
346 PetscCall(VecGetArrayAndMemType(lqn->StFprev, &StFprev, &memtype)); in MatUpdate_LMVMDQN()
348 if (PetscMemTypeHost(memtype)) { in MatUpdate_LMVMDQN()
352 PetscCall(PetscDeviceRegisterMemory(StFprev, memtype, local_n * sizeof(*StFprev))); in MatUpdate_LMVMDQN()
869 PetscMemType memtype; in MatMult_LMVMDBFGS() local
888 PetscCall(VecGetArrayAndMemType(lbfgs->rwork3, &workscalar, &memtype)); in MatMult_LMVMDBFGS()
893 if (PetscMemTypeHost(memtype)) { in MatMult_LMVMDBFGS()
897 … PetscCall(PetscDeviceRegisterMemory(workscalar, memtype, local_n * sizeof(*workscalar))); in MatMult_LMVMDBFGS()
916 PetscCall(VecGetArrayAndMemType(lbfgs->rwork1, &workscalar, &memtype)); in MatMult_LMVMDBFGS()
921 if (PetscMemTypeHost(memtype)) { in MatMult_LMVMDBFGS()
[all …]
/petsc/src/dm/interface/
H A Ddmceed.c49 PetscMemType memtype; in VecGetCeedVector() local
55 PetscCall(VecGetArrayAndMemType(X, &x, &memtype)); in VecGetCeedVector()
57 PetscCallCEED(CeedVectorSetArray(*cx, PetscMemType2Ceed(memtype), CEED_USE_POINTER, x)); in VecGetCeedVector()
71 PetscMemType memtype; in VecGetCeedVectorRead() local
77 PetscCall(VecGetArrayReadAndMemType(X, &x, &memtype)); in VecGetCeedVectorRead()
79 …PetscCallCEED(CeedVectorSetArray(*cx, PetscMemType2Ceed(memtype), CEED_USE_POINTER, (PetscScalar *… in VecGetCeedVectorRead()
/petsc/src/mat/impls/aij/mpi/mpihipsparse/
H A Dmpiaijhipsparse.hip.cxx164 PetscMemType memtype; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
191 PetscCall(PetscGetMemType(v, &memtype)); in MatSetValuesCOO_MPIAIJHIPSPARSE()
192 if (PetscMemTypeHost(memtype)) { /* If user gave v[] in host, we need to copy it to device */ in MatSetValuesCOO_MPIAIJHIPSPARSE()
235 if (PetscMemTypeHost(memtype)) PetscCallHIP(hipFree((void *)v1)); in MatSetValuesCOO_MPIAIJHIPSPARSE()
/petsc/src/vec/vec/impls/mpi/kokkos/
H A Dmpikok.kokkos.cxx164 PetscMemType memtype; in VecSetValuesCOO_MPIKokkos() local
169 PetscCall(PetscGetMemType(v, &memtype)); in VecSetValuesCOO_MPIKokkos()
170 …if (PetscMemTypeHost(memtype)) { /* If user gave v[] in host, we might need to copy it to device i… in VecSetValuesCOO_MPIKokkos()
/petsc/src/mat/impls/aij/mpi/mpicusparse/
H A Dmpiaijcusparse.cu164 PetscMemType memtype; in MatSetValuesCOO_MPIAIJCUSPARSE() local
191 PetscCall(PetscGetMemType(v, &memtype)); in MatSetValuesCOO_MPIAIJCUSPARSE()
192 if (PetscMemTypeHost(memtype)) { /* If user gave v[] in host, we need to copy it to device */ in MatSetValuesCOO_MPIAIJCUSPARSE()
235 if (PetscMemTypeHost(memtype)) PetscCallCUDA(cudaFree((void *)v1)); in MatSetValuesCOO_MPIAIJCUSPARSE()
/petsc/src/ksp/ksp/utils/lmvm/
H A Dlmproducts.c509 PetscMemType memtype; in LMProductsInsertNextDiagonalValue() local
511 PetscCall(VecGetArrayAndMemType(dots->diagonal_global, &array, &memtype)); in LMProductsInsertNextDiagonalValue()
513 if (PetscMemTypeHost(memtype)) { in LMProductsInsertNextDiagonalValue()
/petsc/src/tao/unconstrained/tutorials/
H A Drosenbrock4.h212 PetscMemType memtype; in CreateHessian() local
215 PetscCall(VecGetArrayReadAndMemType(user->fvector, &a, &memtype)); in CreateHessian()
218 if (memtype == PETSC_MEMTYPE_DEVICE) { in CreateHessian()
/petsc/src/dm/dt/dualspace/impls/sum/
H A Ddualspacesum.c319 PetscMemType memtype; in PetscDualSpaceSumCreateMatrix() local
334 PetscCall(MatSeqAIJGetCSRAndMemType(smat, &si, &sj, &sv, &memtype)); in PetscDualSpaceSumCreateMatrix()
335 …PetscCheck(memtype == PETSC_MEMTYPE_HOST, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Could not convert matr… in PetscDualSpaceSumCreateMatrix()
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm_impl.hpp2250 PetscMemType memtype; in SetValuesCOO() local
2256 PetscCall(PetscGetMemType(v, &memtype)); in SetValuesCOO()
2257 if (PetscMemTypeHost(memtype)) { in SetValuesCOO()
2273 if (PetscMemTypeHost(memtype)) PetscCall(PetscDeviceFree(dctx, vv)); in SetValuesCOO()
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx1583 PetscMemType memtype; in VecSetValuesCOO_SeqKokkos() local
1587 PetscCall(PetscGetMemType(v, &memtype)); in VecSetValuesCOO_SeqKokkos()
1588 …if (PetscMemTypeHost(memtype)) { /* If user gave v[] in host, we might need to copy it to device i… in VecSetValuesCOO_SeqKokkos()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx1583 PetscMemType memtype; in MatSetValuesCOO_MPIAIJKokkos() local
1611 PetscCall(PetscGetMemType(v, &memtype)); /* Return PETSC_MEMTYPE_HOST when v is NULL */ in MatSetValuesCOO_MPIAIJKokkos()
1612 …if (PetscMemTypeHost(memtype)) { /* If user gave v[] in host, we need to copy it to device… in MatSetValuesCOO_MPIAIJKokkos()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.kokkos.cxx1397 PetscMemType memtype; in MatSetValuesCOO_SeqAIJKokkos() local
1410 PetscCall(PetscGetMemType(v, &memtype)); in MatSetValuesCOO_SeqAIJKokkos()
1411 …if (PetscMemTypeHost(memtype)) { /* If user gave v[] in host, we might need to copy it to device i… in MatSetValuesCOO_SeqAIJKokkos()
/petsc/src/mat/impls/hypre/
H A Dmhypre.c608 PetscMemType memtype = PETSC_MEMTYPE_HOST; local
612 PetscCall(MatGetCurrentMemType(A, &memtype));
614 boundtocpu = PetscMemTypeHost(memtype) ? PETSC_TRUE : PETSC_FALSE;
/petsc/src/ksp/pc/impls/hypre/
H A Dhypre.c311 PetscMemType memtype; in PCSetUp_HYPRE() local
314 PetscCall(MatGetCurrentMemType(jac->hpmat, &memtype)); in PCSetUp_HYPRE()
315 if (PetscMemTypeDevice(memtype)) { in PCSetUp_HYPRE()
/petsc/src/mat/impls/aij/seq/seqhipsparse/
H A Daijhipsparse.hip.cxx3786 PetscMemType memtype; in MatSetValuesCOO_SeqAIJHIPSPARSE() local
3798 PetscCall(PetscGetMemType(v, &memtype)); in MatSetValuesCOO_SeqAIJHIPSPARSE()
3799 …if (PetscMemTypeHost(memtype)) { /* If user gave v[] in host, we might need to copy it to device i… in MatSetValuesCOO_SeqAIJHIPSPARSE()
3817 if (PetscMemTypeHost(memtype)) PetscCallHIP(hipFree((void *)v1)); in MatSetValuesCOO_SeqAIJHIPSPARSE()
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu4429 PetscMemType memtype; in MatSetValuesCOO_SeqAIJCUSPARSE() local
4441 PetscCall(PetscGetMemType(v, &memtype)); in MatSetValuesCOO_SeqAIJCUSPARSE()
4442 …if (PetscMemTypeHost(memtype)) { /* If user gave v[] in host, we might need to copy it to device i… in MatSetValuesCOO_SeqAIJCUSPARSE()
4460 if (PetscMemTypeHost(memtype)) PetscCallCUDA(cudaFree((void *)v1)); in MatSetValuesCOO_SeqAIJCUSPARSE()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.c8016 PETSC_INTERN PetscErrorCode MatGetCurrentMemType_MPIAIJ(Mat A, PetscMemType *memtype) in MatGetCurrentMemType_MPIAIJ() argument
8024 *memtype = (mD == mO) ? mD : PETSC_MEMTYPE_HOST; in MatGetCurrentMemType_MPIAIJ()