Searched refs:PetscMemTypeCUDA (Results 1 – 4 of 4) sorted by relevance
65 #define PetscMemTypeCUDA(m) (((m) & 0xF) == PETSC_MEMTYPE_CUDA) macro89 static_assert(PetscMemTypeCUDA(PETSC_MEMTYPE_CUDA), "");90 static_assert(PetscMemTypeCUDA(PETSC_MEMTYPE_NVSHMEM), "");
359 …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()
191 …PetscInt oneCuda = (!rootdata || PetscMemTypeCUDA(rootmtype)) && (!leafdata || PetscMemTypeCUDA(le… in PetscSFLinkNvshmemCheck()
1149 …else if (PetscMemTypeCUDA(mtype)) PetscCallCUDA(cudaMalloc((void **)&array, sizeof(PetscScalar) * … in PCSetUpOnBlocks_FieldSplit_Schur()1288 …else if (PetscMemTypeCUDA(mtype)) PetscCall(VecCreateMPICUDAWithArray(PetscObjectComm((PetscObject… in PCApply_FieldSplit_Schur()1485 else if (PetscMemTypeCUDA(mtype)) { in PCMatApply_FieldSplit_Schur()1502 …else if (PetscMemTypeCUDA(mtype)) PetscCall(MatCreateDenseCUDA(PetscObjectComm((PetscObject)jac->s… in PCMatApply_FieldSplit_Schur()