| /petsc/src/vec/vec/impls/mpi/cupm/cuda/ |
| H A D | vecmpicupm.cu | 130 …Int bs, PetscInt n, PetscInt N, const PetscScalar cpuarray[], const PetscScalar gpuarray[], Vec *v) in VecCreateMPICUDAWithArrays() argument
133 PetscCall(VecCreateMPICUPMWithArrays<DeviceType::CUDA>(comm, bs, n, N, cpuarray, gpuarray, v)); in VecCreateMPICUDAWithArrays()
164 …WithArray(MPI_Comm comm, PetscInt bs, PetscInt n, PetscInt N, const PetscScalar gpuarray[], Vec *v) in VecCreateMPICUDAWithArray() argument
167 PetscCall(VecCreateMPICUPMWithArray<DeviceType::CUDA>(comm, bs, n, N, gpuarray, v)); in VecCreateMPICUDAWithArray()
|
| /petsc/src/vec/vec/impls/mpi/cupm/hip/ |
| H A D | vecmpicupm.hip.cxx | 130 …Int bs, PetscInt n, PetscInt N, const PetscScalar cpuarray[], const PetscScalar gpuarray[], Vec *v) in VecCreateMPIHIPWithArrays() argument
133 PetscCall(VecCreateMPICUPMWithArrays<DeviceType::HIP>(comm, bs, n, N, cpuarray, gpuarray, v)); in VecCreateMPIHIPWithArrays()
164 …WithArray(MPI_Comm comm, PetscInt bs, PetscInt n, PetscInt N, const PetscScalar gpuarray[], Vec *v) in VecCreateMPIHIPWithArray() argument
167 PetscCall(VecCreateMPICUPMWithArray<DeviceType::HIP>(comm, bs, n, N, gpuarray, v)); in VecCreateMPIHIPWithArray()
|
| /petsc/src/vec/vec/impls/seq/cupm/cuda/ |
| H A D | vecseqcupm.cu | 104 … comm, PetscInt bs, PetscInt n, const PetscScalar cpuarray[], const PetscScalar gpuarray[], Vec *v) in VecCreateSeqCUDAWithArrays() argument
107 PetscCall(VecCreateSeqCUPMWithArraysAsync<DeviceType::CUDA>(comm, bs, n, cpuarray, gpuarray, v)); in VecCreateSeqCUDAWithArrays()
146 …reateSeqCUDAWithArray(MPI_Comm comm, PetscInt bs, PetscInt n, const PetscScalar gpuarray[], Vec *v) in VecCreateSeqCUDAWithArray() argument
149 PetscCall(VecCreateSeqCUDAWithArrays(comm, bs, n, nullptr, gpuarray, v)); in VecCreateSeqCUDAWithArray()
|
| /petsc/src/vec/vec/impls/seq/cupm/hip/ |
| H A D | vecseqcupm.hip.cxx | 104 … comm, PetscInt bs, PetscInt n, const PetscScalar cpuarray[], const PetscScalar gpuarray[], Vec *v) in VecCreateSeqHIPWithArrays() argument
107 PetscCall(VecCreateSeqCUPMWithArraysAsync<DeviceType::HIP>(comm, bs, n, cpuarray, gpuarray, v)); in VecCreateSeqHIPWithArrays()
146 …CreateSeqHIPWithArray(MPI_Comm comm, PetscInt bs, PetscInt n, const PetscScalar gpuarray[], Vec *v) in VecCreateSeqHIPWithArray() argument
149 PetscCall(VecCreateSeqHIPWithArrays(comm, bs, n, nullptr, gpuarray, v)); in VecCreateSeqHIPWithArray()
|
| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm.hpp | 68 …Int bs, PetscInt n, PetscInt N, const PetscScalar cpuarray[], const PetscScalar gpuarray[], Vec *v) in VecCreateMPICUPMWithArrays() argument
73 PetscCall(impl::VecMPI_CUPM<T>::CreateMPICUPMWithArrays(comm, bs, n, N, cpuarray, gpuarray, v)); in VecCreateMPICUPMWithArrays()
78 …WithArray(MPI_Comm comm, PetscInt bs, PetscInt n, PetscInt N, const PetscScalar gpuarray[], Vec *v) in VecCreateMPICUPMWithArray() argument
81 PetscCall(VecCreateMPICUPMWithArrays<T>(comm, bs, n, N, nullptr, gpuarray, v)); in VecCreateMPICUPMWithArray()
|
| /petsc/src/binding/petsc4py/src/lib-petsc/compat/ |
| H A D | cuda.h | 11 …SC_UNUSED PetscInt M, PETSC_UNUSED PetscInt N, PETSC_UNUSED PetscScalar gpuarray[], PETSC_UNUSED M… in MatCreateDenseCUDA() argument
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm.hpp | 190 …etscInt bs, PetscInt n, const PetscScalar cpuarray[], const PetscScalar gpuarray[], Vec *v) noexce… in VecCreateSeqCUPMWithArraysAsync() argument
195 PetscCall(impl::VecSeq_CUPM<T>::CreateSeqCUPMWithBothArrays(comm, bs, n, cpuarray, gpuarray, v)); in VecCreateSeqCUPMWithArraysAsync()
|
| /petsc/src/binding/petsc4py/src/petsc4py/PETSc/ |
| H A D | Vec.pyx | 402 cdef PetscScalar *gpuarray = NULL
404 gpuarray = <PetscScalar*>(<Py_uintptr_t>cudahandle)
419 CHKERR(VecCreateSeqCUDAWithArrays(ccomm, bs, N, sa, gpuarray, &newvec))
421 CHKERR(VecCreateMPICUDAWithArrays(ccomm, bs, n, N, sa, gpuarray, &newvec))
460 cdef PetscScalar *gpuarray = NULL
462 gpuarray = <PetscScalar*>(<Py_uintptr_t>hiphandle)
477 CHKERR(VecCreateSeqHIPWithArrays(ccomm, bs, N, sa, gpuarray, &newvec))
479 CHKERR(VecCreateMPIHIPWithArrays(ccomm, bs, n, N, sa, gpuarray, &newvec))
|