Home
last modified time | relevance | path

Searched refs:PETSC_MEMTYPE_HIP (Results 1 – 10 of 10) sorted by relevance

/petsc/include/
H A Dpetscdevicetypes.h50 PETSC_MEMTYPE_HIP = 3, /* 0x03 */ enumerator
56 #define PETSC_MEMTYPE_KOKKOS PETSC_MEMTYPE_HIP
66 #define PetscMemTypeHIP(m) (((m) & 0xF) == PETSC_MEMTYPE_HIP)
78 static_assert(!PetscMemTypeHost(PETSC_MEMTYPE_HIP), "");
85 static_assert(PetscMemTypeDevice(PETSC_MEMTYPE_HIP), "");
110 PETSC_CASE_NAME(PETSC_MEMTYPE_HIP); in PetscMemTypeToString()
/petsc/src/mat/tests/
H A Dex217.c31 …else if (iship) PetscCheck(memtype == PETSC_MEMTYPE_HIP, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "wrong … in main()
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/
H A Dpetscdevice.pxi15 PETSC_MEMTYPE_HIP
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/mpihipsparse/
H A Dmpiaijhipsparse.hip.cxx213 …ll(PetscSFReduceWithMemTypeBegin(coo->sf, MPIU_SCALAR, PETSC_MEMTYPE_HIP, vsend, PETSC_MEMTYPE_HIP… in MatSetValuesCOO_MPIAIJHIPSPARSE()
/petsc/src/binding/petsc4py/src/lib-petsc/
H A Dcustom.h185 else if (iship) *m = PETSC_MEMTYPE_HIP; in VecGetCurrentMemType()
/petsc/include/petsc/private/
H A Dcupminterface.hpp118 …_MEMTYPE_CUPM() noexcept { return T == DeviceType::CUDA ? PETSC_MEMTYPE_CUDA : PETSC_MEMTYPE_HIP; } in PETSC_MEMTYPE_CUPM()
/petsc/src/mat/impls/aij/seq/seqhipsparse/
H A Daijhipsparse.hip.cxx2007 if (mtype) *mtype = PETSC_MEMTYPE_HIP; in MatSeqAIJGetCSRAndMemType_SeqAIJHIPSPARSE()
3391 *m = PETSC_MEMTYPE_HIP; in MatGetCurrentMemType_SeqAIJHIPSPARSE()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.c7392 else if (iship) mmdata->mtype = PETSC_MEMTYPE_HIP; in MatProductSymbolic_MPIAIJBACKEND()