| /petsc/src/vec/vec/impls/mpi/cupm/ |
| H A D | vecmpicupm.hpp | 68 …s(MPI_Comm comm, PetscInt bs, PetscInt n, PetscInt N, const PetscScalar cpuarray[], const PetscSca… in VecCreateMPICUPMWithArrays() argument
71 if (n && cpuarray) PetscAssertPointer(cpuarray, 5); in VecCreateMPICUPMWithArrays()
73 PetscCall(impl::VecMPI_CUPM<T>::CreateMPICUPMWithArrays(comm, bs, n, N, cpuarray, gpuarray, v)); in VecCreateMPICUPMWithArrays()
|
| /petsc/src/binding/petsc4py/src/lib-petsc/compat/ |
| H A D | viennacl.h | 11 …TSC_UNUSED PetscInt bs,PETSC_UNUSED PetscInt n,PETSC_UNUSED PetscScalar cpuarray[],PETSC_UNUSED Pe… in VecCreateSeqViennaCLWithArrays() argument
12 …ETSC_UNUSED PetscInt n,PETSC_UNUSED PetscInt N,PETSC_UNUSED PetscScalar cpuarray[],PETSC_UNUSED Pe… in VecCreateMPIViennaCLWithArrays() argument
|
| /petsc/src/vec/vec/impls/mpi/cupm/cuda/ |
| H A D | vecmpicupm.cu | 130 …s(MPI_Comm comm, PetscInt bs, PetscInt n, PetscInt N, const PetscScalar cpuarray[], const PetscSca… in VecCreateMPICUDAWithArrays() argument
133 PetscCall(VecCreateMPICUPMWithArrays<DeviceType::CUDA>(comm, bs, n, N, cpuarray, gpuarray, v)); in VecCreateMPICUDAWithArrays()
|
| /petsc/src/vec/vec/impls/mpi/cupm/hip/ |
| H A D | vecmpicupm.hip.cxx | 130 …s(MPI_Comm comm, PetscInt bs, PetscInt n, PetscInt N, const PetscScalar cpuarray[], const PetscSca… in VecCreateMPIHIPWithArrays() argument
133 PetscCall(VecCreateMPICUPMWithArrays<DeviceType::HIP>(comm, bs, n, N, cpuarray, gpuarray, v)); in VecCreateMPIHIPWithArrays()
|
| /petsc/src/vec/vec/impls/seq/cupm/cuda/ |
| H A D | vecseqcupm.cu | 104 …UDAWithArrays(MPI_Comm comm, PetscInt bs, PetscInt n, const PetscScalar cpuarray[], const PetscSca… in VecCreateSeqCUDAWithArrays() argument
107 PetscCall(VecCreateSeqCUPMWithArraysAsync<DeviceType::CUDA>(comm, bs, n, cpuarray, gpuarray, v)); in VecCreateSeqCUDAWithArrays()
|
| /petsc/src/vec/vec/impls/seq/cupm/hip/ |
| H A D | vecseqcupm.hip.cxx | 104 …HIPWithArrays(MPI_Comm comm, PetscInt bs, PetscInt n, const PetscScalar cpuarray[], const PetscSca… in VecCreateSeqHIPWithArrays() argument
107 PetscCall(VecCreateSeqCUPMWithArraysAsync<DeviceType::HIP>(comm, bs, n, cpuarray, gpuarray, v)); in VecCreateSeqHIPWithArrays()
|
| /petsc/src/vec/vec/impls/mpi/mpiviennacl/ |
| H A D | mpiviennacl.cxx | 340 …s(MPI_Comm comm, PetscInt bs, PetscInt n, PetscInt N, const PetscScalar cpuarray[], const ViennaCL… in VecCreateMPIViennaCLWithArrays() argument
344 if (cpuarray && viennaclvec) { in VecCreateMPIViennaCLWithArrays()
346 s->array = (PetscScalar *)cpuarray; in VecCreateMPIViennaCLWithArrays()
348 } else if (cpuarray) { in VecCreateMPIViennaCLWithArrays()
350 s->array = (PetscScalar *)cpuarray; in VecCreateMPIViennaCLWithArrays()
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm.hpp | 190 …thArraysAsync(MPI_Comm comm, PetscInt bs, PetscInt n, const PetscScalar cpuarray[], const PetscSca… in VecCreateSeqCUPMWithArraysAsync() argument
193 if (n && cpuarray) PetscAssertPointer(cpuarray, 4); in VecCreateSeqCUPMWithArraysAsync()
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 | 372 cpuarray: Sequence[Scalar] | None = None,
383 cpuarray
405 if cpuarray is not None:
406 cpuarray = iarray_s(cpuarray, &na, &sa)
424 if cpuarray is not None:
425 self.set_attr('__array__', cpuarray)
430 cpuarray: Sequence[Scalar] | None = None,
441 cpuarray
463 if cpuarray is not None:
464 cpuarray = iarray_s(cpuarray, &na, &sa)
[all …]
|
| /petsc/src/vec/vec/impls/seq/seqviennacl/ |
| H A D | vecviennacl.cxx | 1022 …aCLWithArrays(MPI_Comm comm, PetscInt bs, PetscInt n, const PetscScalar cpuarray[], const ViennaCL… in VecCreateSeqViennaCLWithArrays() argument
1033 if (cpuarray && viennaclvec) { in VecCreateSeqViennaCLWithArrays()
1035 s->array = (PetscScalar *)cpuarray; in VecCreateSeqViennaCLWithArrays()
1037 } else if (cpuarray) { in VecCreateSeqViennaCLWithArrays()
1039 s->array = (PetscScalar *)cpuarray; in VecCreateSeqViennaCLWithArrays()
|