| /petsc/src/sys/objects/device/impls/cupm/cuda/ |
| H A D | cupmcontext.cu | 23 PetscCall(PetscDeviceContextGetCurrentContextAssertType_Internal(&dctx, PETSC_DEVICE_CUDA)); in PetscCUBLASGetHandle()
34 PetscCall(PetscDeviceContextGetCurrentContextAssertType_Internal(&dctx, PETSC_DEVICE_CUDA)); in PetscCUSOLVERDnGetHandle()
46 PetscCall(PetscDeviceContextGetCurrentContextAssertType_Internal(&dctx, PETSC_DEVICE_CUDA)); in PetscGetCurrentCUDAStream()
|
| /petsc/include/ |
| H A D | petscdevice_cuda.h | 75 …(_p_cublas_stat__ == CUBLAS_STATUS_ALLOC_FAILED)) && PetscDeviceInitialized(PETSC_DEVICE_CUDA)) { \
116 …solver_stat__ == CUSOLVER_STATUS_INTERNAL_ERROR)) && PetscDeviceInitialized(PETSC_DEVICE_CUDA)) { \
133 …ILED) || (_p_cufft_stat__ == CUFFT_ALLOC_FAILED)) && PetscDeviceInitialized(PETSC_DEVICE_CUDA)) { \
150 …urand_stat__ == CURAND_STATUS_ALLOCATION_FAILED)) && PetscDeviceInitialized(PETSC_DEVICE_CUDA)) { \
|
| H A D | petscdevicetypes.h | 256 PETSC_DEVICE_CUDA, enumerator
|
| /petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/cupm/ |
| H A D | blas_cyclic_cupm.cxx | 40 case PETSC_DEVICE_CUDA: in AXPBYCyclic_CUPM_Private()
65 case PETSC_DEVICE_CUDA: in DMVCyclic_CUPM_Private()
90 case PETSC_DEVICE_CUDA: in DSVCyclic_CUPM_Private()
115 case PETSC_DEVICE_CUDA: in TRSVCyclic_CUPM_Private()
140 case PETSC_DEVICE_CUDA: in HEMVCyclic_CUPM_Private()
165 case PETSC_DEVICE_CUDA: in GEMVCyclic_CUPM_Private()
|
| /petsc/src/sys/logging/handler/impls/nvtx/ |
| H A D | lognvtx.c | 16 if (PetscDeviceInitialized(PETSC_DEVICE_CUDA)) { in PetscLogHandlerEventBegin_NVTX()
27 if (PetscDeviceInitialized(PETSC_DEVICE_CUDA)) (void)nvtxRangePop(); in PetscLogHandlerEventEnd_NVTX()
|
| /petsc/src/sys/classes/random/impls/curand/ |
| H A D | curand.c | 84 PetscCall(PetscDeviceInitialize(PETSC_DEVICE_CUDA)); in PetscRandomCreate_CURAND()
85 PetscCall(PetscDeviceContextGetCurrentContextAssertType_Internal(&dctx, PETSC_DEVICE_CUDA)); in PetscRandomCreate_CURAND()
|
| /petsc/src/mat/impls/sell/mpi/mpicuda/ |
| H A D | mpisellcuda.cu | 59 PetscCall(PetscDeviceInitialize(PETSC_DEVICE_CUDA)); in MatConvert_MPISELL_MPISELLCUDA()
84 PetscCall(PetscDeviceInitialize(PETSC_DEVICE_CUDA)); in MatCreate_MPISELLCUDA()
|
| /petsc/src/sys/objects/device/util/ |
| H A D | memory.c | 35 if (PetscDeviceInitialized(PETSC_DEVICE_CUDA)) { in PetscGetMemType()
|
| /petsc/src/ksp/ksp/utils/lmvm/dense/cd_cupm/ |
| H A D | cd_cupm.cxx | 111 case PETSC_DEVICE_CUDA: in MatUpperTriangularSolveInPlace_CUPM()
138 case PETSC_DEVICE_CUDA: in MatUpperTriangularSolveInPlaceCyclic_CUPM()
|
| /petsc/src/vec/vec/impls/mpi/cupm/cuda/ |
| H A D | vecmpicupm.cu | 63 PetscCall(PetscDeviceContextGetCurrentContextAssertType_Internal(&dctx, PETSC_DEVICE_CUDA)); in VecCUDAGetArrays_Private()
|
| /petsc/src/sys/objects/kokkos/ |
| H A D | kinit.kokkos.cxx | 85 …if (dtype == PETSC_DEVICE_CUDA) PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExe… in PetscKokkosInitializeCheck()
|
| /petsc/src/sys/objects/device/interface/ |
| H A D | global_dcontext.cxx | 30 if (dtype == PETSC_DEVICE_CUDA) { in PetscSetDefaultCUPMStreamFromDeviceContext()
|
| H A D | device.cxx | 188 case PETSC_DEVICE_CUDA: in PetscDeviceConfigure()
660 static_assert(Petsc::util::to_underlying(PETSC_DEVICE_CUDA) == 1, ""); in PetscDeviceInitializeFromOptions_Internal()
|
| /petsc/src/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | petscdevice.pxi | 20 PETSC_DEVICE_CUDA
|
| H A D | Device.pyx | 31 ("CUDA" , PETSC_DEVICE_CUDA),
|
| /petsc/include/petsc/private/ |
| H A D | deviceimpl.h | 170 #define PETSC_DEVICE_HARDWARE_DEFAULT_TYPE PETSC_DEVICE_CUDA
287 case PETSC_DEVICE_CUDA: in PetscDeviceConfiguredFor_Internal()
|
| H A D | cupminterface.hpp | 116 …eType PETSC_DEVICE_CUPM() noexcept { return T == DeviceType::CUDA ? PETSC_DEVICE_CUDA : PETSC_DEVI… in PETSC_DEVICE_CUPM()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 461 PetscCall(PetscDeviceInitialize(PETSC_DEVICE_CUDA)); in MatConvert_MPIAIJ_MPIAIJCUSPARSE()
502 PetscCall(PetscDeviceInitialize(PETSC_DEVICE_CUDA)); in MatCreate_MPIAIJCUSPARSE()
|
| /petsc/src/sys/mpiuni/ |
| H A D | mpi.c | 65 if (PetscDeviceInitialized(PETSC_DEVICE_CUDA)) { in MPIUNI_Memcpy()
|
| /petsc/src/vec/is/sf/impls/basic/nvshmem/ |
| H A D | sfnvshmem.cu | 13 PetscCall(PetscDeviceInitialize(PETSC_DEVICE_CUDA)); in PetscNvshmemInitializeCheck()
|
| /petsc/src/mat/impls/h2opus/cuda/ |
| H A D | math2opus.cu | 1285 PetscCall(PetscDeviceInitialize(PETSC_DEVICE_CUDA)); in MatCreate_H2OPUS()
|
| /petsc/src/mat/impls/hypre/ |
| H A D | mhypre.c | 2461 PetscCall(PetscDeviceInitialize(PETSC_DEVICE_CUDA));
|
| /petsc/src/ksp/pc/impls/hypre/ |
| H A D | hypre.c | 2713 PetscCall(PetscDeviceInitialize(PETSC_DEVICE_CUDA)); in PCCreate_HYPRE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 4077 …PetscCall(PetscDeviceInitialize(PETSC_DEVICE_CUDA)); /* first use of CUSPARSE may be via MatConver… in MatConvert_SeqAIJ_SeqAIJCUSPARSE()
|