Home
last modified time | relevance | path

Searched refs:iscuda (Results 1 – 19 of 19) sorted by relevance

/petsc/src/vec/is/sf/tests/
H A Dex9.c16 PetscBool iscuda = PETSC_FALSE; /* Option to use VECCUDA vectors */ in main() local
36 if (compareflag) iscuda = PETSC_TRUE; in main()
50 if (iscuda) { in main()
73 if (iscuda) { in main()
81 if (iscuda) { in main()
91 if (iscuda) { in main()
112 if (iscuda) { in main()
135 if (iscuda) { in main()
188 if (iscuda) { in main()
216 if (iscuda) { in main()
[all …]
/petsc/src/mat/utils/
H A Dveccreatematdense.c28 PetscBool isstd, iscuda, iship, iskokkos; in MatCreateDenseFromVecType() local
32 PetscCall(PetscStrcmpAny(vtype, &iscuda, VECCUDA, VECMPICUDA, VECSEQCUDA, "")); in MatCreateDenseFromVecType()
35 PetscCheck(isstd || iscuda || iship || iskokkos, comm, PETSC_ERR_SUP, "Not for type %s", vtype); in MatCreateDenseFromVecType()
36 if (iscuda) root_type = VECCUDA; in MatCreateDenseFromVecType()
41 if (PetscDefined(HAVE_MACRO_KOKKOS_ENABLE_CUDA)) iscuda = PETSC_TRUE; in MatCreateDenseFromVecType()
53 } else if (iscuda) { in MatCreateDenseFromVecType()
/petsc/src/mat/tests/
H A Dex217.c11 PetscBool ishypre, iskokkos, iscuda, iship; in main() local
19 …PetscCall(PetscObjectTypeCompareAny((PetscObject)A, &iscuda, MATMPIAIJCUSPARSE, MATSEQAIJCUSPARSE,… in main()
30 …if (iscuda) PetscCheck(memtype == PETSC_MEMTYPE_CUDA, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "wrong mem… in main()
/petsc/src/vec/vec/tests/
H A Dex60.c8 PetscBool iscuda, iskokkos, iship; in main() local
21 PetscCall(PetscObjectTypeCompare((PetscObject)x, VECSEQCUDA, &iscuda)); in main()
25 if (iscuda) { in main()
H A Dex38.c14 PetscBool iscuda; in main() local
31 PetscCall(PetscObjectTypeCompareAny((PetscObject)X, &iscuda, VECSEQCUDA, VECMPICUDA, "")); in main()
32 if (iscuda) { /* trigger a copy of the data on the GPU */ in main()
/petsc/src/mat/impls/h2opus/cuda/
H A Dmath2opusutils.cu93 PetscBool iscuda; in VecSetDelta() local
100 PetscCall(PetscObjectTypeCompareAny((PetscObject)x, &iscuda, VECSEQCUDA, VECMPICUDA, "")); in VecSetDelta()
101 iscuda = (PetscBool)(iscuda && !x->boundtocpu); in VecSetDelta()
102 if (iscuda) { in VecSetDelta()
H A Dmath2opus.cu1678 PetscBool isstd, iscuda, iskok; in MatCreateH2OpusFromMat() local
1682 PetscCall(PetscStrcmpAny(vtype, &iscuda, VECCUDA, VECSEQCUDA, VECMPICUDA, "")); in MatCreateH2OpusFromMat()
1684 PetscCheck(isstd || iscuda || iskok, comm, PETSC_ERR_SUP, "Not for type %s", vtype); in MatCreateH2OpusFromMat()
1685 if (iscuda && !B->boundtocpu) boundtocpu = PETSC_FALSE; in MatCreateH2OpusFromMat()
/petsc/src/vec/vec/interface/
H A Dvecreg.c196 PetscBool iscuda, iship, iskokkos, isvcl; in VecGetRootType_Private() local
201 …PetscCall(PetscObjectTypeCompareAny((PetscObject)vec, &iscuda, VECCUDA, VECMPICUDA, VECSEQCUDA, ""… in VecGetRootType_Private()
205 if (iscuda) { in VecGetRootType_Private()
H A Drvector.c1606 PetscBool isstd, iscuda, iship; in VecGetSubVector() local
1609 PetscCall(PetscObjectTypeCompareAny((PetscObject)X, &iscuda, VECSEQCUDA, VECMPICUDA, "")); in VecGetSubVector()
1611 if (iscuda) { in VecGetSubVector()
1728 PetscBool iscuda, iship; in VecRestoreSubVector() local
1729 PetscCall(PetscObjectTypeCompareAny((PetscObject)X, &iscuda, VECSEQCUDA, VECMPICUDA, "")); in VecRestoreSubVector()
1732 if (iscuda) { in VecRestoreSubVector()
/petsc/src/mat/interface/
H A Dmatproduct.c292 PetscBool t = PETSC_TRUE, iscuda = PETSC_FALSE; in MatProductNumeric_X_Dense() local
309 PetscCall(PetscStrcmp(vtype, VECCUDA, &iscuda)); in MatProductNumeric_X_Dense()
310 if (!iscuda) PetscCall(PetscStrcmp(vtype, VECSEQCUDA, &iscuda)); in MatProductNumeric_X_Dense()
311 if (!iscuda) PetscCall(PetscStrcmp(vtype, VECMPICUDA, &iscuda)); in MatProductNumeric_X_Dense()
312 if (iscuda) { /* Make sure we have up-to-date data on the GPU */ in MatProductNumeric_X_Dense()
347 if (iscuda) { in MatProductNumeric_X_Dense()
/petsc/src/binding/petsc4py/src/lib-petsc/
H A Dcustom.h178 char *iscuda = NULL, *iship = NULL, *iskok = NULL; in VecGetCurrentMemType() local
181 PetscCall(PetscStrstr(rtype,"cuda",&iscuda)); in VecGetCurrentMemType()
184 if (iscuda) *m = PETSC_MEMTYPE_CUDA; in VecGetCurrentMemType()
/petsc/src/mat/impls/hypre/
H A Dmhypre.c807 PetscBool iscuda, iship; local
818 …PetscCall(PetscObjectTypeCompareAny((PetscObject)A, &iscuda, MATSEQAIJCUSPARSE, MATMPIAIJCUSPARSE,…
826 if (!boundtocpu && (iscuda || iship)) {
828 if (iscuda) {
861 if (!boundtocpu && (iscuda || iship)) {
863 if (iscuda) {
959 PetscBool iscuda, iship; local
963 …PetscCall(PetscObjectTypeCompareAny((PetscObject)A, &iscuda, MATSEQAIJCUSPARSE, MATMPIAIJCUSPARSE,…
966 if (iscuda) sameint = PETSC_TRUE;
/petsc/src/mat/impls/dense/mpi/
H A Dmpidense.c208 PetscBool iscuda; in MatDenseSetLDA_MPIDense() local
209 PetscCall(PetscObjectTypeCompare((PetscObject)A, MATMPIDENSECUDA, &iscuda)); in MatDenseSetLDA_MPIDense()
210 if (iscuda) mtype = MATSEQDENSECUDA; in MatDenseSetLDA_MPIDense()
1381 PetscBool iscuda; in MatMPIDenseSetPreallocation_MPIDense() local
1382 PetscCall(PetscObjectTypeCompare((PetscObject)mat, MATMPIDENSECUDA, &iscuda)); in MatMPIDenseSetPreallocation_MPIDense()
1383 if (iscuda) mtype = MATSEQDENSECUDA; in MatMPIDenseSetPreallocation_MPIDense()
/petsc/src/dm/impls/swarm/
H A Dswarm.c315 PetscBool iscuda, iskokkos, iship; in DMSwarmCreateVectorFromField_Private() local
325 PetscCall(PetscStrcmp(dm->vectype, VECCUDA, &iscuda)); in DMSwarmCreateVectorFromField_Private()
331 else if (iscuda) PetscCall(VecSetType(*vec, VECCUDA)); in DMSwarmCreateVectorFromField_Private()
/petsc/src/ksp/pc/impls/hypre/
H A Dhypre.c288 PetscBool iscuda, iship, iskokkos; in PCSetUp_HYPRE() local
290 …PetscCall(PetscObjectTypeCompareAny((PetscObject)pc->pmat, &iscuda, MATSEQAIJCUSPARSE, MATMPIAIJCU… in PCSetUp_HYPRE()
293 if (iscuda || iship || iskokkos) PetscCall(MatDestroy(&jac->hpmat)); in PCSetUp_HYPRE()
/petsc/src/mat/impls/dense/seq/
H A Ddense.c3369 PetscBool isstd, iskok, iscuda, iship; in MatDenseCreateColumnVec_Private() local
3380 PetscCall(PetscStrcmpAny(A->defaultvectype, &iscuda, VECCUDA, VECSEQCUDA, VECMPICUDA, "")); in MatDenseCreateColumnVec_Private()
3392 } else if (iscuda) { in MatDenseCreateColumnVec_Private()
/petsc/src/dm/impls/plex/
H A Dplex.c10721 PetscBool iscuda, iskokkos; in DMPlexCreateConstraintMatrix_Anchors() local
10729 PetscCall(PetscStrcmp(dm->mattype, MATSEQAIJCUSPARSE, &iscuda)); in DMPlexCreateConstraintMatrix_Anchors()
10730 if (!iscuda) PetscCall(PetscStrcmp(dm->mattype, MATMPIAIJCUSPARSE, &iscuda)); in DMPlexCreateConstraintMatrix_Anchors()
10733 if (iscuda) mtype = MATSEQAIJCUSPARSE; in DMPlexCreateConstraintMatrix_Anchors()
/petsc/src/ksp/pc/impls/bddc/
H A Dbddcprivate.c5434 PetscBool iscuda, iship, iskokkos; in PCBDDCSetUpCorrection() local
5437 …PetscCall(PetscObjectTypeCompareAny((PetscObject)pcis->vec1_N, &iscuda, VECCUDA, VECMPICUDA, VECSE… in PCBDDCSetUpCorrection()
5441 if (PetscDefined(HAVE_MACRO_KOKKOS_ENABLE_CUDA)) iscuda = PETSC_TRUE; in PCBDDCSetUpCorrection()
5444 …if (iskokkos) mtype = multi_element ? MATSEQAIJKOKKOS : (iscuda ? MATSEQDENSECUDA : MATSEQDENSEHIP… in PCBDDCSetUpCorrection()
5446 else if (iscuda) mtype = multi_element ? MATSEQAIJCUSPARSE : MATSEQDENSECUDA; in PCBDDCSetUpCorrection()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.c7097 PetscBool mptmp[MAX_NUMBER_INTERMEDIATE], hasoffproc = PETSC_FALSE, iscuda, iship, iskokk; in MatProductSymbolic_MPIAIJBACKEND() local
7388 …PetscCall(PetscObjectTypeCompareAny((PetscObject)C, &iscuda, MATSEQAIJCUSPARSE, MATMPIAIJCUSPARSE,… in MatProductSymbolic_MPIAIJBACKEND()
7391 if (iscuda) mmdata->mtype = PETSC_MEMTYPE_CUDA; in MatProductSymbolic_MPIAIJBACKEND()