Searched refs:aptr (Results 1 – 5 of 5) sorted by relevance
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 857 …scInt size, PetscScalar *PETSC_RESTRICT xptr, const PetscScalar *PETSC_RESTRICT aptr, Args... yptr) in MAXPY_kernel() argument
866 if (tx < N) aptr_shmem[tx] = aptr[tx]; in MAXPY_kernel()
907 …eContext dctx, cupmStream_t stream, PetscScalar *xptr, const PetscScalar *aptr, const Vec *yin, Pe… in MAXPY_kernel_dispatch_() argument
915 size, xptr, aptr, DeviceArrayRead(dctx, yin[Idx]).data()... in MAXPY_kernel_dispatch_()
924 …eContext dctx, cupmStream_t stream, PetscScalar *xptr, const PetscScalar *aptr, const Vec *yin, Pe… in MAXPY_kernel_dispatch_() argument
927 …PetscCall(MAXPY_kernel_dispatch_(dctx, stream, xptr, aptr + yidx, yin + yidx, size, util::make_ind… in MAXPY_kernel_dispatch_()
1568 const auto aptr = cupmScalarPtrCast(&alpha); in AXPBYAsync() local
1583 PetscCallCUPMBLAS(cupmBlasXscal(cupmBlasHandle, n, aptr, yptr.cupmdata(), 1)); in AXPBYAsync()
1589 … PetscCallCUPMBLAS(cupmBlasXaxpy(cupmBlasHandle, n, aptr, xptr.cupmdata(), 1, yptr.cupmdata(), 1)); in AXPBYAsync()
1874 const auto aptr = ConditionalDeviceArrayRead(vatol); in ErrorWnorm() local
[all …]
|
| /petsc/include/petsc/private/ |
| H A D | veccupmimpl.h | 749 const auto aptr = const_cast<PetscScalar *>(a); in ReplaceArray() local
771 host_array = aptr; in ReplaceArray()
781 PetscCall(ResetAllocatedDevicePtr_(dctx, v, aptr)); in ReplaceArray()
784 if (aptr) v->offloadmask = PETSC_OFFLOAD_GPU; in ReplaceArray()
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aij.c | 2249 PetscInt *adx, *bdx, *aii, *bii, *aptr, *bptr; in MatIsTranspose_SeqAIJ() local
2266 PetscCall(PetscMalloc1(ma, &aptr)); in MatIsTranspose_SeqAIJ()
2268 for (i = 0; i < ma; i++) aptr[i] = aii[i]; in MatIsTranspose_SeqAIJ()
2273 while (aptr[i] < aii[i + 1]) { in MatIsTranspose_SeqAIJ()
2277 idc = adx[aptr[i]]; in MatIsTranspose_SeqAIJ()
2279 vc = va[aptr[i]]; in MatIsTranspose_SeqAIJ()
2285 aptr[i]++; in MatIsTranspose_SeqAIJ()
2291 PetscCall(PetscFree(aptr)); in MatIsTranspose_SeqAIJ()
2301 PetscInt *adx, *bdx, *aii, *bii, *aptr, *bptr; in MatIsHermitianTranspose_SeqAIJ() local
2318 PetscCall(PetscMalloc1(ma, &aptr)); in MatIsHermitianTranspose_SeqAIJ()
[all …]
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 1493 const auto aptr = thrust::device_pointer_cast(da.data()); in Conjugate()
1498 aptr, aptr + N, aptr, in Conjugate()
|
| /petsc/src/vec/vec/impls/seq/ |
| H A D | dvec2.c | 670 #pragma disjoint(*xx, **yptr, *aptr) in VecMAXPY_Seq()
|