Home
last modified time | relevance | path

Searched refs:PETSC_MEMTYPE_CUDA (Results 1 – 12 of 12) sorted by relevance

/petsc/include/
H A Dpetscdevicetypes.h48 PETSC_MEMTYPE_CUDA = 1, /* 0x01 */ enumerator
54 #define PETSC_MEMTYPE_KOKKOS PETSC_MEMTYPE_CUDA
65 #define PetscMemTypeCUDA(m) (((m) & 0xF) == PETSC_MEMTYPE_CUDA)
77 static_assert(!PetscMemTypeHost(PETSC_MEMTYPE_CUDA), "");
84 static_assert(PetscMemTypeDevice(PETSC_MEMTYPE_CUDA), "");
89 static_assert(PetscMemTypeCUDA(PETSC_MEMTYPE_CUDA), "");
99 static_assert(PETSC_MEMTYPE_CUDA == PETSC_MEMTYPE_DEVICE, ""); in PetscMemTypeToString()
108 PETSC_CASE_NAME(PETSC_MEMTYPE_CUDA); in PetscMemTypeToString()
/petsc/src/vec/is/sf/impls/basic/nvshmem/
H A Dsfnvshmem.cu60 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, bas->leafbufdisp_d)); in PetscSFReset_Basic_NVSHMEM()
61 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, bas->leafsigdisp_d)); in PetscSFReset_Basic_NVSHMEM()
62 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, bas->iranks_d)); in PetscSFReset_Basic_NVSHMEM()
63 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, bas->ioffset_d)); in PetscSFReset_Basic_NVSHMEM()
66 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, sf->rootbufdisp_d)); in PetscSFReset_Basic_NVSHMEM()
67 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, sf->rootsigdisp_d)); in PetscSFReset_Basic_NVSHMEM()
68 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, sf->ranks_d)); in PetscSFReset_Basic_NVSHMEM()
69 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_CUDA, sf->roffset_d)); in PetscSFReset_Basic_NVSHMEM()
/petsc/src/mat/tests/
H A Dex217.c30 …if (iscuda) PetscCheck(memtype == PETSC_MEMTYPE_CUDA, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "wrong mem… in main()
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/
H A Dpetscdevice.pxi14 PETSC_MEMTYPE_CUDA
H A Dpetscvec.pxi482 …devType_ = {PETSC_MEMTYPE_HOST : kDLCPU, PETSC_MEMTYPE_CUDA : kDLCUDA, PETSC_MEMTYPE_HIP : kDLROCM}
H A Dpetscmat.pxi1248 …devType_ = {PETSC_MEMTYPE_HOST : kDLCPU, PETSC_MEMTYPE_CUDA : kDLCUDA, PETSC_MEMTYPE_HIP : kDLROCM}
/petsc/src/mat/impls/aij/mpi/mpicusparse/
H A Dmpiaijcusparse.cu213 …l(PetscSFReduceWithMemTypeBegin(coo->sf, MPIU_SCALAR, PETSC_MEMTYPE_CUDA, vsend, PETSC_MEMTYPE_CUD… in MatSetValuesCOO_MPIAIJCUSPARSE()
/petsc/src/ksp/ksp/impls/hpddm/
H A Dhpddm.cxx359 …if (PetscMemTypeCUDA(type[0])) PetscCall(KSPSolve_HPDDM_Private<PETSC_MEMTYPE_CUDA>(ksp, b, x, 1)); in KSPSolve_HPDDM()
532 …if (PetscMemTypeCUDA(type[0])) PetscCall(KSPSolve_HPDDM_Private<PETSC_MEMTYPE_CUDA>(ksp, b, x, n)); in KSPMatSolve_HPDDM()
/petsc/src/binding/petsc4py/src/lib-petsc/
H A Dcustom.h184 if (iscuda) *m = PETSC_MEMTYPE_CUDA; in VecGetCurrentMemType()
/petsc/include/petsc/private/
H A Dcupminterface.hpp118 …ype PETSC_MEMTYPE_CUPM() noexcept { return T == DeviceType::CUDA ? PETSC_MEMTYPE_CUDA : PETSC_MEMT… in PETSC_MEMTYPE_CUPM()
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu2419 if (mtype) *mtype = PETSC_MEMTYPE_CUDA; in MatSeqAIJGetCSRAndMemType_SeqAIJCUSPARSE()
4000 *m = PETSC_MEMTYPE_CUDA; in MatGetCurrentMemType_SeqAIJCUSPARSE()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.c7391 if (iscuda) mmdata->mtype = PETSC_MEMTYPE_CUDA; in MatProductSymbolic_MPIAIJBACKEND()