Searched refs:mem_type (Results 1 – 4 of 4) sorted by relevance
| /honee/include/ |
| H A D | petsc-ceed-utils.h | 54 …c inline CeedMemType MemTypePetscToCeed(PetscMemType mem_type) { return PetscMemTypeDevice(mem_typ… in MemTypePetscToCeed() argument
125 static inline PetscErrorCode VecPetscToCeed(Vec X_petsc, PetscMemType *mem_type, CeedVector x_ceed)… in VecPetscToCeed() argument
129 PetscCall(VecGetArrayAndMemType(X_petsc, &x, mem_type)); in VecPetscToCeed()
130 …ectorReturnCeed(x_ceed), CeedVectorSetArray(x_ceed, MemTypePetscToCeed(*mem_type), CEED_USE_POINTE… in VecPetscToCeed()
145 static inline PetscErrorCode VecCeedToPetsc(CeedVector x_ceed, PetscMemType mem_type, Vec X_petsc) { in VecCeedToPetsc() argument
149 …lCeed(CeedVectorReturnCeed(x_ceed), CeedVectorTakeArray(x_ceed, MemTypePetscToCeed(mem_type), &x)); in VecCeedToPetsc()
165 static inline PetscErrorCode VecReadPetscToCeed(Vec X_petsc, PetscMemType *mem_type, CeedVector x_c… in VecReadPetscToCeed() argument
169 PetscCall(VecGetArrayReadAndMemType(X_petsc, (const PetscScalar **)&x, mem_type)); in VecReadPetscToCeed()
170 …ectorReturnCeed(x_ceed), CeedVectorSetArray(x_ceed, MemTypePetscToCeed(*mem_type), CEED_USE_POINTE… in VecReadPetscToCeed()
185 static inline PetscErrorCode VecReadCeedToPetsc(CeedVector x_ceed, PetscMemType mem_type, Vec X_pet… in VecReadCeedToPetsc() argument
[all …]
|
| H A D | mat-ceed-impl.h | 16 PetscMemType mem_type; member
|
| /honee/problems/torch/ |
| H A D | sgs_model_torch.cpp | 34 static PetscErrorCode PetscMemTypeToDeviceType(PetscMemType mem_type, torch::DeviceType *device_typ… in PetscMemTypeToDeviceType() argument
36 switch (mem_type) { in PetscMemTypeToDeviceType()
50 …TSC_ERR_SUP, "PetscMemType %s not supported by PyTorch inference", PetscMemTypeToString(mem_type)); in PetscMemTypeToDeviceType()
|
| /honee/src/ |
| H A D | mat-ceed.c | 120 CeedMemType mem_type = CEED_MEM_HOST; in MatCeedAssemblePointBlockDiagonalCOO() local
124 mem_type = MemTypePetscToCeed(mem_type_petsc); in MatCeedAssemblePointBlockDiagonalCOO()
128 PetscCallCeed(ctx->ceed, CeedVectorGetArrayRead(ctx->coo_values_pbd, mem_type, &values)); in MatCeedAssemblePointBlockDiagonalCOO()
759 CeedMemType mem_type = CEED_MEM_HOST; in MatCeedAssembleCOO() local
764 mem_type = MemTypePetscToCeed(mem_type_petsc); in MatCeedAssembleCOO()
768 PetscCallCeed(ctx->ceed, CeedVectorGetArrayRead(ctx->coo_values_full, mem_type, &values)); in MatCeedAssembleCOO()
1440 PetscCall(VecGetArrayReadAndMemType(X, &x, &(*ctx)->mem_type)); in MatCeedContextCreate()
1591 PetscMemType mem_type; in MatGetDiagonal_Ceed() local
1601 PetscCall(VecPetscToCeed(D_loc, &mem_type, ctx->x_loc)); in MatGetDiagonal_Ceed()
1609 PetscCall(VecCeedToPetsc(ctx->x_loc, mem_type, D_loc)); in MatGetDiagonal_Ceed()
[all …]
|