Home
last modified time | relevance | path

Searched refs:xarray (Results 1 – 25 of 34) sorted by relevance

12

/petsc/src/vec/vec/tutorials/
H A Dex21f90.F9088 type(MyStruct), pointer :: xarray(:)
107 PetscCallA(VecGetArrayMyStruct(x, xarray, ierr))
109 xarray(i)%a = i
110 xarray(i)%b = 100*i
111 xarray(i)%c = 10000*i
114 PetscCallA(VecRestoreArrayMyStruct(x, xarray, ierr))
116 PetscCallA(VecGetArrayMyStruct(x, xarray, ierr))
118 write (*, *) abs(xarray(i)%a), abs(xarray(i)%b), abs(xarray(i)%c)
120 PetscCallA(VecRestoreArrayMyStruct(x, xarray, ierr))
H A Dex18.c24 PetscScalar dummy, result = 0, h = 1.0 / numPoints, *xarray; in main() local
69 PetscCall(VecGetArray(x, &xarray)); in main()
72 xarray[k] = (PetscScalar)i * h; in main()
73 xarray[k] = func(xarray[k]); in main()
76 PetscCall(VecRestoreArray(x, &xarray)); in main()
H A Dex18f.F9017 PetscScalar, pointer, dimension(:) :: xarray
64 PetscCallA(VecGetArray(x, xarray, ierr))
67 xarray(k) = real(i)*h
68 xarray(k) = func(xarray(k))
71 PetscCallA(VecRestoreArray(x, xarray, ierr))
/petsc/src/vec/is/sf/tests/
H A Dex22.c14 const PetscScalar *xarray; in main() local
38 PetscCall(VecGetArrayReadAndMemType(x, &xarray, &xmtype)); in main()
42 …PetscCall(PetscSFFetchAndOpWithMemTypeBegin(vscat, MPIU_SCALAR, ymtype, yarray, xmtype, xarray, zm… in main()
43 PetscCall(PetscSFFetchAndOpEnd(vscat, MPIU_SCALAR, yarray, xarray, zarray, MPI_SUM)); in main()
45 PetscCall(VecRestoreArrayReadAndMemType(x, &xarray)); in main()
/petsc/src/vec/vec/impls/seq/
H A Dbvec1.c59 PetscScalar *xarray; in VecScale_Seq() local
63 PetscCall(VecGetArray(xin, &xarray)); in VecScale_Seq()
64 PetscCallBLAS("BLASscal", BLASscal_(&bn, &alpha, xarray, &one)); in VecScale_Seq()
65 PetscCall(VecRestoreArray(xin, &xarray)); in VecScale_Seq()
75 const PetscScalar *xarray; in VecAXPY_Seq() local
82 PetscCall(VecGetArrayRead(xin, &xarray)); in VecAXPY_Seq()
84 PetscCallBLAS("BLASaxpy", BLASaxpy_(&bn, &alpha, xarray, &one, yarray, &one)); in VecAXPY_Seq()
85 PetscCall(VecRestoreArrayRead(xin, &xarray)); in VecAXPY_Seq()
H A Ddvec2.c518 const PetscScalar *xarray, *yarray, *yfirst, *ynext; in VecMultiDot_Seq_GEMV() local
532 PetscCall(VecGetArrayRead(xin, &xarray)); in VecMultiDot_Seq_GEMV()
557 …PetscCallBLAS("BLASgemv", BLASgemv_(trans, &n, &m, &one, yarray, &lda2, xarray, &ione, &zero, z + … in VecMultiDot_Seq_GEMV()
568 PetscCall(VecRestoreArrayRead(xin, &xarray)); in VecMultiDot_Seq_GEMV()
706 const PetscScalar *xfirst, *xnext, *xarray; in VecMAXPY_Seq_GEMV() local
723 xarray = xfirst; in VecMAXPY_Seq_GEMV()
727 lda = xnext - xarray; in VecMAXPY_Seq_GEMV()
729 } else if (lda * (j - i) != xnext - xarray) { // not in the same stride? if so, stop here in VecMAXPY_Seq_GEMV()
741 …PetscCallBLAS("BLASgemv", BLASgemv_("N", &n, &m, &one, xarray, &lda2, alpha + i, &incx, &one, yarr… in VecMAXPY_Seq_GEMV()
/petsc/src/vec/vec/impls/mpi/
H A Dpdvec.c60 const PetscScalar *xarray; in VecView_MPI_ASCII() local
78 PetscCall(VecGetArrayRead(xin, &xarray)); in VecView_MPI_ASCII()
95 if (PetscImaginaryPart(xarray[i]) > 0.0) { in VecView_MPI_ASCII()
96 …viewer, "%18.16e + %18.16ei\n", (double)PetscRealPart(xarray[i]), (double)PetscImaginaryPart(xarra… in VecView_MPI_ASCII()
97 } else if (PetscImaginaryPart(xarray[i]) < 0.0) { in VecView_MPI_ASCII()
98 …iewer, "%18.16e - %18.16ei\n", (double)PetscRealPart(xarray[i]), -(double)PetscImaginaryPart(xarra… in VecView_MPI_ASCII()
100 PetscCall(PetscViewerASCIIPrintf(viewer, "%18.16e\n", (double)PetscRealPart(xarray[i]))); in VecView_MPI_ASCII()
103 PetscCall(PetscViewerASCIIPrintf(viewer, "%18.16e\n", (double)xarray[i])); in VecView_MPI_ASCII()
129 …tf(viewer, "%18.16e %18.16e\n", (double)PetscRealPart(xarray[i]), (double)PetscImaginaryPart(xarra… in VecView_MPI_ASCII()
131 PetscCall(PetscViewerASCIIPrintf(viewer, "%18.16e\n", (double)xarray[i])); in VecView_MPI_ASCII()
[all …]
H A Dpbvec.c294 PetscScalar *xarray; in VecAssemblyEnd_MPI_BTS() local
305 PetscCall(VecGetArray(X, &xarray)); in VecAssemblyEnd_MPI_BTS()
340 xarray[loc] += *recvscalar++; in VecAssemblyEnd_MPI_BTS()
343 xarray[loc] = *recvscalar++; in VecAssemblyEnd_MPI_BTS()
362 for (k = loc; k < loc + bs; k++) xarray[k] += *recvscalar++; in VecAssemblyEnd_MPI_BTS()
365 for (k = loc; k < loc + bs; k++) xarray[k] = *recvscalar++; in VecAssemblyEnd_MPI_BTS()
374 PetscCall(VecRestoreArray(X, &xarray)); in VecAssemblyEnd_MPI_BTS()
/petsc/src/snes/tutorials/
H A Dex47cu.cu84 PetscScalar const *xarray; in ComputeFunction() local
98 PetscCall(VecCUDAGetArrayRead(xlocal, &xarray)); in ComputeFunction()
112 thrust::device_ptr<const PetscScalar>(xarray + xstartshift), in ComputeFunction()
113 thrust::device_ptr<const PetscScalar>(xarray + xstartshift + 1), in ComputeFunction()
114 thrust::device_ptr<const PetscScalar>(xarray + xstartshift - 1), in ComputeFunction()
121 thrust::device_ptr<const PetscScalar>(xarray + lsize - xendshift), in ComputeFunction()
122 thrust::device_ptr<const PetscScalar>(xarray + lsize - xendshift + 1), in ComputeFunction()
123 thrust::device_ptr<const PetscScalar>(xarray + lsize - xendshift - 1), in ComputeFunction()
133 PetscCall(VecCUDARestoreArrayRead(xlocal, &xarray)); in ComputeFunction()
/petsc/src/mat/tests/
H A Dex211.c10 PetscScalar *xarray, *cmaparray; in ISGetSeqIS_SameColDist_Private() local
39 PetscCall(VecGetArray(x, &xarray)); in ISGetSeqIS_SameColDist_Private()
43 xarray[is_idx[i] - cstart] = (PetscScalar)is_idx[i]; in ISGetSeqIS_SameColDist_Private()
47 PetscCall(VecRestoreArray(x, &xarray)); in ISGetSeqIS_SameColDist_Private()
91 PetscCall(VecGetArray(lvec, &xarray)); in ISGetSeqIS_SameColDist_Private()
94 if (PetscRealPart(xarray[i]) > -1.0) { in ISGetSeqIS_SameColDist_Private()
101 PetscCall(VecRestoreArray(lvec, &xarray)); in ISGetSeqIS_SameColDist_Private()
/petsc/src/ksp/pc/impls/kaczmarz/
H A Dkaczmarz.c20 const PetscScalar *vals, *xarray; in PCApply_Kaczmarz() local
30 PetscCall(VecGetArrayRead(x, &xarray)); in PCApply_Kaczmarz()
35 r = xarray[i - xs]; in PCApply_Kaczmarz()
51 r = xarray[i - xs]; in PCApply_Kaczmarz()
66 PetscCall(VecRestoreArrayRead(x, &xarray)); in PCApply_Kaczmarz()
/petsc/src/ksp/pc/impls/rowscalingviennacl/
H A Drowscalingviennacl.cxx84 viennacl::vector<PetscScalar> const *xarray = NULL; in PCApply_ROWSCALINGVIENNACL() local
94 PetscCall(VecViennaCLGetArrayRead(x, &xarray)); in PCApply_ROWSCALINGVIENNACL()
100 *yarray = *xarray; in PCApply_ROWSCALINGVIENNACL()
106 PetscCall(VecViennaCLRestoreArrayRead(x, &xarray)); in PCApply_ROWSCALINGVIENNACL()
/petsc/src/ksp/pc/impls/chowiluviennacl/
H A Dchowiluviennacl.cxx84 viennacl::vector<PetscScalar> const *xarray = NULL; in PCApply_CHOWILUVIENNACL() local
94 PetscCall(VecViennaCLGetArrayRead(x, &xarray)); in PCApply_CHOWILUVIENNACL()
100 *yarray = *xarray; in PCApply_CHOWILUVIENNACL()
106 PetscCall(VecViennaCLRestoreArrayRead(x, &xarray)); in PCApply_CHOWILUVIENNACL()
/petsc/src/ksp/pc/impls/saviennacl/
H A Dsaviennacl.cxx86 viennacl::vector<PetscScalar> const *xarray = NULL; in PCApply_SAVIENNACL() local
95 PetscCall(VecViennaCLGetArrayRead(x, &xarray)); in PCApply_SAVIENNACL()
99 *yarray = *xarray; in PCApply_SAVIENNACL()
105 PetscCall(VecViennaCLRestoreArrayRead(x, &xarray)); in PCApply_SAVIENNACL()
/petsc/src/mat/impls/aij/seq/mkl_pardiso/
H A Dmkl_pardiso.c379 PetscScalar *xarray; in MatSolve_MKL_PARDISO() local
384 PetscCall(VecGetArrayWrite(x, &xarray)); in MatSolve_MKL_PARDISO()
390 if (barray == xarray) { /* if the two vectors share the same memory */ in MatSolve_MKL_PARDISO()
399 …pardiso->nrhs, mat_mkl_pardiso->iparm, &mat_mkl_pardiso->msglvl, (void *)xarray, (void *)work, &ma… in MatSolve_MKL_PARDISO()
404 …at_mkl_pardiso->iparm, &mat_mkl_pardiso->msglvl, (void *)barray, (void *)xarray, &mat_mkl_pardiso-… in MatSolve_MKL_PARDISO()
419 …PetscCall(MatMKLPardisoScatterSchur_Private(mat_mkl_pardiso, xarray, mat_mkl_pardiso->schur_work, … in MatSolve_MKL_PARDISO()
421 …PetscCall(MatMKLPardisoScatterSchur_Private(mat_mkl_pardiso, xarray, mat_mkl_pardiso->schur_work +… in MatSolve_MKL_PARDISO()
424 for (i = 0; i < mat_mkl_pardiso->schur_size; i++) xarray[mat_mkl_pardiso->schur_idxs[i]] = 0.; in MatSolve_MKL_PARDISO()
431 …pardiso->nrhs, mat_mkl_pardiso->iparm, &mat_mkl_pardiso->msglvl, (void *)xarray, (void *)mat_mkl_p… in MatSolve_MKL_PARDISO()
436 PetscCall(VecRestoreArrayWrite(x, &xarray)); in MatSolve_MKL_PARDISO()
[all …]
/petsc/src/mat/impls/aij/mpi/mkl_cpardiso/
H A Dmkl_cpardiso.c379 PetscScalar *xarray; in MatSolve_MKL_CPARDISO() local
384 PetscCall(VecGetArray(x, &xarray)); in MatSolve_MKL_CPARDISO()
390 …_mkl_cpardiso->iparm, &mat_mkl_cpardiso->msglvl, (void *)barray, (void *)xarray, &mat_mkl_cpardiso… in MatSolve_MKL_CPARDISO()
393 PetscCall(VecRestoreArray(x, &xarray)); in MatSolve_MKL_CPARDISO()
402 PetscScalar *xarray; in MatForwardSolve_MKL_CPARDISO() local
407 PetscCall(VecGetArray(x, &xarray)); in MatForwardSolve_MKL_CPARDISO()
413 …_mkl_cpardiso->iparm, &mat_mkl_cpardiso->msglvl, (void *)barray, (void *)xarray, &mat_mkl_cpardiso… in MatForwardSolve_MKL_CPARDISO()
416 PetscCall(VecRestoreArray(x, &xarray)); in MatForwardSolve_MKL_CPARDISO()
425 PetscScalar *xarray; in MatBackwardSolve_MKL_CPARDISO() local
430 PetscCall(VecGetArray(x, &xarray)); in MatBackwardSolve_MKL_CPARDISO()
[all …]
/petsc/src/ksp/ksp/tests/
H A Dex37.c25 PetscScalar *barray, *xarray, *uarray, *array, one = 1.0; in main() local
126 PetscCall(VecGetArray(x, &xarray)); in main()
129 PetscCall(VecPlaceArray(subx, xarray)); in main()
165 PetscCall(VecRestoreArray(x, &xarray)); in main()
/petsc/src/vec/vec/utils/
H A Dvecio.c18 const PetscScalar *xarray; in VecView_Binary() local
34 PetscCall(VecGetArrayRead(vec, &xarray)); in VecView_Binary()
35 PetscCall(PetscViewerBinaryWriteAll(viewer, xarray, n, s, N, PETSC_SCALAR)); in VecView_Binary()
36 PetscCall(VecRestoreArrayRead(vec, &xarray)); in VecView_Binary()
/petsc/src/mat/impls/scalapack/
H A Dmatscalapack.c249 const PetscScalar *xarray; in MatMult_ScaLAPACK() local
253 PetscCall(VecGetArrayRead(x, &xarray)); in MatMult_ScaLAPACK()
255 PetscCall(MatMultXXXYYY_ScaLAPACK(A, PETSC_FALSE, PETSC_FALSE, 0.0, xarray, yarray)); in MatMult_ScaLAPACK()
256 PetscCall(VecRestoreArrayRead(x, &xarray)); in MatMult_ScaLAPACK()
263 const PetscScalar *xarray; in MatMultTranspose_ScaLAPACK() local
267 PetscCall(VecGetArrayRead(x, &xarray)); in MatMultTranspose_ScaLAPACK()
269 PetscCall(MatMultXXXYYY_ScaLAPACK(A, PETSC_TRUE, PETSC_FALSE, 0.0, xarray, yarray)); in MatMultTranspose_ScaLAPACK()
270 PetscCall(VecRestoreArrayRead(x, &xarray)); in MatMultTranspose_ScaLAPACK()
277 const PetscScalar *xarray; in MatMultHermitianTranspose_ScaLAPACK() local
281 PetscCall(VecGetArrayRead(x, &xarray)); in MatMultHermitianTranspose_ScaLAPACK()
[all …]
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu719 PetscScalar *xarray; in MatSolve_SeqAIJCUSPARSE_Cholesky() local
727 PetscCall(VecCUDAGetArrayWrite(x, &xarray)); in MatSolve_SeqAIJCUSPARSE_Cholesky()
729 xGPU = thrust::device_pointer_cast(xarray); in MatSolve_SeqAIJCUSPARSE_Cholesky()
756 PetscCallCUSPARSE(cusparseDnVecSetValues(fs->dnVecDescr_X, xarray)); in MatSolve_SeqAIJCUSPARSE_Cholesky()
767 PetscCall(VecCUDARestoreArrayWrite(x, &xarray)); in MatSolve_SeqAIJCUSPARSE_Cholesky()
1413 PetscScalar *xarray; in MatSolve_SeqAIJCUSPARSE_LU() local
1424 PetscCall(VecCUDAGetArrayWrite(x, &xarray)); in MatSolve_SeqAIJCUSPARSE_LU()
1426 xGPU = thrust::device_pointer_cast(xarray); in MatSolve_SeqAIJCUSPARSE_LU()
1446 PetscCallCUSPARSE(cusparseDnVecSetValues(fs->dnVecDescr_X, xarray)); in MatSolve_SeqAIJCUSPARSE_LU()
1456 PetscCall(VecCUDARestoreArrayWrite(x, &xarray)); in MatSolve_SeqAIJCUSPARSE_LU()
[all …]
/petsc/src/mat/impls/aij/seq/superlu/
H A Dsuperlu.c95 PetscScalar *xarray; in MatSolve_SuperLU_Private() local
121 PetscCall(VecGetArray(x, &xarray)); in MatSolve_SuperLU_Private()
126 ((DNformat *)lu->X.Store)->nzval = (singlecomplex *)xarray; in MatSolve_SuperLU_Private()
129 ((DNformat *)lu->X.Store)->nzval = (doublecomplex *)xarray; in MatSolve_SuperLU_Private()
133 ((DNformat *)lu->X.Store)->nzval = xarray; in MatSolve_SuperLU_Private()
167 PetscCall(VecRestoreArray(x, &xarray)); in MatSolve_SuperLU_Private()
/petsc/src/mat/impls/aij/seq/seqhipsparse/
H A Daijhipsparse.hip.cxx1089 PetscScalar *xarray; in MatSolveTranspose_SeqAIJHIPSPARSE() local
1106 PetscCall(VecHIPGetArrayWrite(xx, &xarray)); in MatSolveTranspose_SeqAIJHIPSPARSE()
1108 xGPU = thrust::device_pointer_cast(xarray); in MatSolveTranspose_SeqAIJHIPSPARSE()
1117 …iFactorT->csrMat->column_indices->data().get(), upTriFactorT->solveInfo, xarray, tempGPU->data().g… in MatSolveTranspose_SeqAIJHIPSPARSE()
1121 …n_indices->data().get(), loTriFactorT->solveInfo, tempGPU->data().get(), xarray, loTriFactorT->sol… in MatSolveTranspose_SeqAIJHIPSPARSE()
1131 PetscCall(VecHIPRestoreArrayWrite(xx, &xarray)); in MatSolveTranspose_SeqAIJHIPSPARSE()
1140 PetscScalar *xarray; in MatSolveTranspose_SeqAIJHIPSPARSE_NaturalOrdering() local
1155 PetscCall(VecHIPGetArrayWrite(xx, &xarray)); in MatSolveTranspose_SeqAIJHIPSPARSE_NaturalOrdering()
1165 …n_indices->data().get(), loTriFactorT->solveInfo, tempGPU->data().get(), xarray, loTriFactorT->sol… in MatSolveTranspose_SeqAIJHIPSPARSE_NaturalOrdering()
1169 PetscCall(VecHIPRestoreArrayWrite(xx, &xarray)); in MatSolveTranspose_SeqAIJHIPSPARSE_NaturalOrdering()
[all …]
/petsc/src/ts/tutorials/network/
H A Dpipes.c236 const PetscScalar *xarray; in WASHSetInitialSolution() local
253 PetscCall(VecGetArrayRead(pipe->x, &xarray)); in WASHSetInitialSolution()
255 for (k = 0; k < nx; k++) (xarr + varoffset)[k] = xarray[k]; in WASHSetInitialSolution()
271 PetscCall(VecRestoreArrayRead(pipe->x, &xarray)); in WASHSetInitialSolution()
/petsc/src/mat/impls/shell/
H A Dshell.c102 const PetscScalar *d, *xarray; in MatShellPreScaleLeft() local
106 PetscCall(VecGetArrayRead(x, &xarray)); in MatShellPreScaleLeft()
108 for (i = 0; i < m; i++) w[i] = PetscConj(d[i]) * xarray[i]; in MatShellPreScaleLeft()
110 PetscCall(VecRestoreArrayRead(x, &xarray)); in MatShellPreScaleLeft()
161 PetscScalar *xarray; in MatShellPostScaleRight() local
164 PetscCall(VecGetArray(x, &xarray)); in MatShellPostScaleRight()
165 for (i = 0; i < m; i++) xarray[i] = PetscConj(d[i]) * xarray[i]; in MatShellPostScaleRight()
167 PetscCall(VecRestoreArray(x, &xarray)); in MatShellPostScaleRight()
/petsc/src/ksp/pc/impls/ml/
H A Dml.c153 const PetscScalar *xarray; in MatMult_ML() local
158 PetscCall(VecGetArrayRead(x, &xarray)); in MatMult_ML()
162 …perator_Apply", ML_Operator_Apply(shell->mlmat, x_length, (PetscScalar *)xarray, y_length, yarray)… in MatMult_ML()
163 PetscCall(VecRestoreArrayRead(x, &xarray)); in MatMult_ML()

12