| /petsc/src/vec/is/sf/tests/ |
| H A D | ex22.c | 15 PetscScalar *yarray, *zarray; in main() local
40 PetscCall(VecGetArrayWriteAndMemType(z, &zarray, &zmtype)); in main()
42 …ndOpWithMemTypeBegin(vscat, MPIU_SCALAR, ymtype, yarray, xmtype, xarray, zmtype, zarray, MPI_SUM)); in main()
43 PetscCall(PetscSFFetchAndOpEnd(vscat, MPIU_SCALAR, yarray, xarray, zarray, MPI_SUM)); in main()
47 PetscCall(VecRestoreArrayWriteAndMemType(z, &zarray)); in main()
|
| /petsc/src/mat/impls/baij/seq/ |
| H A D | baij2.c | 284 PetscScalar *z = NULL, sum1, sum2, *zarray; in MatMult_SeqBAIJ_2() local
293 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_2()
301 PetscCall(PetscArrayzero(zarray, 2 * a->mbs)); in MatMult_SeqBAIJ_2()
305 z = zarray; in MatMult_SeqBAIJ_2()
323 if (usecprow) z = zarray + 2 * ridx[i]; in MatMult_SeqBAIJ_2()
329 PetscCall(VecRestoreArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_2()
337 PetscScalar *z = NULL, sum1, sum2, sum3, x1, x2, x3, *zarray; in MatMult_SeqBAIJ_3() local
349 PetscCall(VecGetArrayWrite(zz, &zarray)); in MatMult_SeqBAIJ_3()
357 PetscCall(PetscArrayzero(zarray, 3 * a->mbs)); in MatMult_SeqBAIJ_3()
361 z = zarray; in MatMult_SeqBAIJ_3()
[all …]
|
| /petsc/src/mat/impls/scalapack/ |
| H A D | matscalapack.c | 292 PetscScalar *zarray; in MatMultAdd_ScaLAPACK() local
297 PetscCall(VecGetArray(z, &zarray)); in MatMultAdd_ScaLAPACK()
298 PetscCall(MatMultXXXYYY_ScaLAPACK(A, PETSC_FALSE, PETSC_FALSE, 1.0, xarray, zarray)); in MatMultAdd_ScaLAPACK()
300 PetscCall(VecRestoreArray(z, &zarray)); in MatMultAdd_ScaLAPACK()
307 PetscScalar *zarray; in MatMultTransposeAdd_ScaLAPACK() local
312 PetscCall(VecGetArray(z, &zarray)); in MatMultTransposeAdd_ScaLAPACK()
313 PetscCall(MatMultXXXYYY_ScaLAPACK(A, PETSC_TRUE, PETSC_FALSE, 1.0, xarray, zarray)); in MatMultTransposeAdd_ScaLAPACK()
315 PetscCall(VecRestoreArray(z, &zarray)); in MatMultTransposeAdd_ScaLAPACK()
322 PetscScalar *zarray; in MatMultHermitianTransposeAdd_ScaLAPACK() local
327 PetscCall(VecGetArray(z, &zarray)); in MatMultHermitianTransposeAdd_ScaLAPACK()
[all …]
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 3026 PetscScalar *xarray, *zarray, *dptr, *beta, *xptr; in MatMultAddKernel_SeqAIJHIPSPARSE() local
3056 …if (yy == zz) PetscCall(VecHIPGetArray(zz, &zarray)); /* read & write zz, so need to get up-to-dat… in MatMultAddKernel_SeqAIJHIPSPARSE()
3057 …else PetscCall(VecHIPGetArrayWrite(zz, &zarray)); /* write zz, so no need to init zarray on GP… in MatMultAddKernel_SeqAIJHIPSPARSE()
3066 dptr = compressed ? hipsparsestruct->workVector->data().get() : zarray; in MatMultAddKernel_SeqAIJHIPSPARSE()
3082 dptr = zarray; in MatMultAddKernel_SeqAIJHIPSPARSE()
3150 thrust::device_ptr<PetscScalar> zptr = thrust::device_pointer_cast(zarray); in MatMultAddKernel_SeqAIJHIPSPARSE()
3157 …ream, n, matstruct->cprowIndices->data().get(), hipsparsestruct->workVector->data().get(), zarray); in MatMultAddKernel_SeqAIJHIPSPARSE()
3165 if (yy == zz) PetscCall(VecHIPRestoreArray(zz, &zarray)); in MatMultAddKernel_SeqAIJHIPSPARSE()
3166 else PetscCall(VecHIPRestoreArrayWrite(zz, &zarray)); in MatMultAddKernel_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 3560 PetscScalar *xarray, *zarray, *dptr, *beta, *xptr; in MatMultAddKernel_SeqAIJCUSPARSE() local
3593 …if (yy == zz) PetscCall(VecCUDAGetArray(zz, &zarray)); /* read & write zz, so need to get up-to-da… in MatMultAddKernel_SeqAIJCUSPARSE()
3594 …else PetscCall(VecCUDAGetArrayWrite(zz, &zarray)); /* write zz, so no need to init zarray on G… in MatMultAddKernel_SeqAIJCUSPARSE()
3603 dptr = compressed ? cusparsestruct->workVector->data().get() : zarray; in MatMultAddKernel_SeqAIJCUSPARSE()
3621 dptr = zarray; in MatMultAddKernel_SeqAIJCUSPARSE()
3708 …eam>>>(n, matstruct->cprowIndices->data().get(), cusparsestruct->workVector->data().get(), zarray); in MatMultAddKernel_SeqAIJCUSPARSE()
3715 if (yy == zz) PetscCall(VecCUDARestoreArray(zz, &zarray)); in MatMultAddKernel_SeqAIJCUSPARSE()
3716 else PetscCall(VecCUDARestoreArrayWrite(zz, &zarray)); in MatMultAddKernel_SeqAIJCUSPARSE()
|