Lines Matching refs:diag
5 static PetscErrorCode CheckDiagonal(Mat A, Vec diag, PetscScalar dval) in CheckDiagonal() argument
27 PetscCall(VecSetRandom(diag, NULL)); in CheckDiagonal()
28 PetscCall(MatGetDiagonal(A, diag)); in CheckDiagonal()
29 PetscCall(VecViewFromOptions(diag, NULL, "-diag_vec_view")); in CheckDiagonal()
31 PetscCall(VecGetLocalSize(diag, &n)); in CheckDiagonal()
32 PetscCall(VecGetArrayRead(diag, &arr)); in CheckDiagonal()
46 PetscCall(VecRestoreArrayRead(diag, &arr)); in CheckDiagonal()
86 Vec diag; in main() local
97 PetscCall(MatCreateVecs(A, &diag, NULL)); in main()
99 PetscCall(CheckDiagonal(A, diag, 0.0)); in main()
100 PetscCall(CheckDiagonal(A, diag, 1.0)); in main()
101 PetscCall(CheckDiagonal(A, diag, 2.0)); in main()
103 PetscCall(VecDestroy(&diag)); in main()