Home
last modified time | relevance | path

Searched refs:PETSC_OFFLOAD_GPU (Results 1 – 18 of 18) sorted by relevance

/petsc/include/
H A Dpetscdevicetypes.h141 PETSC_OFFLOAD_GPU = 2, /* 0x2 */ enumerator
149 #define PetscOffloadDevice(m) (((m) & PETSC_OFFLOAD_GPU) == PETSC_OFFLOAD_GPU)
159 static_assert(!PetscOffloadHost(PETSC_OFFLOAD_GPU), "");
165 static_assert(PetscOffloadDevice(PETSC_OFFLOAD_GPU), "");
171 static_assert(!PetscOffloadBoth(PETSC_OFFLOAD_GPU), "");
172 static_assert(!PetscOffloadBoth(PETSC_OFFLOAD_GPU), "");
188 PETSC_CASE_RETURN(PETSC_OFFLOAD_GPU); in PetscOffloadMaskToString()
203 case PETSC_OFFLOAD_GPU: in PetscOffloadMaskToMemType()
/petsc/src/vec/vec/tests/
H A Dex65.c31 …PetscCheck(mask == PETSC_OFFLOAD_GPU, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unexpected offload state"); in main()
42 …PetscCheck(mask == PETSC_OFFLOAD_GPU, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Local vector has synced wi… in main()
44 …PetscCheck(mask == PETSC_OFFLOAD_GPU, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Global vector has synced w… in main()
/petsc/src/vec/vec/impls/seq/seqviennacl/
H A Dvecviennacl.cxx34 v->offloadmask = PETSC_OFFLOAD_GPU; in VecViennaCLRestoreArray()
73 v->offloadmask = PETSC_OFFLOAD_GPU; in VecViennaCLRestoreArrayWrite()
237 if (v->offloadmask == PETSC_OFFLOAD_GPU) { in VecViennaCLCopyFromGPU()
586 if (xin->offloadmask == PETSC_OFFLOAD_GPU) { in VecCopy_SeqViennaCL()
610 } else if (yin->offloadmask == PETSC_OFFLOAD_GPU) { in VecCopy_SeqViennaCL()
1042 (*V)->offloadmask = PETSC_OFFLOAD_GPU; in VecCreateSeqViennaCLWithArrays()
1078 vin->offloadmask = PETSC_OFFLOAD_GPU; in VecViennaCLPlaceArray()
1103 vin->offloadmask = PETSC_OFFLOAD_GPU; in VecViennaCLResetArray()
1150 if (v->offloadmask == PETSC_OFFLOAD_GPU) { in VecGetArray_SeqViennaCL()
/petsc/include/petsc/private/
H A Dveccupmimpl.h407 v->offloadmask = (vimp && vimp->array) ? PETSC_OFFLOAD_CPU : PETSC_OFFLOAD_GPU; in DeviceAllocateCheck_()
435 if (v->offloadmask == PETSC_OFFLOAD_GPU) { in CopyToHost_()
584 … if (((oldmask == PETSC_OFFLOAD_GPU) && hostmem) || ((oldmask == PETSC_OFFLOAD_CPU) && !hostmem)) { in GetArray()
613 mask = hostmem ? PETSC_OFFLOAD_CPU : PETSC_OFFLOAD_GPU; in GetArray()
645 v->offloadmask = PetscMemTypeHost(mtype) ? PETSC_OFFLOAD_CPU : PETSC_OFFLOAD_GPU; in RestoreArray()
738 if (a) v->offloadmask = PETSC_OFFLOAD_GPU; in PlaceArray()
784 if (aptr) v->offloadmask = PETSC_OFFLOAD_GPU; in ReplaceArray()
829 v->offloadmask = PETSC_OFFLOAD_GPU; in ResetArray()
924 v->offloadmask = device_array ? PETSC_OFFLOAD_GPU : PETSC_OFFLOAD_UNALLOCATED; in Initialize_CUPMBase()
H A Dmatdensecupmimpl.h561 A->offloadmask = PETSC_OFFLOAD_GPU; in MatDenseCUPMRestoreArray_Private()
622 A->offloadmask = PETSC_OFFLOAD_GPU; in MatDenseCUPMPlaceArray()
633 A->offloadmask = PETSC_OFFLOAD_GPU; in MatDenseCUPMReplaceArray()
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/
H A Dpetscdevice.pxi8 PETSC_OFFLOAD_GPU
H A DVec.pyx1575 - 2: ``PETSC_OFFLOAD_GPU`` - GPU has valid entries
/petsc/src/vec/vec/interface/
H A Drvector.c1743 } else if (ymask == PETSC_OFFLOAD_GPU) { in VecRestoreSubVector()
1744 X->offloadmask = PETSC_OFFLOAD_GPU; in VecRestoreSubVector()
1747 case PETSC_OFFLOAD_GPU: in VecRestoreSubVector()
1751 if (ymask == PETSC_OFFLOAD_GPU) PetscCall(VecResetArray(*Y)); in VecRestoreSubVector()
1769 } else if (ymask == PETSC_OFFLOAD_GPU) { in VecRestoreSubVector()
1770 X->offloadmask = PETSC_OFFLOAD_GPU; in VecRestoreSubVector()
1773 case PETSC_OFFLOAD_GPU: in VecRestoreSubVector()
1777 if (ymask == PETSC_OFFLOAD_GPU) PetscCall(VecResetArray(*Y)); in VecRestoreSubVector()
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp229 m->offloadmask = PETSC_OFFLOAD_GPU; in SetPreallocation_()
276 const auto copy = m->offloadmask == PETSC_OFFLOAD_GPU; in DeviceToHost_()
1184 m->offloadmask = hostmem ? PETSC_OFFLOAD_CPU : PETSC_OFFLOAD_GPU; in GetArray()
1198 m->offloadmask = PetscMemTypeHost(mtype) ? PETSC_OFFLOAD_CPU : PETSC_OFFLOAD_GPU; in RestoreArray()
1863 cmat->offloadmask = PETSC_OFFLOAD_GPU; in GetSubMatrix()
1903 A->offloadmask = was_offload_host ? PETSC_OFFLOAD_CPU : PETSC_OFFLOAD_GPU; in RestoreSubMatrix()
/petsc/src/vec/vec/impls/mpi/mpiviennacl/
H A Dmpiviennacl.cxx353 (*vv)->offloadmask = PETSC_OFFLOAD_GPU; in VecCreateMPIViennaCLWithArrays()
/petsc/src/mat/impls/aij/seq/seqviennacl/
H A Daijviennacl.cxx157 if (!Agpu && A->offloadmask != PETSC_OFFLOAD_GPU) PetscFunctionReturn(PETSC_SUCCESS); in MatViennaCLCopyFromGPU()
/petsc/src/mat/impls/h2opus/cuda/
H A Dmath2opus.cu941 A->offloadmask = boundtocpu ? PETSC_OFFLOAD_CPU : PETSC_OFFLOAD_GPU; in MatAssemblyEnd_H2OPUS()
1216 if (flg && A->offloadmask == PETSC_OFFLOAD_GPU) { in MatBindToCPU_H2OPUS()
1249 A->offloadmask = PETSC_OFFLOAD_GPU; in MatBindToCPU_H2OPUS()
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu1749 fact->offloadmask = PETSC_OFFLOAD_GPU; in MatILUFactorNumeric_SeqAIJCUSPARSE_ILU0()
1999 fact->offloadmask = PETSC_OFFLOAD_GPU; in MatICCFactorNumeric_SeqAIJCUSPARSE_ICC0()
2325 if (A->offloadmask == PETSC_OFFLOAD_GPU) { in MatSeqAIJCUSPARSECopyFromGPU()
3010 C->offloadmask = PETSC_OFFLOAD_GPU; in MatProductNumeric_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3387 C->offloadmask = PETSC_OFFLOAD_GPU; in MatProductSymbolic_SeqAIJCUSPARSE_SeqAIJCUSPARSE()
3762 if (A->offloadmask == PETSC_OFFLOAD_BOTH || A->offloadmask == PETSC_OFFLOAD_GPU) { in MatGetDiagonal_SeqAIJCUSPARSE()
3989 if (gpu) A->offloadmask = PETSC_OFFLOAD_GPU; in MatZeroEntries_SeqAIJCUSPARSE()
4630 A->offloadmask = PETSC_OFFLOAD_GPU; in MatSeqAIJCUSPARSEGetArray()
4692 A->offloadmask = PETSC_OFFLOAD_GPU; in MatSeqAIJCUSPARSEGetArrayWrite()
5029 (*C)->offloadmask = PETSC_OFFLOAD_GPU; in MatSeqAIJCUSPARSEMergeMats()
/petsc/src/mat/impls/aij/seq/seqhipsparse/
H A Daijhipsparse.hip.cxx1384 fact->offloadmask = PETSC_OFFLOAD_GPU; in MatILUFactorNumeric_SeqAIJHIPSPARSE_ILU0()
1629 fact->offloadmask = PETSC_OFFLOAD_GPU; in MatICCFactorNumeric_SeqAIJHIPSPARSE_ICC0()
1913 if (A->offloadmask == PETSC_OFFLOAD_GPU) { in MatSeqAIJHIPSPARSECopyFromGPU()
2508 C->offloadmask = PETSC_OFFLOAD_GPU; in MatProductNumeric_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
2860 C->offloadmask = PETSC_OFFLOAD_GPU; in MatProductSymbolic_SeqAIJHIPSPARSE_SeqAIJHIPSPARSE()
3986 A->offloadmask = PETSC_OFFLOAD_GPU; in MatSeqAIJHIPSPARSEGetArray()
4046 A->offloadmask = PETSC_OFFLOAD_GPU; in MatSeqAIJHIPSPARSEGetArrayWrite()
4348 (*C)->offloadmask = PETSC_OFFLOAD_GPU; in MatSeqAIJHIPSPARSEMergeMats()
/petsc/src/mat/impls/aij/mpi/mpihipsparse/
H A Dmpiaijhipsparse.hip.cxx236 mat->offloadmask = PETSC_OFFLOAD_GPU; in MatSetValuesCOO_MPIAIJHIPSPARSE()
/petsc/src/mat/impls/aij/mpi/mpicusparse/
H A Dmpiaijcusparse.cu236 mat->offloadmask = PETSC_OFFLOAD_GPU; in MatSetValuesCOO_MPIAIJCUSPARSE()
/petsc/src/mat/impls/dense/seq/
H A Ddense.c1014 …PetscCheck(A->offloadmask != PETSC_OFFLOAD_GPU, PETSC_COMM_SELF, PETSC_ERR_SUP, "Not implemented"); in MatSOR_SeqDense()
1292 A->offloadmask = PETSC_OFFLOAD_GPU; in MatSetValues_SeqDense()
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm_impl.hpp1456 case PETSC_OFFLOAD_GPU: in CopyAsync()