167e560aaSBarry Smith /* 267e560aaSBarry Smith Defines the basic matrix operations for sequential dense. 347d993e7Ssuyashtn Portions of this code are under: 447d993e7Ssuyashtn Copyright (c) 2022 Advanced Micro Devices, Inc. All rights reserved. 567e560aaSBarry Smith */ 6289bc588SBarry Smith 7dec5eb66SMatthew G Knepley #include <../src/mat/impls/dense/seq/dense.h> /*I "petscmat.h" I*/ 8cd3f9d89SJunchao Zhang #include <../src/mat/impls/dense/mpi/mpidense.h> 9c6db04a5SJed Brown #include <petscblaslapack.h> 106a63e612SBarry Smith #include <../src/mat/impls/aij/seq/aij.h> 11b2573a8aSBarry Smith 12d71ae5a4SJacob Faibussowitsch PetscErrorCode MatSeqDenseSymmetrize_Private(Mat A, PetscBool hermitian) 13d71ae5a4SJacob Faibussowitsch { 148c178816SStefano Zampini Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 158c178816SStefano Zampini PetscInt j, k, n = A->rmap->n; 16ca15aa20SStefano Zampini PetscScalar *v; 178c178816SStefano Zampini 188c178816SStefano Zampini PetscFunctionBegin; 1908401ef6SPierre Jolivet PetscCheck(A->rmap->n == A->cmap->n, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "Cannot symmetrize a rectangular matrix"); 209566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, &v)); 218c178816SStefano Zampini if (!hermitian) { 228c178816SStefano Zampini for (k = 0; k < n; k++) { 23ad540459SPierre Jolivet for (j = k; j < n; j++) v[j * mat->lda + k] = v[k * mat->lda + j]; 248c178816SStefano Zampini } 258c178816SStefano Zampini } else { 268c178816SStefano Zampini for (k = 0; k < n; k++) { 27ad540459SPierre Jolivet for (j = k; j < n; j++) v[j * mat->lda + k] = PetscConj(v[k * mat->lda + j]); 288c178816SStefano Zampini } 298c178816SStefano Zampini } 309566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, &v)); 318c178816SStefano Zampini PetscFunctionReturn(0); 328c178816SStefano Zampini } 338c178816SStefano Zampini 34d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode MatSeqDenseInvertFactors_Private(Mat A) 35d71ae5a4SJacob Faibussowitsch { 368c178816SStefano Zampini Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 378c178816SStefano Zampini PetscBLASInt info, n; 388c178816SStefano Zampini 398c178816SStefano Zampini PetscFunctionBegin; 408c178816SStefano Zampini if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 419566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->cmap->n, &n)); 428c178816SStefano Zampini if (A->factortype == MAT_FACTOR_LU) { 4328b400f6SJacob Faibussowitsch PetscCheck(mat->pivots, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Pivots not present"); 448c178816SStefano Zampini if (!mat->fwork) { 458c178816SStefano Zampini mat->lfwork = n; 469566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(mat->lfwork, &mat->fwork)); 478c178816SStefano Zampini } 489566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 49792fecdfSBarry Smith PetscCallBLAS("LAPACKgetri", LAPACKgetri_(&n, mat->v, &mat->lda, mat->pivots, mat->fwork, &mat->lfwork, &info)); 509566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 519566063dSJacob Faibussowitsch PetscCall(PetscLogFlops((1.0 * A->cmap->n * A->cmap->n * A->cmap->n) / 3.0)); 528c178816SStefano Zampini } else if (A->factortype == MAT_FACTOR_CHOLESKY) { 53b94d7dedSBarry Smith if (A->spd == PETSC_BOOL3_TRUE) { 549566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 55792fecdfSBarry Smith PetscCallBLAS("LAPACKpotri", LAPACKpotri_("L", &n, mat->v, &mat->lda, &info)); 569566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 579566063dSJacob Faibussowitsch PetscCall(MatSeqDenseSymmetrize_Private(A, PETSC_TRUE)); 588c178816SStefano Zampini #if defined(PETSC_USE_COMPLEX) 59b94d7dedSBarry Smith } else if (A->hermitian == PETSC_BOOL3_TRUE) { 6028b400f6SJacob Faibussowitsch PetscCheck(mat->pivots, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Pivots not present"); 6128b400f6SJacob Faibussowitsch PetscCheck(mat->fwork, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Fwork not present"); 629566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 63792fecdfSBarry Smith PetscCallBLAS("LAPACKhetri", LAPACKhetri_("L", &n, mat->v, &mat->lda, mat->pivots, mat->fwork, &info)); 649566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 659566063dSJacob Faibussowitsch PetscCall(MatSeqDenseSymmetrize_Private(A, PETSC_TRUE)); 668c178816SStefano Zampini #endif 678c178816SStefano Zampini } else { /* symmetric case */ 6828b400f6SJacob Faibussowitsch PetscCheck(mat->pivots, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Pivots not present"); 6928b400f6SJacob Faibussowitsch PetscCheck(mat->fwork, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Fwork not present"); 709566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 71792fecdfSBarry Smith PetscCallBLAS("LAPACKsytri", LAPACKsytri_("L", &n, mat->v, &mat->lda, mat->pivots, mat->fwork, &info)); 729566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 739566063dSJacob Faibussowitsch PetscCall(MatSeqDenseSymmetrize_Private(A, PETSC_FALSE)); 748c178816SStefano Zampini } 7528b400f6SJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_MAT_CH_ZRPVT, "Bad Inversion: zero pivot in row %" PetscInt_FMT, (PetscInt)info - 1); 769566063dSJacob Faibussowitsch PetscCall(PetscLogFlops((1.0 * A->cmap->n * A->cmap->n * A->cmap->n) / 3.0)); 778c178816SStefano Zampini } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "Matrix must be factored to solve"); 788c178816SStefano Zampini 798c178816SStefano Zampini A->ops->solve = NULL; 808c178816SStefano Zampini A->ops->matsolve = NULL; 818c178816SStefano Zampini A->ops->solvetranspose = NULL; 828c178816SStefano Zampini A->ops->matsolvetranspose = NULL; 838c178816SStefano Zampini A->ops->solveadd = NULL; 848c178816SStefano Zampini A->ops->solvetransposeadd = NULL; 858c178816SStefano Zampini A->factortype = MAT_FACTOR_NONE; 869566063dSJacob Faibussowitsch PetscCall(PetscFree(A->solvertype)); 878c178816SStefano Zampini PetscFunctionReturn(0); 888c178816SStefano Zampini } 898c178816SStefano Zampini 90d71ae5a4SJacob Faibussowitsch PetscErrorCode MatZeroRowsColumns_SeqDense(Mat A, PetscInt N, const PetscInt rows[], PetscScalar diag, Vec x, Vec b) 91d71ae5a4SJacob Faibussowitsch { 923f49a652SStefano Zampini Mat_SeqDense *l = (Mat_SeqDense *)A->data; 933f49a652SStefano Zampini PetscInt m = l->lda, n = A->cmap->n, r = A->rmap->n, i, j; 94ca15aa20SStefano Zampini PetscScalar *slot, *bb, *v; 953f49a652SStefano Zampini const PetscScalar *xx; 963f49a652SStefano Zampini 973f49a652SStefano Zampini PetscFunctionBegin; 9876bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 993f49a652SStefano Zampini for (i = 0; i < N; i++) { 10008401ef6SPierre Jolivet PetscCheck(rows[i] >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Negative row requested to be zeroed"); 10108401ef6SPierre Jolivet PetscCheck(rows[i] < A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row %" PetscInt_FMT " requested to be zeroed greater than or equal number of rows %" PetscInt_FMT, rows[i], A->rmap->n); 10208401ef6SPierre Jolivet PetscCheck(rows[i] < A->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Col %" PetscInt_FMT " requested to be zeroed greater than or equal number of cols %" PetscInt_FMT, rows[i], A->cmap->n); 1033f49a652SStefano Zampini } 10476bd3646SJed Brown } 105ca15aa20SStefano Zampini if (!N) PetscFunctionReturn(0); 1063f49a652SStefano Zampini 1073f49a652SStefano Zampini /* fix right hand side if needed */ 1083f49a652SStefano Zampini if (x && b) { 1096c4d906cSStefano Zampini Vec xt; 1106c4d906cSStefano Zampini 11108401ef6SPierre Jolivet PetscCheck(A->rmap->n == A->cmap->n, PETSC_COMM_SELF, PETSC_ERR_SUP, "Only coded for square matrices"); 1129566063dSJacob Faibussowitsch PetscCall(VecDuplicate(x, &xt)); 1139566063dSJacob Faibussowitsch PetscCall(VecCopy(x, xt)); 1149566063dSJacob Faibussowitsch PetscCall(VecScale(xt, -1.0)); 1159566063dSJacob Faibussowitsch PetscCall(MatMultAdd(A, xt, b, b)); 1169566063dSJacob Faibussowitsch PetscCall(VecDestroy(&xt)); 1179566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(x, &xx)); 1189566063dSJacob Faibussowitsch PetscCall(VecGetArray(b, &bb)); 1193f49a652SStefano Zampini for (i = 0; i < N; i++) bb[rows[i]] = diag * xx[rows[i]]; 1209566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(x, &xx)); 1219566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(b, &bb)); 1223f49a652SStefano Zampini } 1233f49a652SStefano Zampini 1249566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, &v)); 1253f49a652SStefano Zampini for (i = 0; i < N; i++) { 126ca15aa20SStefano Zampini slot = v + rows[i] * m; 1279566063dSJacob Faibussowitsch PetscCall(PetscArrayzero(slot, r)); 1283f49a652SStefano Zampini } 1293f49a652SStefano Zampini for (i = 0; i < N; i++) { 130ca15aa20SStefano Zampini slot = v + rows[i]; 1319371c9d4SSatish Balay for (j = 0; j < n; j++) { 1329371c9d4SSatish Balay *slot = 0.0; 1339371c9d4SSatish Balay slot += m; 1349371c9d4SSatish Balay } 1353f49a652SStefano Zampini } 1363f49a652SStefano Zampini if (diag != 0.0) { 13708401ef6SPierre Jolivet PetscCheck(A->rmap->n == A->cmap->n, PETSC_COMM_SELF, PETSC_ERR_SUP, "Only coded for square matrices"); 1383f49a652SStefano Zampini for (i = 0; i < N; i++) { 139ca15aa20SStefano Zampini slot = v + (m + 1) * rows[i]; 1403f49a652SStefano Zampini *slot = diag; 1413f49a652SStefano Zampini } 1423f49a652SStefano Zampini } 1439566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, &v)); 1443f49a652SStefano Zampini PetscFunctionReturn(0); 1453f49a652SStefano Zampini } 1463f49a652SStefano Zampini 147d71ae5a4SJacob Faibussowitsch PetscErrorCode MatPtAPNumeric_SeqDense_SeqDense(Mat A, Mat P, Mat C) 148d71ae5a4SJacob Faibussowitsch { 149abc3b08eSStefano Zampini Mat_SeqDense *c = (Mat_SeqDense *)(C->data); 150abc3b08eSStefano Zampini 151abc3b08eSStefano Zampini PetscFunctionBegin; 152ca15aa20SStefano Zampini if (c->ptapwork) { 1539566063dSJacob Faibussowitsch PetscCall((*C->ops->matmultnumeric)(A, P, c->ptapwork)); 1549566063dSJacob Faibussowitsch PetscCall((*C->ops->transposematmultnumeric)(P, c->ptapwork, C)); 1554222ddf1SHong Zhang } else SETERRQ(PetscObjectComm((PetscObject)C), PETSC_ERR_SUP, "Must call MatPtAPSymbolic_SeqDense_SeqDense() first"); 156abc3b08eSStefano Zampini PetscFunctionReturn(0); 157abc3b08eSStefano Zampini } 158abc3b08eSStefano Zampini 159d71ae5a4SJacob Faibussowitsch PetscErrorCode MatPtAPSymbolic_SeqDense_SeqDense(Mat A, Mat P, PetscReal fill, Mat C) 160d71ae5a4SJacob Faibussowitsch { 161abc3b08eSStefano Zampini Mat_SeqDense *c; 16247d993e7Ssuyashtn PetscBool cisdense = PETSC_FALSE; 163abc3b08eSStefano Zampini 164abc3b08eSStefano Zampini PetscFunctionBegin; 1659566063dSJacob Faibussowitsch PetscCall(MatSetSizes(C, P->cmap->n, P->cmap->n, P->cmap->N, P->cmap->N)); 16647d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) 1679566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompareAny((PetscObject)C, &cisdense, MATSEQDENSE, MATSEQDENSECUDA, "")); 16847d993e7Ssuyashtn #elif (PETSC_HAVE_HIP) 16947d993e7Ssuyashtn PetscCall(PetscObjectTypeCompareAny((PetscObject)C, &cisdense, MATSEQDENSE, MATSEQDENSEHIP, "")); 17047d993e7Ssuyashtn #endif 17147d993e7Ssuyashtn 1727a3c3d58SStefano Zampini if (!cisdense) { 1737a3c3d58SStefano Zampini PetscBool flg; 1747a3c3d58SStefano Zampini 1759566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)P, ((PetscObject)A)->type_name, &flg)); 1769566063dSJacob Faibussowitsch PetscCall(MatSetType(C, flg ? ((PetscObject)A)->type_name : MATDENSE)); 1777a3c3d58SStefano Zampini } 1789566063dSJacob Faibussowitsch PetscCall(MatSetUp(C)); 1794222ddf1SHong Zhang c = (Mat_SeqDense *)C->data; 1809566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &c->ptapwork)); 1819566063dSJacob Faibussowitsch PetscCall(MatSetSizes(c->ptapwork, A->rmap->n, P->cmap->n, A->rmap->N, P->cmap->N)); 1829566063dSJacob Faibussowitsch PetscCall(MatSetType(c->ptapwork, ((PetscObject)C)->type_name)); 1839566063dSJacob Faibussowitsch PetscCall(MatSetUp(c->ptapwork)); 184abc3b08eSStefano Zampini PetscFunctionReturn(0); 185abc3b08eSStefano Zampini } 186abc3b08eSStefano Zampini 187d71ae5a4SJacob Faibussowitsch PETSC_INTERN PetscErrorCode MatConvert_SeqAIJ_SeqDense(Mat A, MatType newtype, MatReuse reuse, Mat *newmat) 188d71ae5a4SJacob Faibussowitsch { 189a13144ffSStefano Zampini Mat B = NULL; 190b49cda9fSStefano Zampini Mat_SeqAIJ *a = (Mat_SeqAIJ *)A->data; 191b49cda9fSStefano Zampini Mat_SeqDense *b; 192b49cda9fSStefano Zampini PetscInt *ai = a->i, *aj = a->j, m = A->rmap->N, n = A->cmap->N, i; 1932e5835c6SStefano Zampini const MatScalar *av; 194a13144ffSStefano Zampini PetscBool isseqdense; 195b49cda9fSStefano Zampini 196b49cda9fSStefano Zampini PetscFunctionBegin; 197a13144ffSStefano Zampini if (reuse == MAT_REUSE_MATRIX) { 1989566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)*newmat, MATSEQDENSE, &isseqdense)); 19928b400f6SJacob Faibussowitsch PetscCheck(isseqdense, PetscObjectComm((PetscObject)*newmat), PETSC_ERR_USER, "Cannot reuse matrix of type %s", ((PetscObject)(*newmat))->type_name); 200a13144ffSStefano Zampini } 201a13144ffSStefano Zampini if (reuse != MAT_REUSE_MATRIX) { 2029566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B)); 2039566063dSJacob Faibussowitsch PetscCall(MatSetSizes(B, m, n, m, n)); 2049566063dSJacob Faibussowitsch PetscCall(MatSetType(B, MATSEQDENSE)); 2059566063dSJacob Faibussowitsch PetscCall(MatSeqDenseSetPreallocation(B, NULL)); 206b49cda9fSStefano Zampini b = (Mat_SeqDense *)(B->data); 207a13144ffSStefano Zampini } else { 208a13144ffSStefano Zampini b = (Mat_SeqDense *)((*newmat)->data); 2099566063dSJacob Faibussowitsch PetscCall(PetscArrayzero(b->v, m * n)); 210a13144ffSStefano Zampini } 2119566063dSJacob Faibussowitsch PetscCall(MatSeqAIJGetArrayRead(A, &av)); 212b49cda9fSStefano Zampini for (i = 0; i < m; i++) { 213b49cda9fSStefano Zampini PetscInt j; 214b49cda9fSStefano Zampini for (j = 0; j < ai[1] - ai[0]; j++) { 215b49cda9fSStefano Zampini b->v[*aj * m + i] = *av; 216b49cda9fSStefano Zampini aj++; 217b49cda9fSStefano Zampini av++; 218b49cda9fSStefano Zampini } 219b49cda9fSStefano Zampini ai++; 220b49cda9fSStefano Zampini } 2219566063dSJacob Faibussowitsch PetscCall(MatSeqAIJRestoreArrayRead(A, &av)); 222b49cda9fSStefano Zampini 223511c6705SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 2249566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(B, MAT_FINAL_ASSEMBLY)); 2259566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(B, MAT_FINAL_ASSEMBLY)); 2269566063dSJacob Faibussowitsch PetscCall(MatHeaderReplace(A, &B)); 227b49cda9fSStefano Zampini } else { 228a13144ffSStefano Zampini if (B) *newmat = B; 2299566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(*newmat, MAT_FINAL_ASSEMBLY)); 2309566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(*newmat, MAT_FINAL_ASSEMBLY)); 231b49cda9fSStefano Zampini } 232b49cda9fSStefano Zampini PetscFunctionReturn(0); 233b49cda9fSStefano Zampini } 234b49cda9fSStefano Zampini 235d71ae5a4SJacob Faibussowitsch PETSC_INTERN PetscErrorCode MatConvert_SeqDense_SeqAIJ(Mat A, MatType newtype, MatReuse reuse, Mat *newmat) 236d71ae5a4SJacob Faibussowitsch { 2376d4ec7b0SPierre Jolivet Mat B = NULL; 2386a63e612SBarry Smith Mat_SeqDense *a = (Mat_SeqDense *)A->data; 2399399e1b8SMatthew G. Knepley PetscInt i, j; 2409399e1b8SMatthew G. Knepley PetscInt *rows, *nnz; 2419399e1b8SMatthew G. Knepley MatScalar *aa = a->v, *vals; 2426a63e612SBarry Smith 2436a63e612SBarry Smith PetscFunctionBegin; 2449566063dSJacob Faibussowitsch PetscCall(PetscCalloc3(A->rmap->n, &rows, A->rmap->n, &nnz, A->rmap->n, &vals)); 2456d4ec7b0SPierre Jolivet if (reuse != MAT_REUSE_MATRIX) { 2469566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &B)); 2479566063dSJacob Faibussowitsch PetscCall(MatSetSizes(B, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N)); 2489566063dSJacob Faibussowitsch PetscCall(MatSetType(B, MATSEQAIJ)); 2499399e1b8SMatthew G. Knepley for (j = 0; j < A->cmap->n; j++) { 2509371c9d4SSatish Balay for (i = 0; i < A->rmap->n; i++) 2519371c9d4SSatish Balay if (aa[i] != 0.0 || (i == j && A->cmap->n == A->rmap->n)) ++nnz[i]; 2526a63e612SBarry Smith aa += a->lda; 2536a63e612SBarry Smith } 2549566063dSJacob Faibussowitsch PetscCall(MatSeqAIJSetPreallocation(B, PETSC_DETERMINE, nnz)); 2556d4ec7b0SPierre Jolivet } else B = *newmat; 2569399e1b8SMatthew G. Knepley aa = a->v; 2579399e1b8SMatthew G. Knepley for (j = 0; j < A->cmap->n; j++) { 2589399e1b8SMatthew G. Knepley PetscInt numRows = 0; 2599371c9d4SSatish Balay for (i = 0; i < A->rmap->n; i++) 2609371c9d4SSatish Balay if (aa[i] != 0.0 || (i == j && A->cmap->n == A->rmap->n)) { 2619371c9d4SSatish Balay rows[numRows] = i; 2629371c9d4SSatish Balay vals[numRows++] = aa[i]; 2639371c9d4SSatish Balay } 2649566063dSJacob Faibussowitsch PetscCall(MatSetValues(B, numRows, rows, 1, &j, vals, INSERT_VALUES)); 2659399e1b8SMatthew G. Knepley aa += a->lda; 2669399e1b8SMatthew G. Knepley } 2679566063dSJacob Faibussowitsch PetscCall(PetscFree3(rows, nnz, vals)); 2689566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(B, MAT_FINAL_ASSEMBLY)); 2699566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(B, MAT_FINAL_ASSEMBLY)); 2706a63e612SBarry Smith 271511c6705SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 2729566063dSJacob Faibussowitsch PetscCall(MatHeaderReplace(A, &B)); 2736d4ec7b0SPierre Jolivet } else if (reuse != MAT_REUSE_MATRIX) *newmat = B; 2746a63e612SBarry Smith PetscFunctionReturn(0); 2756a63e612SBarry Smith } 2766a63e612SBarry Smith 277d71ae5a4SJacob Faibussowitsch PetscErrorCode MatAXPY_SeqDense(Mat Y, PetscScalar alpha, Mat X, MatStructure str) 278d71ae5a4SJacob Faibussowitsch { 2791987afe7SBarry Smith Mat_SeqDense *x = (Mat_SeqDense *)X->data, *y = (Mat_SeqDense *)Y->data; 280ca15aa20SStefano Zampini const PetscScalar *xv; 281ca15aa20SStefano Zampini PetscScalar *yv; 28223fff9afSBarry Smith PetscBLASInt N, m, ldax = 0, lday = 0, one = 1; 2833a40ed3dSBarry Smith 2843a40ed3dSBarry Smith PetscFunctionBegin; 2859566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(X, &xv)); 2869566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(Y, &yv)); 2879566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(X->rmap->n * X->cmap->n, &N)); 2889566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(X->rmap->n, &m)); 2899566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(x->lda, &ldax)); 2909566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(y->lda, &lday)); 291a5ce6ee0Svictorle if (ldax > m || lday > m) { 292ca15aa20SStefano Zampini PetscInt j; 293ca15aa20SStefano Zampini 29448a46eb9SPierre Jolivet for (j = 0; j < X->cmap->n; j++) PetscCallBLAS("BLASaxpy", BLASaxpy_(&m, &alpha, xv + j * ldax, &one, yv + j * lday, &one)); 295a5ce6ee0Svictorle } else { 296792fecdfSBarry Smith PetscCallBLAS("BLASaxpy", BLASaxpy_(&N, &alpha, xv, &one, yv, &one)); 297a5ce6ee0Svictorle } 2989566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(X, &xv)); 2999566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(Y, &yv)); 3009566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(PetscMax(2.0 * N - 1, 0))); 3013a40ed3dSBarry Smith PetscFunctionReturn(0); 3021987afe7SBarry Smith } 3031987afe7SBarry Smith 304d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetInfo_SeqDense(Mat A, MatInfoType flag, MatInfo *info) 305d71ae5a4SJacob Faibussowitsch { 306ca15aa20SStefano Zampini PetscLogDouble N = A->rmap->n * A->cmap->n; 3073a40ed3dSBarry Smith 3083a40ed3dSBarry Smith PetscFunctionBegin; 3094e220ebcSLois Curfman McInnes info->block_size = 1.0; 310ca15aa20SStefano Zampini info->nz_allocated = N; 311ca15aa20SStefano Zampini info->nz_used = N; 312ca15aa20SStefano Zampini info->nz_unneeded = 0; 313ca15aa20SStefano Zampini info->assemblies = A->num_ass; 3144e220ebcSLois Curfman McInnes info->mallocs = 0; 3154dfa11a4SJacob Faibussowitsch info->memory = 0; /* REVIEW ME */ 3164e220ebcSLois Curfman McInnes info->fill_ratio_given = 0; 3174e220ebcSLois Curfman McInnes info->fill_ratio_needed = 0; 3184e220ebcSLois Curfman McInnes info->factor_mallocs = 0; 3193a40ed3dSBarry Smith PetscFunctionReturn(0); 320289bc588SBarry Smith } 321289bc588SBarry Smith 322d71ae5a4SJacob Faibussowitsch PetscErrorCode MatScale_SeqDense(Mat A, PetscScalar alpha) 323d71ae5a4SJacob Faibussowitsch { 324273d9f13SBarry Smith Mat_SeqDense *a = (Mat_SeqDense *)A->data; 325ca15aa20SStefano Zampini PetscScalar *v; 32623fff9afSBarry Smith PetscBLASInt one = 1, j, nz, lda = 0; 32780cd9d93SLois Curfman McInnes 3283a40ed3dSBarry Smith PetscFunctionBegin; 3299566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, &v)); 3309566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(a->lda, &lda)); 331d0f46423SBarry Smith if (lda > A->rmap->n) { 3329566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->rmap->n, &nz)); 33348a46eb9SPierre Jolivet for (j = 0; j < A->cmap->n; j++) PetscCallBLAS("BLASscal", BLASscal_(&nz, &alpha, v + j * lda, &one)); 334a5ce6ee0Svictorle } else { 3359566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->rmap->n * A->cmap->n, &nz)); 336792fecdfSBarry Smith PetscCallBLAS("BLASscal", BLASscal_(&nz, &alpha, v, &one)); 337a5ce6ee0Svictorle } 33804cbc005SJose E. Roman PetscCall(PetscLogFlops(A->rmap->n * A->cmap->n)); 3399566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, &v)); 3403a40ed3dSBarry Smith PetscFunctionReturn(0); 34180cd9d93SLois Curfman McInnes } 34280cd9d93SLois Curfman McInnes 343d71ae5a4SJacob Faibussowitsch PetscErrorCode MatShift_SeqDense(Mat A, PetscScalar alpha) 344d71ae5a4SJacob Faibussowitsch { 3452f605a99SJose E. Roman Mat_SeqDense *a = (Mat_SeqDense *)A->data; 3462f605a99SJose E. Roman PetscScalar *v; 3472f605a99SJose E. Roman PetscInt j, k; 3482f605a99SJose E. Roman 3492f605a99SJose E. Roman PetscFunctionBegin; 3502f605a99SJose E. Roman PetscCall(MatDenseGetArray(A, &v)); 3512f605a99SJose E. Roman k = PetscMin(A->rmap->n, A->cmap->n); 3522f605a99SJose E. Roman for (j = 0; j < k; j++) v[j + j * a->lda] += alpha; 3532f605a99SJose E. Roman PetscCall(PetscLogFlops(k)); 3542f605a99SJose E. Roman PetscCall(MatDenseRestoreArray(A, &v)); 3552f605a99SJose E. Roman PetscFunctionReturn(0); 3562f605a99SJose E. Roman } 3572f605a99SJose E. Roman 358d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatIsHermitian_SeqDense(Mat A, PetscReal rtol, PetscBool *fl) 359d71ae5a4SJacob Faibussowitsch { 3601cbb95d3SBarry Smith Mat_SeqDense *a = (Mat_SeqDense *)A->data; 361ca15aa20SStefano Zampini PetscInt i, j, m = A->rmap->n, N = a->lda; 362ca15aa20SStefano Zampini const PetscScalar *v; 3631cbb95d3SBarry Smith 3641cbb95d3SBarry Smith PetscFunctionBegin; 3651cbb95d3SBarry Smith *fl = PETSC_FALSE; 366d0f46423SBarry Smith if (A->rmap->n != A->cmap->n) PetscFunctionReturn(0); 3679566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &v)); 3681cbb95d3SBarry Smith for (i = 0; i < m; i++) { 369ca15aa20SStefano Zampini for (j = i; j < m; j++) { 370ad540459SPierre Jolivet if (PetscAbsScalar(v[i + j * N] - PetscConj(v[j + i * N])) > rtol) goto restore; 3711cbb95d3SBarry Smith } 372637a0070SStefano Zampini } 3731cbb95d3SBarry Smith *fl = PETSC_TRUE; 374637a0070SStefano Zampini restore: 3759566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &v)); 376637a0070SStefano Zampini PetscFunctionReturn(0); 377637a0070SStefano Zampini } 378637a0070SStefano Zampini 379d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatIsSymmetric_SeqDense(Mat A, PetscReal rtol, PetscBool *fl) 380d71ae5a4SJacob Faibussowitsch { 381637a0070SStefano Zampini Mat_SeqDense *a = (Mat_SeqDense *)A->data; 382637a0070SStefano Zampini PetscInt i, j, m = A->rmap->n, N = a->lda; 383637a0070SStefano Zampini const PetscScalar *v; 384637a0070SStefano Zampini 385637a0070SStefano Zampini PetscFunctionBegin; 386637a0070SStefano Zampini *fl = PETSC_FALSE; 387637a0070SStefano Zampini if (A->rmap->n != A->cmap->n) PetscFunctionReturn(0); 3889566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &v)); 389637a0070SStefano Zampini for (i = 0; i < m; i++) { 390637a0070SStefano Zampini for (j = i; j < m; j++) { 391ad540459SPierre Jolivet if (PetscAbsScalar(v[i + j * N] - v[j + i * N]) > rtol) goto restore; 392637a0070SStefano Zampini } 393637a0070SStefano Zampini } 394637a0070SStefano Zampini *fl = PETSC_TRUE; 395637a0070SStefano Zampini restore: 3969566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &v)); 3971cbb95d3SBarry Smith PetscFunctionReturn(0); 3981cbb95d3SBarry Smith } 3991cbb95d3SBarry Smith 400d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDuplicateNoCreate_SeqDense(Mat newi, Mat A, MatDuplicateOption cpvalues) 401d71ae5a4SJacob Faibussowitsch { 402ca15aa20SStefano Zampini Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 40323fc5dcaSStefano Zampini PetscInt lda = (PetscInt)mat->lda, j, m, nlda = lda; 40475f6d85dSStefano Zampini PetscBool isdensecpu; 405b24902e0SBarry Smith 406b24902e0SBarry Smith PetscFunctionBegin; 4079566063dSJacob Faibussowitsch PetscCall(PetscLayoutReference(A->rmap, &newi->rmap)); 4089566063dSJacob Faibussowitsch PetscCall(PetscLayoutReference(A->cmap, &newi->cmap)); 40923fc5dcaSStefano Zampini if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { /* propagate LDA */ 4109566063dSJacob Faibussowitsch PetscCall(MatDenseSetLDA(newi, lda)); 41123fc5dcaSStefano Zampini } 4129566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)newi, MATSEQDENSE, &isdensecpu)); 4139566063dSJacob Faibussowitsch if (isdensecpu) PetscCall(MatSeqDenseSetPreallocation(newi, NULL)); 414b24902e0SBarry Smith if (cpvalues == MAT_COPY_VALUES) { 415ca15aa20SStefano Zampini const PetscScalar *av; 416ca15aa20SStefano Zampini PetscScalar *v; 417ca15aa20SStefano Zampini 4189566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &av)); 4199566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayWrite(newi, &v)); 4209566063dSJacob Faibussowitsch PetscCall(MatDenseGetLDA(newi, &nlda)); 421d0f46423SBarry Smith m = A->rmap->n; 42223fc5dcaSStefano Zampini if (lda > m || nlda > m) { 42348a46eb9SPierre Jolivet for (j = 0; j < A->cmap->n; j++) PetscCall(PetscArraycpy(v + j * nlda, av + j * lda, m)); 424b24902e0SBarry Smith } else { 4259566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(v, av, A->rmap->n * A->cmap->n)); 426b24902e0SBarry Smith } 4279566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayWrite(newi, &v)); 4289566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &av)); 429b24902e0SBarry Smith } 430b24902e0SBarry Smith PetscFunctionReturn(0); 431b24902e0SBarry Smith } 432b24902e0SBarry Smith 433d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDuplicate_SeqDense(Mat A, MatDuplicateOption cpvalues, Mat *newmat) 434d71ae5a4SJacob Faibussowitsch { 4353a40ed3dSBarry Smith PetscFunctionBegin; 4369566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), newmat)); 4379566063dSJacob Faibussowitsch PetscCall(MatSetSizes(*newmat, A->rmap->n, A->cmap->n, A->rmap->n, A->cmap->n)); 4389566063dSJacob Faibussowitsch PetscCall(MatSetType(*newmat, ((PetscObject)A)->type_name)); 4399566063dSJacob Faibussowitsch PetscCall(MatDuplicateNoCreate_SeqDense(*newmat, A, cpvalues)); 440b24902e0SBarry Smith PetscFunctionReturn(0); 441b24902e0SBarry Smith } 442b24902e0SBarry Smith 443d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolve_SeqDense_Internal_LU(Mat A, PetscScalar *x, PetscBLASInt ldx, PetscBLASInt m, PetscBLASInt nrhs, PetscBLASInt k, PetscBool T) 444d71ae5a4SJacob Faibussowitsch { 445c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 4464396437dSToby Isaac PetscBLASInt info; 44767e560aaSBarry Smith 4483a40ed3dSBarry Smith PetscFunctionBegin; 4499566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 450792fecdfSBarry Smith PetscCallBLAS("LAPACKgetrs", LAPACKgetrs_(T ? "T" : "N", &m, &nrhs, mat->v, &mat->lda, mat->pivots, x, &m, &info)); 4519566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 45205fcb23eSStefano Zampini PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "GETRS - Bad solve %d", (int)info); 4539566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(nrhs * (2.0 * m * m - m))); 4544396437dSToby Isaac PetscFunctionReturn(0); 4554396437dSToby Isaac } 4564396437dSToby Isaac 4574396437dSToby Isaac static PetscErrorCode MatConjugate_SeqDense(Mat); 4584396437dSToby Isaac 459d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolve_SeqDense_Internal_Cholesky(Mat A, PetscScalar *x, PetscBLASInt ldx, PetscBLASInt m, PetscBLASInt nrhs, PetscBLASInt k, PetscBool T) 460d71ae5a4SJacob Faibussowitsch { 4614396437dSToby Isaac Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 4624396437dSToby Isaac PetscBLASInt info; 4634396437dSToby Isaac 4644396437dSToby Isaac PetscFunctionBegin; 465b94d7dedSBarry Smith if (A->spd == PETSC_BOOL3_TRUE) { 4669566063dSJacob Faibussowitsch if (PetscDefined(USE_COMPLEX) && T) PetscCall(MatConjugate_SeqDense(A)); 4679566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 468792fecdfSBarry Smith PetscCallBLAS("LAPACKpotrs", LAPACKpotrs_("L", &m, &nrhs, mat->v, &mat->lda, x, &m, &info)); 4699566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 47005fcb23eSStefano Zampini PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "POTRS Bad solve %d", (int)info); 4719566063dSJacob Faibussowitsch if (PetscDefined(USE_COMPLEX) && T) PetscCall(MatConjugate_SeqDense(A)); 472a49dc2a2SStefano Zampini #if defined(PETSC_USE_COMPLEX) 473b94d7dedSBarry Smith } else if (A->hermitian == PETSC_BOOL3_TRUE) { 4749566063dSJacob Faibussowitsch if (T) PetscCall(MatConjugate_SeqDense(A)); 4759566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 476792fecdfSBarry Smith PetscCallBLAS("LAPACKhetrs", LAPACKhetrs_("L", &m, &nrhs, mat->v, &mat->lda, mat->pivots, x, &m, &info)); 4779566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 47805fcb23eSStefano Zampini PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "HETRS Bad solve %d", (int)info); 4799566063dSJacob Faibussowitsch if (T) PetscCall(MatConjugate_SeqDense(A)); 480a49dc2a2SStefano Zampini #endif 481a49dc2a2SStefano Zampini } else { /* symmetric case */ 4829566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 483792fecdfSBarry Smith PetscCallBLAS("LAPACKsytrs", LAPACKsytrs_("L", &m, &nrhs, mat->v, &mat->lda, mat->pivots, x, &m, &info)); 4849566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 48505fcb23eSStefano Zampini PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "SYTRS Bad solve %d", (int)info); 486a49dc2a2SStefano Zampini } 4879566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(nrhs * (2.0 * m * m - m))); 4884396437dSToby Isaac PetscFunctionReturn(0); 4894396437dSToby Isaac } 49085e2c93fSHong Zhang 491d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolve_SeqDense_Internal_QR(Mat A, PetscScalar *x, PetscBLASInt ldx, PetscBLASInt m, PetscBLASInt nrhs, PetscBLASInt k) 492d71ae5a4SJacob Faibussowitsch { 4934396437dSToby Isaac Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 4944396437dSToby Isaac PetscBLASInt info; 4954396437dSToby Isaac char trans; 4964396437dSToby Isaac 4974396437dSToby Isaac PetscFunctionBegin; 4984905a7bcSToby Isaac if (PetscDefined(USE_COMPLEX)) { 4994905a7bcSToby Isaac trans = 'C'; 5004905a7bcSToby Isaac } else { 5014905a7bcSToby Isaac trans = 'T'; 5024905a7bcSToby Isaac } 5039566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 50405fcb23eSStefano Zampini { /* lwork depends on the number of right-hand sides */ 50505fcb23eSStefano Zampini PetscBLASInt nlfwork, lfwork = -1; 50605fcb23eSStefano Zampini PetscScalar fwork; 50705fcb23eSStefano Zampini 508792fecdfSBarry Smith PetscCallBLAS("LAPACKormqr", LAPACKormqr_("L", &trans, &m, &nrhs, &mat->rank, mat->v, &mat->lda, mat->tau, x, &ldx, &fwork, &lfwork, &info)); 50905fcb23eSStefano Zampini nlfwork = (PetscBLASInt)PetscRealPart(fwork); 51005fcb23eSStefano Zampini if (nlfwork > mat->lfwork) { 51105fcb23eSStefano Zampini mat->lfwork = nlfwork; 51205fcb23eSStefano Zampini PetscCall(PetscFree(mat->fwork)); 51305fcb23eSStefano Zampini PetscCall(PetscMalloc1(mat->lfwork, &mat->fwork)); 51405fcb23eSStefano Zampini } 51505fcb23eSStefano Zampini } 516792fecdfSBarry Smith PetscCallBLAS("LAPACKormqr", LAPACKormqr_("L", &trans, &m, &nrhs, &mat->rank, mat->v, &mat->lda, mat->tau, x, &ldx, mat->fwork, &mat->lfwork, &info)); 5179566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 51805fcb23eSStefano Zampini PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "ORMQR - Bad orthogonal transform %d", (int)info); 5199566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 520792fecdfSBarry Smith PetscCallBLAS("LAPACKtrtrs", LAPACKtrtrs_("U", "N", "N", &mat->rank, &nrhs, mat->v, &mat->lda, x, &ldx, &info)); 5219566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 52205fcb23eSStefano Zampini PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "TRTRS - Bad triangular solve %d", (int)info); 5234905a7bcSToby Isaac for (PetscInt j = 0; j < nrhs; j++) { 524ad540459SPierre Jolivet for (PetscInt i = mat->rank; i < k; i++) x[j * ldx + i] = 0.; 5254905a7bcSToby Isaac } 5269566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(nrhs * (4.0 * m * mat->rank - PetscSqr(mat->rank)))); 5274905a7bcSToby Isaac PetscFunctionReturn(0); 5284905a7bcSToby Isaac } 5294905a7bcSToby Isaac 530d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolveTranspose_SeqDense_Internal_QR(Mat A, PetscScalar *x, PetscBLASInt ldx, PetscBLASInt m, PetscBLASInt nrhs, PetscBLASInt k) 531d71ae5a4SJacob Faibussowitsch { 5324396437dSToby Isaac Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 5334396437dSToby Isaac PetscBLASInt info; 5344396437dSToby Isaac 5354396437dSToby Isaac PetscFunctionBegin; 5364396437dSToby Isaac if (A->rmap->n == A->cmap->n && mat->rank == A->rmap->n) { 5379566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 538792fecdfSBarry Smith PetscCallBLAS("LAPACKtrtrs", LAPACKtrtrs_("U", "T", "N", &m, &nrhs, mat->v, &mat->lda, x, &ldx, &info)); 5399566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 54005fcb23eSStefano Zampini PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "TRTRS - Bad triangular solve %d", (int)info); 5419566063dSJacob Faibussowitsch if (PetscDefined(USE_COMPLEX)) PetscCall(MatConjugate_SeqDense(A)); 54205fcb23eSStefano Zampini { /* lwork depends on the number of right-hand sides */ 54305fcb23eSStefano Zampini PetscBLASInt nlfwork, lfwork = -1; 54405fcb23eSStefano Zampini PetscScalar fwork; 54505fcb23eSStefano Zampini 546792fecdfSBarry Smith PetscCallBLAS("LAPACKormqr", LAPACKormqr_("L", "N", &m, &nrhs, &mat->rank, mat->v, &mat->lda, mat->tau, x, &ldx, &fwork, &lfwork, &info)); 54705fcb23eSStefano Zampini nlfwork = (PetscBLASInt)PetscRealPart(fwork); 54805fcb23eSStefano Zampini if (nlfwork > mat->lfwork) { 54905fcb23eSStefano Zampini mat->lfwork = nlfwork; 55005fcb23eSStefano Zampini PetscCall(PetscFree(mat->fwork)); 55105fcb23eSStefano Zampini PetscCall(PetscMalloc1(mat->lfwork, &mat->fwork)); 55205fcb23eSStefano Zampini } 55305fcb23eSStefano Zampini } 5549566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 555792fecdfSBarry Smith PetscCallBLAS("LAPACKormqr", LAPACKormqr_("L", "N", &m, &nrhs, &mat->rank, mat->v, &mat->lda, mat->tau, x, &ldx, mat->fwork, &mat->lfwork, &info)); 5569566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 55705fcb23eSStefano Zampini PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "ORMQR - Bad orthogonal transform %d", (int)info); 5589566063dSJacob Faibussowitsch if (PetscDefined(USE_COMPLEX)) PetscCall(MatConjugate_SeqDense(A)); 5594396437dSToby Isaac } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "QR factored matrix cannot be used for transpose solve"); 5609566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(nrhs * (4.0 * m * mat->rank - PetscSqr(mat->rank)))); 5614396437dSToby Isaac PetscFunctionReturn(0); 5624396437dSToby Isaac } 5634396437dSToby Isaac 564d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolve_SeqDense_SetUp(Mat A, Vec xx, Vec yy, PetscScalar **_y, PetscBLASInt *_m, PetscBLASInt *_k) 565d71ae5a4SJacob Faibussowitsch { 5664396437dSToby Isaac Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 5674905a7bcSToby Isaac PetscScalar *y; 5684905a7bcSToby Isaac PetscBLASInt m = 0, k = 0; 5694905a7bcSToby Isaac 5704905a7bcSToby Isaac PetscFunctionBegin; 5719566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->rmap->n, &m)); 5729566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->cmap->n, &k)); 5734905a7bcSToby Isaac if (k < m) { 5749566063dSJacob Faibussowitsch PetscCall(VecCopy(xx, mat->qrrhs)); 5759566063dSJacob Faibussowitsch PetscCall(VecGetArray(mat->qrrhs, &y)); 5764905a7bcSToby Isaac } else { 5779566063dSJacob Faibussowitsch PetscCall(VecCopy(xx, yy)); 5789566063dSJacob Faibussowitsch PetscCall(VecGetArray(yy, &y)); 5794905a7bcSToby Isaac } 5804396437dSToby Isaac *_y = y; 5814396437dSToby Isaac *_k = k; 5824396437dSToby Isaac *_m = m; 5834396437dSToby Isaac PetscFunctionReturn(0); 5844396437dSToby Isaac } 5854396437dSToby Isaac 586d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolve_SeqDense_TearDown(Mat A, Vec xx, Vec yy, PetscScalar **_y, PetscBLASInt *_m, PetscBLASInt *_k) 587d71ae5a4SJacob Faibussowitsch { 5884396437dSToby Isaac Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 58942e9364cSSatish Balay PetscScalar *y = NULL; 5904396437dSToby Isaac PetscBLASInt m, k; 5914396437dSToby Isaac 5924396437dSToby Isaac PetscFunctionBegin; 5934396437dSToby Isaac y = *_y; 5944396437dSToby Isaac *_y = NULL; 5954396437dSToby Isaac k = *_k; 5964396437dSToby Isaac m = *_m; 5974905a7bcSToby Isaac if (k < m) { 5984905a7bcSToby Isaac PetscScalar *yv; 5999566063dSJacob Faibussowitsch PetscCall(VecGetArray(yy, &yv)); 6009566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(yv, y, k)); 6019566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(yy, &yv)); 6029566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(mat->qrrhs, &y)); 6034905a7bcSToby Isaac } else { 6049566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(yy, &y)); 6054905a7bcSToby Isaac } 6064905a7bcSToby Isaac PetscFunctionReturn(0); 6074905a7bcSToby Isaac } 6084905a7bcSToby Isaac 609d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolve_SeqDense_LU(Mat A, Vec xx, Vec yy) 610d71ae5a4SJacob Faibussowitsch { 61142e9364cSSatish Balay PetscScalar *y = NULL; 61242e9364cSSatish Balay PetscBLASInt m = 0, k = 0; 6134396437dSToby Isaac 6144396437dSToby Isaac PetscFunctionBegin; 6159566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_SetUp(A, xx, yy, &y, &m, &k)); 6169566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_Internal_LU(A, y, m, m, 1, k, PETSC_FALSE)); 6179566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_TearDown(A, xx, yy, &y, &m, &k)); 6184396437dSToby Isaac PetscFunctionReturn(0); 6194396437dSToby Isaac } 6204396437dSToby Isaac 621d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolveTranspose_SeqDense_LU(Mat A, Vec xx, Vec yy) 622d71ae5a4SJacob Faibussowitsch { 62342e9364cSSatish Balay PetscScalar *y = NULL; 62442e9364cSSatish Balay PetscBLASInt m = 0, k = 0; 6254396437dSToby Isaac 6264396437dSToby Isaac PetscFunctionBegin; 6279566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_SetUp(A, xx, yy, &y, &m, &k)); 6289566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_Internal_LU(A, y, m, m, 1, k, PETSC_TRUE)); 6299566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_TearDown(A, xx, yy, &y, &m, &k)); 6304396437dSToby Isaac PetscFunctionReturn(0); 6314396437dSToby Isaac } 6324396437dSToby Isaac 633d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolve_SeqDense_Cholesky(Mat A, Vec xx, Vec yy) 634d71ae5a4SJacob Faibussowitsch { 635e54beecaSStefano Zampini PetscScalar *y = NULL; 636e54beecaSStefano Zampini PetscBLASInt m = 0, k = 0; 6374396437dSToby Isaac 6384396437dSToby Isaac PetscFunctionBegin; 6399566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_SetUp(A, xx, yy, &y, &m, &k)); 6409566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_Internal_Cholesky(A, y, m, m, 1, k, PETSC_FALSE)); 6419566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_TearDown(A, xx, yy, &y, &m, &k)); 6424396437dSToby Isaac PetscFunctionReturn(0); 6434396437dSToby Isaac } 6444396437dSToby Isaac 645d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolveTranspose_SeqDense_Cholesky(Mat A, Vec xx, Vec yy) 646d71ae5a4SJacob Faibussowitsch { 647e54beecaSStefano Zampini PetscScalar *y = NULL; 648e54beecaSStefano Zampini PetscBLASInt m = 0, k = 0; 6494396437dSToby Isaac 6504396437dSToby Isaac PetscFunctionBegin; 6519566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_SetUp(A, xx, yy, &y, &m, &k)); 6529566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_Internal_Cholesky(A, y, m, m, 1, k, PETSC_TRUE)); 6539566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_TearDown(A, xx, yy, &y, &m, &k)); 6544396437dSToby Isaac PetscFunctionReturn(0); 6554396437dSToby Isaac } 6564396437dSToby Isaac 657d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolve_SeqDense_QR(Mat A, Vec xx, Vec yy) 658d71ae5a4SJacob Faibussowitsch { 659e54beecaSStefano Zampini PetscScalar *y = NULL; 660e54beecaSStefano Zampini PetscBLASInt m = 0, k = 0; 6614396437dSToby Isaac 6624396437dSToby Isaac PetscFunctionBegin; 6639566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_SetUp(A, xx, yy, &y, &m, &k)); 6649566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_Internal_QR(A, y, PetscMax(m, k), m, 1, k)); 6659566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_TearDown(A, xx, yy, &y, &m, &k)); 6664396437dSToby Isaac PetscFunctionReturn(0); 6674396437dSToby Isaac } 6684396437dSToby Isaac 669d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSolveTranspose_SeqDense_QR(Mat A, Vec xx, Vec yy) 670d71ae5a4SJacob Faibussowitsch { 67142e9364cSSatish Balay PetscScalar *y = NULL; 67242e9364cSSatish Balay PetscBLASInt m = 0, k = 0; 6734396437dSToby Isaac 6744396437dSToby Isaac PetscFunctionBegin; 6759566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_SetUp(A, xx, yy, &y, &m, &k)); 6769566063dSJacob Faibussowitsch PetscCall(MatSolveTranspose_SeqDense_Internal_QR(A, y, PetscMax(m, k), m, 1, k)); 6779566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_TearDown(A, xx, yy, &y, &m, &k)); 6784396437dSToby Isaac PetscFunctionReturn(0); 6794396437dSToby Isaac } 6804396437dSToby Isaac 681d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMatSolve_SeqDense_SetUp(Mat A, Mat B, Mat X, PetscScalar **_y, PetscBLASInt *_ldy, PetscBLASInt *_m, PetscBLASInt *_nrhs, PetscBLASInt *_k) 682d71ae5a4SJacob Faibussowitsch { 6834905a7bcSToby Isaac const PetscScalar *b; 6844396437dSToby Isaac PetscScalar *y; 685bf5a80bcSToby Isaac PetscInt n, _ldb, _ldx; 686bf5a80bcSToby Isaac PetscBLASInt nrhs = 0, m = 0, k = 0, ldb = 0, ldx = 0, ldy = 0; 6874905a7bcSToby Isaac 6884905a7bcSToby Isaac PetscFunctionBegin; 6899371c9d4SSatish Balay *_ldy = 0; 6909371c9d4SSatish Balay *_m = 0; 6919371c9d4SSatish Balay *_nrhs = 0; 6929371c9d4SSatish Balay *_k = 0; 6939371c9d4SSatish Balay *_y = NULL; 6949566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->rmap->n, &m)); 6959566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->cmap->n, &k)); 6969566063dSJacob Faibussowitsch PetscCall(MatGetSize(B, NULL, &n)); 6979566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(n, &nrhs)); 6989566063dSJacob Faibussowitsch PetscCall(MatDenseGetLDA(B, &_ldb)); 6999566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(_ldb, &ldb)); 7009566063dSJacob Faibussowitsch PetscCall(MatDenseGetLDA(X, &_ldx)); 7019566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(_ldx, &ldx)); 702bf5a80bcSToby Isaac if (ldx < m) { 7039566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(B, &b)); 7049566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(nrhs * m, &y)); 705bf5a80bcSToby Isaac if (ldb == m) { 7069566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(y, b, ldb * nrhs)); 7074905a7bcSToby Isaac } else { 70848a46eb9SPierre Jolivet for (PetscInt j = 0; j < nrhs; j++) PetscCall(PetscArraycpy(&y[j * m], &b[j * ldb], m)); 7094905a7bcSToby Isaac } 710bf5a80bcSToby Isaac ldy = m; 7119566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(B, &b)); 7124905a7bcSToby Isaac } else { 713bf5a80bcSToby Isaac if (ldb == ldx) { 7149566063dSJacob Faibussowitsch PetscCall(MatCopy(B, X, SAME_NONZERO_PATTERN)); 7159566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(X, &y)); 7164905a7bcSToby Isaac } else { 7179566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(X, &y)); 7189566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(B, &b)); 71948a46eb9SPierre Jolivet for (PetscInt j = 0; j < nrhs; j++) PetscCall(PetscArraycpy(&y[j * ldx], &b[j * ldb], m)); 7209566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(B, &b)); 7214905a7bcSToby Isaac } 722bf5a80bcSToby Isaac ldy = ldx; 7234905a7bcSToby Isaac } 7244396437dSToby Isaac *_y = y; 725bf5a80bcSToby Isaac *_ldy = ldy; 7264396437dSToby Isaac *_k = k; 7274396437dSToby Isaac *_m = m; 7284396437dSToby Isaac *_nrhs = nrhs; 7294396437dSToby Isaac PetscFunctionReturn(0); 7304396437dSToby Isaac } 7314396437dSToby Isaac 732d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMatSolve_SeqDense_TearDown(Mat A, Mat B, Mat X, PetscScalar **_y, PetscBLASInt *_ldy, PetscBLASInt *_m, PetscBLASInt *_nrhs, PetscBLASInt *_k) 733d71ae5a4SJacob Faibussowitsch { 7344396437dSToby Isaac PetscScalar *y; 735bf5a80bcSToby Isaac PetscInt _ldx; 736bf5a80bcSToby Isaac PetscBLASInt k, ldy, nrhs, ldx = 0; 7374396437dSToby Isaac 7384396437dSToby Isaac PetscFunctionBegin; 7394396437dSToby Isaac y = *_y; 7404396437dSToby Isaac *_y = NULL; 7414396437dSToby Isaac k = *_k; 742bf5a80bcSToby Isaac ldy = *_ldy; 7434396437dSToby Isaac nrhs = *_nrhs; 7449566063dSJacob Faibussowitsch PetscCall(MatDenseGetLDA(X, &_ldx)); 7459566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(_ldx, &ldx)); 746bf5a80bcSToby Isaac if (ldx != ldy) { 7474905a7bcSToby Isaac PetscScalar *xv; 7489566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(X, &xv)); 74948a46eb9SPierre Jolivet for (PetscInt j = 0; j < nrhs; j++) PetscCall(PetscArraycpy(&xv[j * ldx], &y[j * ldy], k)); 7509566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(X, &xv)); 7519566063dSJacob Faibussowitsch PetscCall(PetscFree(y)); 7524905a7bcSToby Isaac } else { 7539566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(X, &y)); 7544905a7bcSToby Isaac } 75585e2c93fSHong Zhang PetscFunctionReturn(0); 75685e2c93fSHong Zhang } 75785e2c93fSHong Zhang 758d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMatSolve_SeqDense_LU(Mat A, Mat B, Mat X) 759d71ae5a4SJacob Faibussowitsch { 7604396437dSToby Isaac PetscScalar *y; 761bf5a80bcSToby Isaac PetscBLASInt m, k, ldy, nrhs; 7624396437dSToby Isaac 7634396437dSToby Isaac PetscFunctionBegin; 7649566063dSJacob Faibussowitsch PetscCall(MatMatSolve_SeqDense_SetUp(A, B, X, &y, &ldy, &m, &nrhs, &k)); 7659566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_Internal_LU(A, y, ldy, m, nrhs, k, PETSC_FALSE)); 7669566063dSJacob Faibussowitsch PetscCall(MatMatSolve_SeqDense_TearDown(A, B, X, &y, &ldy, &m, &nrhs, &k)); 7674396437dSToby Isaac PetscFunctionReturn(0); 7684396437dSToby Isaac } 7694396437dSToby Isaac 770d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMatSolveTranspose_SeqDense_LU(Mat A, Mat B, Mat X) 771d71ae5a4SJacob Faibussowitsch { 7724396437dSToby Isaac PetscScalar *y; 773bf5a80bcSToby Isaac PetscBLASInt m, k, ldy, nrhs; 7744396437dSToby Isaac 7754396437dSToby Isaac PetscFunctionBegin; 7769566063dSJacob Faibussowitsch PetscCall(MatMatSolve_SeqDense_SetUp(A, B, X, &y, &ldy, &m, &nrhs, &k)); 7779566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_Internal_LU(A, y, ldy, m, nrhs, k, PETSC_TRUE)); 7789566063dSJacob Faibussowitsch PetscCall(MatMatSolve_SeqDense_TearDown(A, B, X, &y, &ldy, &m, &nrhs, &k)); 7794396437dSToby Isaac PetscFunctionReturn(0); 7804396437dSToby Isaac } 7814396437dSToby Isaac 782d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMatSolve_SeqDense_Cholesky(Mat A, Mat B, Mat X) 783d71ae5a4SJacob Faibussowitsch { 7844396437dSToby Isaac PetscScalar *y; 785bf5a80bcSToby Isaac PetscBLASInt m, k, ldy, nrhs; 7864396437dSToby Isaac 7874396437dSToby Isaac PetscFunctionBegin; 7889566063dSJacob Faibussowitsch PetscCall(MatMatSolve_SeqDense_SetUp(A, B, X, &y, &ldy, &m, &nrhs, &k)); 7899566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_Internal_Cholesky(A, y, ldy, m, nrhs, k, PETSC_FALSE)); 7909566063dSJacob Faibussowitsch PetscCall(MatMatSolve_SeqDense_TearDown(A, B, X, &y, &ldy, &m, &nrhs, &k)); 7914396437dSToby Isaac PetscFunctionReturn(0); 7924396437dSToby Isaac } 7934396437dSToby Isaac 794d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMatSolveTranspose_SeqDense_Cholesky(Mat A, Mat B, Mat X) 795d71ae5a4SJacob Faibussowitsch { 7964396437dSToby Isaac PetscScalar *y; 797bf5a80bcSToby Isaac PetscBLASInt m, k, ldy, nrhs; 7984396437dSToby Isaac 7994396437dSToby Isaac PetscFunctionBegin; 8009566063dSJacob Faibussowitsch PetscCall(MatMatSolve_SeqDense_SetUp(A, B, X, &y, &ldy, &m, &nrhs, &k)); 8019566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_Internal_Cholesky(A, y, ldy, m, nrhs, k, PETSC_TRUE)); 8029566063dSJacob Faibussowitsch PetscCall(MatMatSolve_SeqDense_TearDown(A, B, X, &y, &ldy, &m, &nrhs, &k)); 8034396437dSToby Isaac PetscFunctionReturn(0); 8044396437dSToby Isaac } 8054396437dSToby Isaac 806d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMatSolve_SeqDense_QR(Mat A, Mat B, Mat X) 807d71ae5a4SJacob Faibussowitsch { 8084396437dSToby Isaac PetscScalar *y; 809bf5a80bcSToby Isaac PetscBLASInt m, k, ldy, nrhs; 8104396437dSToby Isaac 8114396437dSToby Isaac PetscFunctionBegin; 8129566063dSJacob Faibussowitsch PetscCall(MatMatSolve_SeqDense_SetUp(A, B, X, &y, &ldy, &m, &nrhs, &k)); 8139566063dSJacob Faibussowitsch PetscCall(MatSolve_SeqDense_Internal_QR(A, y, ldy, m, nrhs, k)); 8149566063dSJacob Faibussowitsch PetscCall(MatMatSolve_SeqDense_TearDown(A, B, X, &y, &ldy, &m, &nrhs, &k)); 8154396437dSToby Isaac PetscFunctionReturn(0); 8164396437dSToby Isaac } 8174396437dSToby Isaac 818d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMatSolveTranspose_SeqDense_QR(Mat A, Mat B, Mat X) 819d71ae5a4SJacob Faibussowitsch { 8204396437dSToby Isaac PetscScalar *y; 821bf5a80bcSToby Isaac PetscBLASInt m, k, ldy, nrhs; 8224396437dSToby Isaac 8234396437dSToby Isaac PetscFunctionBegin; 8249566063dSJacob Faibussowitsch PetscCall(MatMatSolve_SeqDense_SetUp(A, B, X, &y, &ldy, &m, &nrhs, &k)); 8259566063dSJacob Faibussowitsch PetscCall(MatSolveTranspose_SeqDense_Internal_QR(A, y, ldy, m, nrhs, k)); 8269566063dSJacob Faibussowitsch PetscCall(MatMatSolve_SeqDense_TearDown(A, B, X, &y, &ldy, &m, &nrhs, &k)); 8274396437dSToby Isaac PetscFunctionReturn(0); 8284396437dSToby Isaac } 8294396437dSToby Isaac 830db4efbfdSBarry Smith /* ---------------------------------------------------------------*/ 831db4efbfdSBarry Smith /* COMMENT: I have chosen to hide row permutation in the pivots, 832db4efbfdSBarry Smith rather than put it in the Mat->row slot.*/ 833d71ae5a4SJacob Faibussowitsch PetscErrorCode MatLUFactor_SeqDense(Mat A, IS row, IS col, const MatFactorInfo *minfo) 834d71ae5a4SJacob Faibussowitsch { 835db4efbfdSBarry Smith Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 836db4efbfdSBarry Smith PetscBLASInt n, m, info; 837db4efbfdSBarry Smith 838db4efbfdSBarry Smith PetscFunctionBegin; 8399566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->cmap->n, &n)); 8409566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->rmap->n, &m)); 8414dfa11a4SJacob Faibussowitsch if (!mat->pivots) { PetscCall(PetscMalloc1(A->rmap->n, &mat->pivots)); } 842db4efbfdSBarry Smith if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 8439566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 844792fecdfSBarry Smith PetscCallBLAS("LAPACKgetrf", LAPACKgetrf_(&m, &n, mat->v, &mat->lda, mat->pivots, &info)); 8459566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 8468e57ea43SSatish Balay 84705fcb23eSStefano Zampini PetscCheck(info >= 0, PETSC_COMM_SELF, PETSC_ERR_LIB, "Bad argument to LU factorization %d", (int)info); 84805fcb23eSStefano Zampini PetscCheck(info <= 0, PETSC_COMM_SELF, PETSC_ERR_MAT_LU_ZRPVT, "Bad LU factorization %d", (int)info); 8498208b9aeSStefano Zampini 8504396437dSToby Isaac A->ops->solve = MatSolve_SeqDense_LU; 8514396437dSToby Isaac A->ops->matsolve = MatMatSolve_SeqDense_LU; 8524396437dSToby Isaac A->ops->solvetranspose = MatSolveTranspose_SeqDense_LU; 8534396437dSToby Isaac A->ops->matsolvetranspose = MatMatSolveTranspose_SeqDense_LU; 854d5f3da31SBarry Smith A->factortype = MAT_FACTOR_LU; 855db4efbfdSBarry Smith 8569566063dSJacob Faibussowitsch PetscCall(PetscFree(A->solvertype)); 8579566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATSOLVERPETSC, &A->solvertype)); 858f6224b95SHong Zhang 8599566063dSJacob Faibussowitsch PetscCall(PetscLogFlops((2.0 * A->cmap->n * A->cmap->n * A->cmap->n) / 3)); 860db4efbfdSBarry Smith PetscFunctionReturn(0); 861db4efbfdSBarry Smith } 862db4efbfdSBarry Smith 863d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatLUFactorNumeric_SeqDense(Mat fact, Mat A, const MatFactorInfo *info_dummy) 864d71ae5a4SJacob Faibussowitsch { 8654396437dSToby Isaac MatFactorInfo info; 8664396437dSToby Isaac 8674396437dSToby Isaac PetscFunctionBegin; 8689566063dSJacob Faibussowitsch PetscCall(MatDuplicateNoCreate_SeqDense(fact, A, MAT_COPY_VALUES)); 869dbbe0bcdSBarry Smith PetscUseTypeMethod(fact, lufactor, NULL, NULL, &info); 8704396437dSToby Isaac PetscFunctionReturn(0); 8714396437dSToby Isaac } 8724396437dSToby Isaac 873d71ae5a4SJacob Faibussowitsch PetscErrorCode MatLUFactorSymbolic_SeqDense(Mat fact, Mat A, IS row, IS col, const MatFactorInfo *info) 874d71ae5a4SJacob Faibussowitsch { 8754396437dSToby Isaac PetscFunctionBegin; 8764396437dSToby Isaac fact->preallocated = PETSC_TRUE; 8774396437dSToby Isaac fact->assembled = PETSC_TRUE; 8784396437dSToby Isaac fact->ops->lufactornumeric = MatLUFactorNumeric_SeqDense; 8794396437dSToby Isaac PetscFunctionReturn(0); 8804396437dSToby Isaac } 8814396437dSToby Isaac 882a49dc2a2SStefano Zampini /* Cholesky as L*L^T or L*D*L^T and the symmetric/hermitian complex variants */ 883d71ae5a4SJacob Faibussowitsch PetscErrorCode MatCholeskyFactor_SeqDense(Mat A, IS perm, const MatFactorInfo *factinfo) 884d71ae5a4SJacob Faibussowitsch { 885db4efbfdSBarry Smith Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 886c5df96a5SBarry Smith PetscBLASInt info, n; 887db4efbfdSBarry Smith 888db4efbfdSBarry Smith PetscFunctionBegin; 8899566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->cmap->n, &n)); 890db4efbfdSBarry Smith if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 891b94d7dedSBarry Smith if (A->spd == PETSC_BOOL3_TRUE) { 8929566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 893792fecdfSBarry Smith PetscCallBLAS("LAPACKpotrf", LAPACKpotrf_("L", &n, mat->v, &mat->lda, &info)); 8949566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 895a49dc2a2SStefano Zampini #if defined(PETSC_USE_COMPLEX) 896b94d7dedSBarry Smith } else if (A->hermitian == PETSC_BOOL3_TRUE) { 8974dfa11a4SJacob Faibussowitsch if (!mat->pivots) { PetscCall(PetscMalloc1(A->rmap->n, &mat->pivots)); } 898a49dc2a2SStefano Zampini if (!mat->fwork) { 899a49dc2a2SStefano Zampini PetscScalar dummy; 900a49dc2a2SStefano Zampini 901a49dc2a2SStefano Zampini mat->lfwork = -1; 9029566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 903792fecdfSBarry Smith PetscCallBLAS("LAPACKhetrf", LAPACKhetrf_("L", &n, mat->v, &mat->lda, mat->pivots, &dummy, &mat->lfwork, &info)); 9049566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 905a49dc2a2SStefano Zampini mat->lfwork = (PetscInt)PetscRealPart(dummy); 9069566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(mat->lfwork, &mat->fwork)); 907a49dc2a2SStefano Zampini } 9089566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 909792fecdfSBarry Smith PetscCallBLAS("LAPACKhetrf", LAPACKhetrf_("L", &n, mat->v, &mat->lda, mat->pivots, mat->fwork, &mat->lfwork, &info)); 9109566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 911a49dc2a2SStefano Zampini #endif 912a49dc2a2SStefano Zampini } else { /* symmetric case */ 9134dfa11a4SJacob Faibussowitsch if (!mat->pivots) { PetscCall(PetscMalloc1(A->rmap->n, &mat->pivots)); } 914a49dc2a2SStefano Zampini if (!mat->fwork) { 915a49dc2a2SStefano Zampini PetscScalar dummy; 916a49dc2a2SStefano Zampini 917a49dc2a2SStefano Zampini mat->lfwork = -1; 9189566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 919792fecdfSBarry Smith PetscCallBLAS("LAPACKsytrf", LAPACKsytrf_("L", &n, mat->v, &mat->lda, mat->pivots, &dummy, &mat->lfwork, &info)); 9209566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 921a49dc2a2SStefano Zampini mat->lfwork = (PetscInt)PetscRealPart(dummy); 9229566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(mat->lfwork, &mat->fwork)); 923a49dc2a2SStefano Zampini } 9249566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 925792fecdfSBarry Smith PetscCallBLAS("LAPACKsytrf", LAPACKsytrf_("L", &n, mat->v, &mat->lda, mat->pivots, mat->fwork, &mat->lfwork, &info)); 9269566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 927a49dc2a2SStefano Zampini } 92828b400f6SJacob Faibussowitsch PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_MAT_CH_ZRPVT, "Bad factorization: zero pivot in row %" PetscInt_FMT, (PetscInt)info - 1); 9298208b9aeSStefano Zampini 9304396437dSToby Isaac A->ops->solve = MatSolve_SeqDense_Cholesky; 9314396437dSToby Isaac A->ops->matsolve = MatMatSolve_SeqDense_Cholesky; 9324396437dSToby Isaac A->ops->solvetranspose = MatSolveTranspose_SeqDense_Cholesky; 9334396437dSToby Isaac A->ops->matsolvetranspose = MatMatSolveTranspose_SeqDense_Cholesky; 934d5f3da31SBarry Smith A->factortype = MAT_FACTOR_CHOLESKY; 9352205254eSKarl Rupp 9369566063dSJacob Faibussowitsch PetscCall(PetscFree(A->solvertype)); 9379566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATSOLVERPETSC, &A->solvertype)); 938f6224b95SHong Zhang 9399566063dSJacob Faibussowitsch PetscCall(PetscLogFlops((1.0 * A->cmap->n * A->cmap->n * A->cmap->n) / 3.0)); 940db4efbfdSBarry Smith PetscFunctionReturn(0); 941db4efbfdSBarry Smith } 942db4efbfdSBarry Smith 943d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatCholeskyFactorNumeric_SeqDense(Mat fact, Mat A, const MatFactorInfo *info_dummy) 944d71ae5a4SJacob Faibussowitsch { 945db4efbfdSBarry Smith MatFactorInfo info; 946db4efbfdSBarry Smith 947db4efbfdSBarry Smith PetscFunctionBegin; 948db4efbfdSBarry Smith info.fill = 1.0; 9492205254eSKarl Rupp 9509566063dSJacob Faibussowitsch PetscCall(MatDuplicateNoCreate_SeqDense(fact, A, MAT_COPY_VALUES)); 951dbbe0bcdSBarry Smith PetscUseTypeMethod(fact, choleskyfactor, NULL, &info); 952db4efbfdSBarry Smith PetscFunctionReturn(0); 953db4efbfdSBarry Smith } 954db4efbfdSBarry Smith 955d71ae5a4SJacob Faibussowitsch PetscErrorCode MatCholeskyFactorSymbolic_SeqDense(Mat fact, Mat A, IS row, const MatFactorInfo *info) 956d71ae5a4SJacob Faibussowitsch { 957db4efbfdSBarry Smith PetscFunctionBegin; 958c3ef05f6SHong Zhang fact->assembled = PETSC_TRUE; 9591bbcc794SSatish Balay fact->preallocated = PETSC_TRUE; 960719d5645SBarry Smith fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqDense; 961db4efbfdSBarry Smith PetscFunctionReturn(0); 962db4efbfdSBarry Smith } 963db4efbfdSBarry Smith 964d71ae5a4SJacob Faibussowitsch PetscErrorCode MatQRFactor_SeqDense(Mat A, IS col, const MatFactorInfo *minfo) 965d71ae5a4SJacob Faibussowitsch { 9664905a7bcSToby Isaac Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 9674905a7bcSToby Isaac PetscBLASInt n, m, info, min, max; 9684905a7bcSToby Isaac 9694905a7bcSToby Isaac PetscFunctionBegin; 9709566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->cmap->n, &n)); 9719566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->rmap->n, &m)); 9724396437dSToby Isaac max = PetscMax(m, n); 9734396437dSToby Isaac min = PetscMin(m, n); 9744dfa11a4SJacob Faibussowitsch if (!mat->tau) { PetscCall(PetscMalloc1(min, &mat->tau)); } 9754dfa11a4SJacob Faibussowitsch if (!mat->pivots) { PetscCall(PetscMalloc1(n, &mat->pivots)); } 97648a46eb9SPierre Jolivet if (!mat->qrrhs) PetscCall(MatCreateVecs(A, NULL, &(mat->qrrhs))); 9774905a7bcSToby Isaac if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 9784905a7bcSToby Isaac if (!mat->fwork) { 9794905a7bcSToby Isaac PetscScalar dummy; 9804905a7bcSToby Isaac 9814905a7bcSToby Isaac mat->lfwork = -1; 9829566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 983792fecdfSBarry Smith PetscCallBLAS("LAPACKgeqrf", LAPACKgeqrf_(&m, &n, mat->v, &mat->lda, mat->tau, &dummy, &mat->lfwork, &info)); 9849566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 9854905a7bcSToby Isaac mat->lfwork = (PetscInt)PetscRealPart(dummy); 9869566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(mat->lfwork, &mat->fwork)); 9874905a7bcSToby Isaac } 9889566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPush(PETSC_FP_TRAP_OFF)); 989792fecdfSBarry Smith PetscCallBLAS("LAPACKgeqrf", LAPACKgeqrf_(&m, &n, mat->v, &mat->lda, mat->tau, mat->fwork, &mat->lfwork, &info)); 9909566063dSJacob Faibussowitsch PetscCall(PetscFPTrapPop()); 99105fcb23eSStefano Zampini PetscCheck(!info, PETSC_COMM_SELF, PETSC_ERR_LIB, "Bad argument to QR factorization %d", (int)info); 9924905a7bcSToby Isaac // TODO: try to estimate rank or test for and use geqp3 for rank revealing QR. For now just say rank is min of m and n 9934905a7bcSToby Isaac mat->rank = min; 9944905a7bcSToby Isaac 9954396437dSToby Isaac A->ops->solve = MatSolve_SeqDense_QR; 9964396437dSToby Isaac A->ops->matsolve = MatMatSolve_SeqDense_QR; 9974905a7bcSToby Isaac A->factortype = MAT_FACTOR_QR; 9984905a7bcSToby Isaac if (m == n) { 9994396437dSToby Isaac A->ops->solvetranspose = MatSolveTranspose_SeqDense_QR; 10004396437dSToby Isaac A->ops->matsolvetranspose = MatMatSolveTranspose_SeqDense_QR; 10014905a7bcSToby Isaac } 10024905a7bcSToby Isaac 10039566063dSJacob Faibussowitsch PetscCall(PetscFree(A->solvertype)); 10049566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATSOLVERPETSC, &A->solvertype)); 10054905a7bcSToby Isaac 10069566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(2.0 * min * min * (max - min / 3.0))); 10074905a7bcSToby Isaac PetscFunctionReturn(0); 10084905a7bcSToby Isaac } 10094905a7bcSToby Isaac 1010d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatQRFactorNumeric_SeqDense(Mat fact, Mat A, const MatFactorInfo *info_dummy) 1011d71ae5a4SJacob Faibussowitsch { 10124905a7bcSToby Isaac MatFactorInfo info; 10134905a7bcSToby Isaac 10144905a7bcSToby Isaac PetscFunctionBegin; 10154905a7bcSToby Isaac info.fill = 1.0; 10164905a7bcSToby Isaac 10179566063dSJacob Faibussowitsch PetscCall(MatDuplicateNoCreate_SeqDense(fact, A, MAT_COPY_VALUES)); 1018cac4c232SBarry Smith PetscUseMethod(fact, "MatQRFactor_C", (Mat, IS, const MatFactorInfo *), (fact, NULL, &info)); 10194905a7bcSToby Isaac PetscFunctionReturn(0); 10204905a7bcSToby Isaac } 10214905a7bcSToby Isaac 1022d71ae5a4SJacob Faibussowitsch PetscErrorCode MatQRFactorSymbolic_SeqDense(Mat fact, Mat A, IS row, const MatFactorInfo *info) 1023d71ae5a4SJacob Faibussowitsch { 10244905a7bcSToby Isaac PetscFunctionBegin; 10254905a7bcSToby Isaac fact->assembled = PETSC_TRUE; 10264905a7bcSToby Isaac fact->preallocated = PETSC_TRUE; 10279566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)fact, "MatQRFactorNumeric_C", MatQRFactorNumeric_SeqDense)); 10284905a7bcSToby Isaac PetscFunctionReturn(0); 10294905a7bcSToby Isaac } 10304905a7bcSToby Isaac 1031ca15aa20SStefano Zampini /* uses LAPACK */ 1032d71ae5a4SJacob Faibussowitsch PETSC_INTERN PetscErrorCode MatGetFactor_seqdense_petsc(Mat A, MatFactorType ftype, Mat *fact) 1033d71ae5a4SJacob Faibussowitsch { 1034db4efbfdSBarry Smith PetscFunctionBegin; 10359566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), fact)); 10369566063dSJacob Faibussowitsch PetscCall(MatSetSizes(*fact, A->rmap->n, A->cmap->n, A->rmap->n, A->cmap->n)); 10379566063dSJacob Faibussowitsch PetscCall(MatSetType(*fact, MATDENSE)); 103866e17bc3SBarry Smith (*fact)->trivialsymbolic = PETSC_TRUE; 10392a350339SBarry Smith if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU) { 1040db4efbfdSBarry Smith (*fact)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqDense; 10412a350339SBarry Smith (*fact)->ops->ilufactorsymbolic = MatLUFactorSymbolic_SeqDense; 1042bf5a80bcSToby Isaac } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 1043db4efbfdSBarry Smith (*fact)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqDense; 1044bf5a80bcSToby Isaac } else if (ftype == MAT_FACTOR_QR) { 10459566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)(*fact), "MatQRFactorSymbolic_C", MatQRFactorSymbolic_SeqDense)); 1046db4efbfdSBarry Smith } 1047d5f3da31SBarry Smith (*fact)->factortype = ftype; 104800c67f3bSHong Zhang 10499566063dSJacob Faibussowitsch PetscCall(PetscFree((*fact)->solvertype)); 10509566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATSOLVERPETSC, &(*fact)->solvertype)); 10519566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&(*fact)->preferredordering[MAT_FACTOR_LU])); 10529566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&(*fact)->preferredordering[MAT_FACTOR_ILU])); 10539566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&(*fact)->preferredordering[MAT_FACTOR_CHOLESKY])); 10549566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(MATORDERINGEXTERNAL, (char **)&(*fact)->preferredordering[MAT_FACTOR_ICC])); 1055db4efbfdSBarry Smith PetscFunctionReturn(0); 1056db4efbfdSBarry Smith } 1057db4efbfdSBarry Smith 1058289bc588SBarry Smith /* ------------------------------------------------------------------*/ 1059d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSOR_SeqDense(Mat A, Vec bb, PetscReal omega, MatSORType flag, PetscReal shift, PetscInt its, PetscInt lits, Vec xx) 1060d71ae5a4SJacob Faibussowitsch { 1061c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 1062d9ca1df4SBarry Smith PetscScalar *x, *v = mat->v, zero = 0.0, xt; 1063d9ca1df4SBarry Smith const PetscScalar *b; 1064d0f46423SBarry Smith PetscInt m = A->rmap->n, i; 106523fff9afSBarry Smith PetscBLASInt o = 1, bm = 0; 1066289bc588SBarry Smith 10673a40ed3dSBarry Smith PetscFunctionBegin; 106847d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) || defined(PETSC_HAVE_HIP) 106908401ef6SPierre Jolivet PetscCheck(A->offloadmask != PETSC_OFFLOAD_GPU, PETSC_COMM_SELF, PETSC_ERR_SUP, "Not implemented"); 1070ca15aa20SStefano Zampini #endif 1071422a814eSBarry Smith if (shift == -1) shift = 0.0; /* negative shift indicates do not error on zero diagonal; this code never zeros on zero diagonal */ 10729566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(m, &bm)); 1073289bc588SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 10743bffc371SBarry Smith /* this is a hack fix, should have another version without the second BLASdotu */ 10759566063dSJacob Faibussowitsch PetscCall(VecSet(xx, zero)); 1076289bc588SBarry Smith } 10779566063dSJacob Faibussowitsch PetscCall(VecGetArray(xx, &x)); 10789566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(bb, &b)); 1079b965ef7fSBarry Smith its = its * lits; 108008401ef6SPierre Jolivet PetscCheck(its > 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Relaxation requires global its %" PetscInt_FMT " and local its %" PetscInt_FMT " both positive", its, lits); 1081289bc588SBarry Smith while (its--) { 1082fccaa45eSBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) { 1083289bc588SBarry Smith for (i = 0; i < m; i++) { 1084792fecdfSBarry Smith PetscCallBLAS("BLASdotu", xt = b[i] - BLASdotu_(&bm, v + i, &bm, x, &o)); 108555a1b374SBarry Smith x[i] = (1. - omega) * x[i] + omega * (xt + v[i + i * m] * x[i]) / (v[i + i * m] + shift); 1086289bc588SBarry Smith } 1087289bc588SBarry Smith } 1088fccaa45eSBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) { 1089289bc588SBarry Smith for (i = m - 1; i >= 0; i--) { 1090792fecdfSBarry Smith PetscCallBLAS("BLASdotu", xt = b[i] - BLASdotu_(&bm, v + i, &bm, x, &o)); 109155a1b374SBarry Smith x[i] = (1. - omega) * x[i] + omega * (xt + v[i + i * m] * x[i]) / (v[i + i * m] + shift); 1092289bc588SBarry Smith } 1093289bc588SBarry Smith } 1094289bc588SBarry Smith } 10959566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(bb, &b)); 10969566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(xx, &x)); 10973a40ed3dSBarry Smith PetscFunctionReturn(0); 1098289bc588SBarry Smith } 1099289bc588SBarry Smith 1100289bc588SBarry Smith /* -----------------------------------------------------------------*/ 1101d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMultTranspose_SeqDense(Mat A, Vec xx, Vec yy) 1102d71ae5a4SJacob Faibussowitsch { 1103c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 1104d9ca1df4SBarry Smith const PetscScalar *v = mat->v, *x; 1105d9ca1df4SBarry Smith PetscScalar *y; 11060805154bSBarry Smith PetscBLASInt m, n, _One = 1; 1107ea709b57SSatish Balay PetscScalar _DOne = 1.0, _DZero = 0.0; 11083a40ed3dSBarry Smith 11093a40ed3dSBarry Smith PetscFunctionBegin; 11109566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->rmap->n, &m)); 11119566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->cmap->n, &n)); 11129566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(xx, &x)); 11139566063dSJacob Faibussowitsch PetscCall(VecGetArrayWrite(yy, &y)); 11145ac36cfcSBarry Smith if (!A->rmap->n || !A->cmap->n) { 11155ac36cfcSBarry Smith PetscBLASInt i; 11165ac36cfcSBarry Smith for (i = 0; i < n; i++) y[i] = 0.0; 11175ac36cfcSBarry Smith } else { 1118792fecdfSBarry Smith PetscCallBLAS("BLASgemv", BLASgemv_("T", &m, &n, &_DOne, v, &mat->lda, x, &_One, &_DZero, y, &_One)); 11199566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(2.0 * A->rmap->n * A->cmap->n - A->cmap->n)); 11205ac36cfcSBarry Smith } 11219566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(xx, &x)); 11229566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayWrite(yy, &y)); 11233a40ed3dSBarry Smith PetscFunctionReturn(0); 1124289bc588SBarry Smith } 1125800995b7SMatthew Knepley 1126d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMult_SeqDense(Mat A, Vec xx, Vec yy) 1127d71ae5a4SJacob Faibussowitsch { 1128c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 1129d9ca1df4SBarry Smith PetscScalar *y, _DOne = 1.0, _DZero = 0.0; 11300805154bSBarry Smith PetscBLASInt m, n, _One = 1; 1131d9ca1df4SBarry Smith const PetscScalar *v = mat->v, *x; 11323a40ed3dSBarry Smith 11333a40ed3dSBarry Smith PetscFunctionBegin; 11349566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->rmap->n, &m)); 11359566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->cmap->n, &n)); 11369566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(xx, &x)); 11379566063dSJacob Faibussowitsch PetscCall(VecGetArrayWrite(yy, &y)); 11385ac36cfcSBarry Smith if (!A->rmap->n || !A->cmap->n) { 11395ac36cfcSBarry Smith PetscBLASInt i; 11405ac36cfcSBarry Smith for (i = 0; i < m; i++) y[i] = 0.0; 11415ac36cfcSBarry Smith } else { 1142792fecdfSBarry Smith PetscCallBLAS("BLASgemv", BLASgemv_("N", &m, &n, &_DOne, v, &(mat->lda), x, &_One, &_DZero, y, &_One)); 11439566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(2.0 * A->rmap->n * A->cmap->n - A->rmap->n)); 11445ac36cfcSBarry Smith } 11459566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(xx, &x)); 11469566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayWrite(yy, &y)); 11473a40ed3dSBarry Smith PetscFunctionReturn(0); 1148289bc588SBarry Smith } 11496ee01492SSatish Balay 1150d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMultAdd_SeqDense(Mat A, Vec xx, Vec zz, Vec yy) 1151d71ae5a4SJacob Faibussowitsch { 1152c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 1153d9ca1df4SBarry Smith const PetscScalar *v = mat->v, *x; 1154d9ca1df4SBarry Smith PetscScalar *y, _DOne = 1.0; 11550805154bSBarry Smith PetscBLASInt m, n, _One = 1; 11563a40ed3dSBarry Smith 11573a40ed3dSBarry Smith PetscFunctionBegin; 11589566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->rmap->n, &m)); 11599566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->cmap->n, &n)); 11609566063dSJacob Faibussowitsch PetscCall(VecCopy(zz, yy)); 1161d0f46423SBarry Smith if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 11629566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(xx, &x)); 11639566063dSJacob Faibussowitsch PetscCall(VecGetArray(yy, &y)); 1164792fecdfSBarry Smith PetscCallBLAS("BLASgemv", BLASgemv_("N", &m, &n, &_DOne, v, &(mat->lda), x, &_One, &_DOne, y, &_One)); 11659566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(xx, &x)); 11669566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(yy, &y)); 11679566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(2.0 * A->rmap->n * A->cmap->n)); 11683a40ed3dSBarry Smith PetscFunctionReturn(0); 1169289bc588SBarry Smith } 11706ee01492SSatish Balay 1171d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMultTransposeAdd_SeqDense(Mat A, Vec xx, Vec zz, Vec yy) 1172d71ae5a4SJacob Faibussowitsch { 1173c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 1174d9ca1df4SBarry Smith const PetscScalar *v = mat->v, *x; 1175d9ca1df4SBarry Smith PetscScalar *y; 11760805154bSBarry Smith PetscBLASInt m, n, _One = 1; 117787828ca2SBarry Smith PetscScalar _DOne = 1.0; 11783a40ed3dSBarry Smith 11793a40ed3dSBarry Smith PetscFunctionBegin; 11809566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->rmap->n, &m)); 11819566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->cmap->n, &n)); 11829566063dSJacob Faibussowitsch PetscCall(VecCopy(zz, yy)); 1183d0f46423SBarry Smith if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 11849566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(xx, &x)); 11859566063dSJacob Faibussowitsch PetscCall(VecGetArray(yy, &y)); 1186792fecdfSBarry Smith PetscCallBLAS("BLASgemv", BLASgemv_("T", &m, &n, &_DOne, v, &(mat->lda), x, &_One, &_DOne, y, &_One)); 11879566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(xx, &x)); 11889566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(yy, &y)); 11899566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(2.0 * A->rmap->n * A->cmap->n)); 11903a40ed3dSBarry Smith PetscFunctionReturn(0); 1191289bc588SBarry Smith } 1192289bc588SBarry Smith 1193289bc588SBarry Smith /* -----------------------------------------------------------------*/ 1194d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetRow_SeqDense(Mat A, PetscInt row, PetscInt *ncols, PetscInt **cols, PetscScalar **vals) 1195d71ae5a4SJacob Faibussowitsch { 1196c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 119713f74950SBarry Smith PetscInt i; 119867e560aaSBarry Smith 11993a40ed3dSBarry Smith PetscFunctionBegin; 1200d0f46423SBarry Smith *ncols = A->cmap->n; 1201289bc588SBarry Smith if (cols) { 12029566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(A->cmap->n, cols)); 1203d0f46423SBarry Smith for (i = 0; i < A->cmap->n; i++) (*cols)[i] = i; 1204289bc588SBarry Smith } 1205289bc588SBarry Smith if (vals) { 1206ca15aa20SStefano Zampini const PetscScalar *v; 1207ca15aa20SStefano Zampini 12089566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &v)); 12099566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(A->cmap->n, vals)); 1210ca15aa20SStefano Zampini v += row; 12119371c9d4SSatish Balay for (i = 0; i < A->cmap->n; i++) { 12129371c9d4SSatish Balay (*vals)[i] = *v; 12139371c9d4SSatish Balay v += mat->lda; 12149371c9d4SSatish Balay } 12159566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &v)); 1216289bc588SBarry Smith } 12173a40ed3dSBarry Smith PetscFunctionReturn(0); 1218289bc588SBarry Smith } 12196ee01492SSatish Balay 1220d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatRestoreRow_SeqDense(Mat A, PetscInt row, PetscInt *ncols, PetscInt **cols, PetscScalar **vals) 1221d71ae5a4SJacob Faibussowitsch { 1222606d414cSSatish Balay PetscFunctionBegin; 1223cb4a9cd9SHong Zhang if (ncols) *ncols = 0; 12249566063dSJacob Faibussowitsch if (cols) PetscCall(PetscFree(*cols)); 12259566063dSJacob Faibussowitsch if (vals) PetscCall(PetscFree(*vals)); 12263a40ed3dSBarry Smith PetscFunctionReturn(0); 1227289bc588SBarry Smith } 1228289bc588SBarry Smith /* ----------------------------------------------------------------*/ 1229d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSetValues_SeqDense(Mat A, PetscInt m, const PetscInt indexm[], PetscInt n, const PetscInt indexn[], const PetscScalar v[], InsertMode addv) 1230d71ae5a4SJacob Faibussowitsch { 1231c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 1232ca15aa20SStefano Zampini PetscScalar *av; 123313f74950SBarry Smith PetscInt i, j, idx = 0; 123447d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) || defined(PETSC_HAVE_HIP) 1235c70f7ee4SJunchao Zhang PetscOffloadMask oldf; 1236ca15aa20SStefano Zampini #endif 1237d6dfbf8fSBarry Smith 12383a40ed3dSBarry Smith PetscFunctionBegin; 12399566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, &av)); 1240289bc588SBarry Smith if (!mat->roworiented) { 1241dbb450caSBarry Smith if (addv == INSERT_VALUES) { 1242289bc588SBarry Smith for (j = 0; j < n; j++) { 12439371c9d4SSatish Balay if (indexn[j] < 0) { 12449371c9d4SSatish Balay idx += m; 12459371c9d4SSatish Balay continue; 12469371c9d4SSatish Balay } 12476bdcaf15SBarry Smith PetscCheck(indexn[j] < A->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Column too large: col %" PetscInt_FMT " max %" PetscInt_FMT, indexn[j], A->cmap->n - 1); 1248289bc588SBarry Smith for (i = 0; i < m; i++) { 12499371c9d4SSatish Balay if (indexm[i] < 0) { 12509371c9d4SSatish Balay idx++; 12519371c9d4SSatish Balay continue; 12529371c9d4SSatish Balay } 12536bdcaf15SBarry Smith PetscCheck(indexm[i] < A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT, indexm[i], A->rmap->n - 1); 1254ca15aa20SStefano Zampini av[indexn[j] * mat->lda + indexm[i]] = v[idx++]; 1255289bc588SBarry Smith } 1256289bc588SBarry Smith } 12573a40ed3dSBarry Smith } else { 1258289bc588SBarry Smith for (j = 0; j < n; j++) { 12599371c9d4SSatish Balay if (indexn[j] < 0) { 12609371c9d4SSatish Balay idx += m; 12619371c9d4SSatish Balay continue; 12629371c9d4SSatish Balay } 12636bdcaf15SBarry Smith PetscCheck(indexn[j] < A->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Column too large: col %" PetscInt_FMT " max %" PetscInt_FMT, indexn[j], A->cmap->n - 1); 1264289bc588SBarry Smith for (i = 0; i < m; i++) { 12659371c9d4SSatish Balay if (indexm[i] < 0) { 12669371c9d4SSatish Balay idx++; 12679371c9d4SSatish Balay continue; 12689371c9d4SSatish Balay } 12696bdcaf15SBarry Smith PetscCheck(indexm[i] < A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT, indexm[i], A->rmap->n - 1); 1270ca15aa20SStefano Zampini av[indexn[j] * mat->lda + indexm[i]] += v[idx++]; 1271289bc588SBarry Smith } 1272289bc588SBarry Smith } 1273289bc588SBarry Smith } 12743a40ed3dSBarry Smith } else { 1275dbb450caSBarry Smith if (addv == INSERT_VALUES) { 1276e8d4e0b9SBarry Smith for (i = 0; i < m; i++) { 12779371c9d4SSatish Balay if (indexm[i] < 0) { 12789371c9d4SSatish Balay idx += n; 12799371c9d4SSatish Balay continue; 12809371c9d4SSatish Balay } 12816bdcaf15SBarry Smith PetscCheck(indexm[i] < A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT, indexm[i], A->rmap->n - 1); 1282e8d4e0b9SBarry Smith for (j = 0; j < n; j++) { 12839371c9d4SSatish Balay if (indexn[j] < 0) { 12849371c9d4SSatish Balay idx++; 12859371c9d4SSatish Balay continue; 12869371c9d4SSatish Balay } 12876bdcaf15SBarry Smith PetscCheck(indexn[j] < A->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Column too large: col %" PetscInt_FMT " max %" PetscInt_FMT, indexn[j], A->cmap->n - 1); 1288ca15aa20SStefano Zampini av[indexn[j] * mat->lda + indexm[i]] = v[idx++]; 1289e8d4e0b9SBarry Smith } 1290e8d4e0b9SBarry Smith } 12913a40ed3dSBarry Smith } else { 1292289bc588SBarry Smith for (i = 0; i < m; i++) { 12939371c9d4SSatish Balay if (indexm[i] < 0) { 12949371c9d4SSatish Balay idx += n; 12959371c9d4SSatish Balay continue; 12969371c9d4SSatish Balay } 12976bdcaf15SBarry Smith PetscCheck(indexm[i] < A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT, indexm[i], A->rmap->n - 1); 1298289bc588SBarry Smith for (j = 0; j < n; j++) { 12999371c9d4SSatish Balay if (indexn[j] < 0) { 13009371c9d4SSatish Balay idx++; 13019371c9d4SSatish Balay continue; 13029371c9d4SSatish Balay } 13036bdcaf15SBarry Smith PetscCheck(indexn[j] < A->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Column too large: col %" PetscInt_FMT " max %" PetscInt_FMT, indexn[j], A->cmap->n - 1); 1304ca15aa20SStefano Zampini av[indexn[j] * mat->lda + indexm[i]] += v[idx++]; 1305289bc588SBarry Smith } 1306289bc588SBarry Smith } 1307289bc588SBarry Smith } 1308e8d4e0b9SBarry Smith } 1309ca15aa20SStefano Zampini /* hack to prevent unneeded copy to the GPU while returning the array */ 131047d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) || defined(PETSC_HAVE_HIP) 1311c70f7ee4SJunchao Zhang oldf = A->offloadmask; 1312c70f7ee4SJunchao Zhang A->offloadmask = PETSC_OFFLOAD_GPU; 1313ca15aa20SStefano Zampini #endif 13149566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, &av)); 131547d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) || defined(PETSC_HAVE_HIP) 1316c70f7ee4SJunchao Zhang A->offloadmask = (oldf == PETSC_OFFLOAD_UNALLOCATED ? PETSC_OFFLOAD_UNALLOCATED : PETSC_OFFLOAD_CPU); 1317ca15aa20SStefano Zampini #endif 13183a40ed3dSBarry Smith PetscFunctionReturn(0); 1319289bc588SBarry Smith } 1320e8d4e0b9SBarry Smith 1321d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetValues_SeqDense(Mat A, PetscInt m, const PetscInt indexm[], PetscInt n, const PetscInt indexn[], PetscScalar v[]) 1322d71ae5a4SJacob Faibussowitsch { 1323ae80bb75SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 1324ca15aa20SStefano Zampini const PetscScalar *vv; 132513f74950SBarry Smith PetscInt i, j; 1326ae80bb75SLois Curfman McInnes 13273a40ed3dSBarry Smith PetscFunctionBegin; 13289566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &vv)); 1329ae80bb75SLois Curfman McInnes /* row-oriented output */ 1330ae80bb75SLois Curfman McInnes for (i = 0; i < m; i++) { 13319371c9d4SSatish Balay if (indexm[i] < 0) { 13329371c9d4SSatish Balay v += n; 13339371c9d4SSatish Balay continue; 13349371c9d4SSatish Balay } 133508401ef6SPierre Jolivet PetscCheck(indexm[i] < A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row %" PetscInt_FMT " requested larger than number rows %" PetscInt_FMT, indexm[i], A->rmap->n); 1336ae80bb75SLois Curfman McInnes for (j = 0; j < n; j++) { 13379371c9d4SSatish Balay if (indexn[j] < 0) { 13389371c9d4SSatish Balay v++; 13399371c9d4SSatish Balay continue; 13409371c9d4SSatish Balay } 134108401ef6SPierre Jolivet PetscCheck(indexn[j] < A->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Column %" PetscInt_FMT " requested larger than number columns %" PetscInt_FMT, indexn[j], A->cmap->n); 1342ca15aa20SStefano Zampini *v++ = vv[indexn[j] * mat->lda + indexm[i]]; 1343ae80bb75SLois Curfman McInnes } 1344ae80bb75SLois Curfman McInnes } 13459566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &vv)); 13463a40ed3dSBarry Smith PetscFunctionReturn(0); 1347ae80bb75SLois Curfman McInnes } 1348ae80bb75SLois Curfman McInnes 1349289bc588SBarry Smith /* -----------------------------------------------------------------*/ 1350289bc588SBarry Smith 1351d71ae5a4SJacob Faibussowitsch PetscErrorCode MatView_Dense_Binary(Mat mat, PetscViewer viewer) 1352d71ae5a4SJacob Faibussowitsch { 13538491ab44SLisandro Dalcin PetscBool skipHeader; 13548491ab44SLisandro Dalcin PetscViewerFormat format; 13558491ab44SLisandro Dalcin PetscInt header[4], M, N, m, lda, i, j, k; 13568491ab44SLisandro Dalcin const PetscScalar *v; 13578491ab44SLisandro Dalcin PetscScalar *vwork; 1358aabbc4fbSShri Abhyankar 1359aabbc4fbSShri Abhyankar PetscFunctionBegin; 13609566063dSJacob Faibussowitsch PetscCall(PetscViewerSetUp(viewer)); 13619566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryGetSkipHeader(viewer, &skipHeader)); 13629566063dSJacob Faibussowitsch PetscCall(PetscViewerGetFormat(viewer, &format)); 13638491ab44SLisandro Dalcin if (skipHeader) format = PETSC_VIEWER_NATIVE; 1364aabbc4fbSShri Abhyankar 13659566063dSJacob Faibussowitsch PetscCall(MatGetSize(mat, &M, &N)); 13668491ab44SLisandro Dalcin 13678491ab44SLisandro Dalcin /* write matrix header */ 13689371c9d4SSatish Balay header[0] = MAT_FILE_CLASSID; 13699371c9d4SSatish Balay header[1] = M; 13709371c9d4SSatish Balay header[2] = N; 13718491ab44SLisandro Dalcin header[3] = (format == PETSC_VIEWER_NATIVE) ? MATRIX_BINARY_FORMAT_DENSE : M * N; 13729566063dSJacob Faibussowitsch if (!skipHeader) PetscCall(PetscViewerBinaryWrite(viewer, header, 4, PETSC_INT)); 13738491ab44SLisandro Dalcin 13749566063dSJacob Faibussowitsch PetscCall(MatGetLocalSize(mat, &m, NULL)); 13758491ab44SLisandro Dalcin if (format != PETSC_VIEWER_NATIVE) { 13768491ab44SLisandro Dalcin PetscInt nnz = m * N, *iwork; 13778491ab44SLisandro Dalcin /* store row lengths for each row */ 13789566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(nnz, &iwork)); 13798491ab44SLisandro Dalcin for (i = 0; i < m; i++) iwork[i] = N; 13809566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryWriteAll(viewer, iwork, m, PETSC_DETERMINE, PETSC_DETERMINE, PETSC_INT)); 13818491ab44SLisandro Dalcin /* store column indices (zero start index) */ 13828491ab44SLisandro Dalcin for (k = 0, i = 0; i < m; i++) 13839371c9d4SSatish Balay for (j = 0; j < N; j++, k++) iwork[k] = j; 13849566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryWriteAll(viewer, iwork, nnz, PETSC_DETERMINE, PETSC_DETERMINE, PETSC_INT)); 13859566063dSJacob Faibussowitsch PetscCall(PetscFree(iwork)); 13868491ab44SLisandro Dalcin } 13878491ab44SLisandro Dalcin /* store matrix values as a dense matrix in row major order */ 13889566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(m * N, &vwork)); 13899566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(mat, &v)); 13909566063dSJacob Faibussowitsch PetscCall(MatDenseGetLDA(mat, &lda)); 13918491ab44SLisandro Dalcin for (k = 0, i = 0; i < m; i++) 13929371c9d4SSatish Balay for (j = 0; j < N; j++, k++) vwork[k] = v[i + lda * j]; 13939566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(mat, &v)); 13949566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryWriteAll(viewer, vwork, m * N, PETSC_DETERMINE, PETSC_DETERMINE, PETSC_SCALAR)); 13959566063dSJacob Faibussowitsch PetscCall(PetscFree(vwork)); 13968491ab44SLisandro Dalcin PetscFunctionReturn(0); 13978491ab44SLisandro Dalcin } 13988491ab44SLisandro Dalcin 1399d71ae5a4SJacob Faibussowitsch PetscErrorCode MatLoad_Dense_Binary(Mat mat, PetscViewer viewer) 1400d71ae5a4SJacob Faibussowitsch { 14018491ab44SLisandro Dalcin PetscBool skipHeader; 14028491ab44SLisandro Dalcin PetscInt header[4], M, N, m, nz, lda, i, j, k; 14038491ab44SLisandro Dalcin PetscInt rows, cols; 14048491ab44SLisandro Dalcin PetscScalar *v, *vwork; 14058491ab44SLisandro Dalcin 14068491ab44SLisandro Dalcin PetscFunctionBegin; 14079566063dSJacob Faibussowitsch PetscCall(PetscViewerSetUp(viewer)); 14089566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryGetSkipHeader(viewer, &skipHeader)); 14098491ab44SLisandro Dalcin 14108491ab44SLisandro Dalcin if (!skipHeader) { 14119566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryRead(viewer, header, 4, NULL, PETSC_INT)); 141208401ef6SPierre Jolivet PetscCheck(header[0] == MAT_FILE_CLASSID, PetscObjectComm((PetscObject)viewer), PETSC_ERR_FILE_UNEXPECTED, "Not a matrix object in file"); 14139371c9d4SSatish Balay M = header[1]; 14149371c9d4SSatish Balay N = header[2]; 141508401ef6SPierre Jolivet PetscCheck(M >= 0, PetscObjectComm((PetscObject)viewer), PETSC_ERR_FILE_UNEXPECTED, "Matrix row size (%" PetscInt_FMT ") in file is negative", M); 141608401ef6SPierre Jolivet PetscCheck(N >= 0, PetscObjectComm((PetscObject)viewer), PETSC_ERR_FILE_UNEXPECTED, "Matrix column size (%" PetscInt_FMT ") in file is negative", N); 14178491ab44SLisandro Dalcin nz = header[3]; 1418aed4548fSBarry Smith PetscCheck(nz == MATRIX_BINARY_FORMAT_DENSE || nz >= 0, PetscObjectComm((PetscObject)viewer), PETSC_ERR_FILE_UNEXPECTED, "Unknown matrix format %" PetscInt_FMT " in file", nz); 1419aabbc4fbSShri Abhyankar } else { 14209566063dSJacob Faibussowitsch PetscCall(MatGetSize(mat, &M, &N)); 1421aed4548fSBarry Smith PetscCheck(M >= 0 && N >= 0, PETSC_COMM_SELF, PETSC_ERR_USER, "Matrix binary file header was skipped, thus the user must specify the global sizes of input matrix"); 14228491ab44SLisandro Dalcin nz = MATRIX_BINARY_FORMAT_DENSE; 1423e6324fbbSBarry Smith } 1424aabbc4fbSShri Abhyankar 14258491ab44SLisandro Dalcin /* setup global sizes if not set */ 14268491ab44SLisandro Dalcin if (mat->rmap->N < 0) mat->rmap->N = M; 14278491ab44SLisandro Dalcin if (mat->cmap->N < 0) mat->cmap->N = N; 14289566063dSJacob Faibussowitsch PetscCall(MatSetUp(mat)); 14298491ab44SLisandro Dalcin /* check if global sizes are correct */ 14309566063dSJacob Faibussowitsch PetscCall(MatGetSize(mat, &rows, &cols)); 1431aed4548fSBarry Smith PetscCheck(M == rows && N == cols, PetscObjectComm((PetscObject)viewer), PETSC_ERR_FILE_UNEXPECTED, "Matrix in file of different sizes (%" PetscInt_FMT ", %" PetscInt_FMT ") than the input matrix (%" PetscInt_FMT ", %" PetscInt_FMT ")", M, N, rows, cols); 1432aabbc4fbSShri Abhyankar 14339566063dSJacob Faibussowitsch PetscCall(MatGetSize(mat, NULL, &N)); 14349566063dSJacob Faibussowitsch PetscCall(MatGetLocalSize(mat, &m, NULL)); 14359566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(mat, &v)); 14369566063dSJacob Faibussowitsch PetscCall(MatDenseGetLDA(mat, &lda)); 14378491ab44SLisandro Dalcin if (nz == MATRIX_BINARY_FORMAT_DENSE) { /* matrix in file is dense format */ 14388491ab44SLisandro Dalcin PetscInt nnz = m * N; 14398491ab44SLisandro Dalcin /* read in matrix values */ 14409566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(nnz, &vwork)); 14419566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryReadAll(viewer, vwork, nnz, PETSC_DETERMINE, PETSC_DETERMINE, PETSC_SCALAR)); 14428491ab44SLisandro Dalcin /* store values in column major order */ 14438491ab44SLisandro Dalcin for (j = 0; j < N; j++) 14449371c9d4SSatish Balay for (i = 0; i < m; i++) v[i + lda * j] = vwork[i * N + j]; 14459566063dSJacob Faibussowitsch PetscCall(PetscFree(vwork)); 14468491ab44SLisandro Dalcin } else { /* matrix in file is sparse format */ 14478491ab44SLisandro Dalcin PetscInt nnz = 0, *rlens, *icols; 14488491ab44SLisandro Dalcin /* read in row lengths */ 14499566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(m, &rlens)); 14509566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryReadAll(viewer, rlens, m, PETSC_DETERMINE, PETSC_DETERMINE, PETSC_INT)); 14518491ab44SLisandro Dalcin for (i = 0; i < m; i++) nnz += rlens[i]; 14528491ab44SLisandro Dalcin /* read in column indices and values */ 14539566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(nnz, &icols, nnz, &vwork)); 14549566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryReadAll(viewer, icols, nnz, PETSC_DETERMINE, PETSC_DETERMINE, PETSC_INT)); 14559566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryReadAll(viewer, vwork, nnz, PETSC_DETERMINE, PETSC_DETERMINE, PETSC_SCALAR)); 14568491ab44SLisandro Dalcin /* store values in column major order */ 14578491ab44SLisandro Dalcin for (k = 0, i = 0; i < m; i++) 14589371c9d4SSatish Balay for (j = 0; j < rlens[i]; j++, k++) v[i + lda * icols[k]] = vwork[k]; 14599566063dSJacob Faibussowitsch PetscCall(PetscFree(rlens)); 14609566063dSJacob Faibussowitsch PetscCall(PetscFree2(icols, vwork)); 1461aabbc4fbSShri Abhyankar } 14629566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(mat, &v)); 14639566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(mat, MAT_FINAL_ASSEMBLY)); 14649566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(mat, MAT_FINAL_ASSEMBLY)); 1465aabbc4fbSShri Abhyankar PetscFunctionReturn(0); 1466aabbc4fbSShri Abhyankar } 1467aabbc4fbSShri Abhyankar 1468d71ae5a4SJacob Faibussowitsch PetscErrorCode MatLoad_SeqDense(Mat newMat, PetscViewer viewer) 1469d71ae5a4SJacob Faibussowitsch { 1470eb91f321SVaclav Hapla PetscBool isbinary, ishdf5; 1471eb91f321SVaclav Hapla 1472eb91f321SVaclav Hapla PetscFunctionBegin; 1473eb91f321SVaclav Hapla PetscValidHeaderSpecific(newMat, MAT_CLASSID, 1); 1474eb91f321SVaclav Hapla PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 1475eb91f321SVaclav Hapla /* force binary viewer to load .info file if it has not yet done so */ 14769566063dSJacob Faibussowitsch PetscCall(PetscViewerSetUp(viewer)); 14779566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERBINARY, &isbinary)); 14789566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERHDF5, &ishdf5)); 1479eb91f321SVaclav Hapla if (isbinary) { 14809566063dSJacob Faibussowitsch PetscCall(MatLoad_Dense_Binary(newMat, viewer)); 1481eb91f321SVaclav Hapla } else if (ishdf5) { 1482eb91f321SVaclav Hapla #if defined(PETSC_HAVE_HDF5) 14839566063dSJacob Faibussowitsch PetscCall(MatLoad_Dense_HDF5(newMat, viewer)); 1484eb91f321SVaclav Hapla #else 1485eb91f321SVaclav Hapla SETERRQ(PetscObjectComm((PetscObject)newMat), PETSC_ERR_SUP, "HDF5 not supported in this build.\nPlease reconfigure using --download-hdf5"); 1486eb91f321SVaclav Hapla #endif 1487eb91f321SVaclav Hapla } else { 148898921bdaSJacob Faibussowitsch SETERRQ(PetscObjectComm((PetscObject)newMat), PETSC_ERR_SUP, "Viewer type %s not yet supported for reading %s matrices", ((PetscObject)viewer)->type_name, ((PetscObject)newMat)->type_name); 1489eb91f321SVaclav Hapla } 1490eb91f321SVaclav Hapla PetscFunctionReturn(0); 1491eb91f321SVaclav Hapla } 1492eb91f321SVaclav Hapla 1493d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatView_SeqDense_ASCII(Mat A, PetscViewer viewer) 1494d71ae5a4SJacob Faibussowitsch { 1495932b0c3eSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense *)A->data; 149613f74950SBarry Smith PetscInt i, j; 14972dcb1b2aSMatthew Knepley const char *name; 1498ca15aa20SStefano Zampini PetscScalar *v, *av; 1499f3ef73ceSBarry Smith PetscViewerFormat format; 15005f481a85SSatish Balay #if defined(PETSC_USE_COMPLEX) 1501ace3abfcSBarry Smith PetscBool allreal = PETSC_TRUE; 15025f481a85SSatish Balay #endif 1503932b0c3eSLois Curfman McInnes 15043a40ed3dSBarry Smith PetscFunctionBegin; 15059566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, (const PetscScalar **)&av)); 15069566063dSJacob Faibussowitsch PetscCall(PetscViewerGetFormat(viewer, &format)); 1507456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 15083a40ed3dSBarry Smith PetscFunctionReturn(0); /* do nothing for now */ 1509fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_COMMON) { 15109566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer, PETSC_FALSE)); 1511d0f46423SBarry Smith for (i = 0; i < A->rmap->n; i++) { 1512ca15aa20SStefano Zampini v = av + i; 15139566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "row %" PetscInt_FMT ":", i)); 1514d0f46423SBarry Smith for (j = 0; j < A->cmap->n; j++) { 1515aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 1516329f5518SBarry Smith if (PetscRealPart(*v) != 0.0 && PetscImaginaryPart(*v) != 0.0) { 15179566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " (%" PetscInt_FMT ", %g + %g i) ", j, (double)PetscRealPart(*v), (double)PetscImaginaryPart(*v))); 1518329f5518SBarry Smith } else if (PetscRealPart(*v)) { 15199566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " (%" PetscInt_FMT ", %g) ", j, (double)PetscRealPart(*v))); 15206831982aSBarry Smith } 152180cd9d93SLois Curfman McInnes #else 152248a46eb9SPierre Jolivet if (*v) PetscCall(PetscViewerASCIIPrintf(viewer, " (%" PetscInt_FMT ", %g) ", j, (double)*v)); 152380cd9d93SLois Curfman McInnes #endif 15241b807ce4Svictorle v += a->lda; 152580cd9d93SLois Curfman McInnes } 15269566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "\n")); 152780cd9d93SLois Curfman McInnes } 15289566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer, PETSC_TRUE)); 15293a40ed3dSBarry Smith } else { 15309566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer, PETSC_FALSE)); 1531aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 153247989497SBarry Smith /* determine if matrix has all real values */ 1533bcd8d3a4SJose E. Roman for (j = 0; j < A->cmap->n; j++) { 1534bcd8d3a4SJose E. Roman v = av + j * a->lda; 1535bcd8d3a4SJose E. Roman for (i = 0; i < A->rmap->n; i++) { 15369371c9d4SSatish Balay if (PetscImaginaryPart(v[i])) { 15379371c9d4SSatish Balay allreal = PETSC_FALSE; 15389371c9d4SSatish Balay break; 15399371c9d4SSatish Balay } 154047989497SBarry Smith } 1541bcd8d3a4SJose E. Roman } 154247989497SBarry Smith #endif 1543fb9695e5SSatish Balay if (format == PETSC_VIEWER_ASCII_MATLAB) { 15449566063dSJacob Faibussowitsch PetscCall(PetscObjectGetName((PetscObject)A, &name)); 15459566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "%% Size = %" PetscInt_FMT " %" PetscInt_FMT " \n", A->rmap->n, A->cmap->n)); 15469566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "%s = zeros(%" PetscInt_FMT ",%" PetscInt_FMT ");\n", name, A->rmap->n, A->cmap->n)); 15479566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "%s = [\n", name)); 1548ffac6cdbSBarry Smith } 1549ffac6cdbSBarry Smith 1550d0f46423SBarry Smith for (i = 0; i < A->rmap->n; i++) { 1551ca15aa20SStefano Zampini v = av + i; 1552d0f46423SBarry Smith for (j = 0; j < A->cmap->n; j++) { 1553aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 155447989497SBarry Smith if (allreal) { 15559566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "%18.16e ", (double)PetscRealPart(*v))); 155647989497SBarry Smith } else { 15579566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "%18.16e + %18.16ei ", (double)PetscRealPart(*v), (double)PetscImaginaryPart(*v))); 155847989497SBarry Smith } 1559289bc588SBarry Smith #else 15609566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "%18.16e ", (double)*v)); 1561289bc588SBarry Smith #endif 15621b807ce4Svictorle v += a->lda; 1563289bc588SBarry Smith } 15649566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "\n")); 1565289bc588SBarry Smith } 156648a46eb9SPierre Jolivet if (format == PETSC_VIEWER_ASCII_MATLAB) PetscCall(PetscViewerASCIIPrintf(viewer, "];\n")); 15679566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIUseTabs(viewer, PETSC_TRUE)); 1568da3a660dSBarry Smith } 15699566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, (const PetscScalar **)&av)); 15709566063dSJacob Faibussowitsch PetscCall(PetscViewerFlush(viewer)); 15713a40ed3dSBarry Smith PetscFunctionReturn(0); 1572289bc588SBarry Smith } 1573289bc588SBarry Smith 15749804daf3SBarry Smith #include <petscdraw.h> 1575d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatView_SeqDense_Draw_Zoom(PetscDraw draw, void *Aa) 1576d71ae5a4SJacob Faibussowitsch { 1577f1af5d2fSBarry Smith Mat A = (Mat)Aa; 1578383922c3SLisandro Dalcin PetscInt m = A->rmap->n, n = A->cmap->n, i, j; 1579383922c3SLisandro Dalcin int color = PETSC_DRAW_WHITE; 1580ca15aa20SStefano Zampini const PetscScalar *v; 1581b0a32e0cSBarry Smith PetscViewer viewer; 1582b05fc000SLisandro Dalcin PetscReal xl, yl, xr, yr, x_l, x_r, y_l, y_r; 1583f3ef73ceSBarry Smith PetscViewerFormat format; 1584f1af5d2fSBarry Smith 1585f1af5d2fSBarry Smith PetscFunctionBegin; 15869566063dSJacob Faibussowitsch PetscCall(PetscObjectQuery((PetscObject)A, "Zoomviewer", (PetscObject *)&viewer)); 15879566063dSJacob Faibussowitsch PetscCall(PetscViewerGetFormat(viewer, &format)); 15889566063dSJacob Faibussowitsch PetscCall(PetscDrawGetCoordinates(draw, &xl, &yl, &xr, &yr)); 1589f1af5d2fSBarry Smith 1590f1af5d2fSBarry Smith /* Loop over matrix elements drawing boxes */ 15919566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &v)); 1592fb9695e5SSatish Balay if (format != PETSC_VIEWER_DRAW_CONTOUR) { 1593d0609cedSBarry Smith PetscDrawCollectiveBegin(draw); 1594f1af5d2fSBarry Smith /* Blue for negative and Red for positive */ 1595f1af5d2fSBarry Smith for (j = 0; j < n; j++) { 15969371c9d4SSatish Balay x_l = j; 15979371c9d4SSatish Balay x_r = x_l + 1.0; 1598f1af5d2fSBarry Smith for (i = 0; i < m; i++) { 1599f1af5d2fSBarry Smith y_l = m - i - 1.0; 1600f1af5d2fSBarry Smith y_r = y_l + 1.0; 1601ca15aa20SStefano Zampini if (PetscRealPart(v[j * m + i]) > 0.) color = PETSC_DRAW_RED; 1602ca15aa20SStefano Zampini else if (PetscRealPart(v[j * m + i]) < 0.) color = PETSC_DRAW_BLUE; 1603ca15aa20SStefano Zampini else continue; 16049566063dSJacob Faibussowitsch PetscCall(PetscDrawRectangle(draw, x_l, y_l, x_r, y_r, color, color, color, color)); 1605f1af5d2fSBarry Smith } 1606f1af5d2fSBarry Smith } 1607d0609cedSBarry Smith PetscDrawCollectiveEnd(draw); 1608f1af5d2fSBarry Smith } else { 1609f1af5d2fSBarry Smith /* use contour shading to indicate magnitude of values */ 1610f1af5d2fSBarry Smith /* first determine max of all nonzero values */ 1611b05fc000SLisandro Dalcin PetscReal minv = 0.0, maxv = 0.0; 1612b05fc000SLisandro Dalcin PetscDraw popup; 1613b05fc000SLisandro Dalcin 1614f1af5d2fSBarry Smith for (i = 0; i < m * n; i++) { 1615f1af5d2fSBarry Smith if (PetscAbsScalar(v[i]) > maxv) maxv = PetscAbsScalar(v[i]); 1616f1af5d2fSBarry Smith } 1617383922c3SLisandro Dalcin if (minv >= maxv) maxv = minv + PETSC_SMALL; 16189566063dSJacob Faibussowitsch PetscCall(PetscDrawGetPopup(draw, &popup)); 16199566063dSJacob Faibussowitsch PetscCall(PetscDrawScalePopup(popup, minv, maxv)); 1620383922c3SLisandro Dalcin 1621d0609cedSBarry Smith PetscDrawCollectiveBegin(draw); 1622f1af5d2fSBarry Smith for (j = 0; j < n; j++) { 1623f1af5d2fSBarry Smith x_l = j; 1624f1af5d2fSBarry Smith x_r = x_l + 1.0; 1625f1af5d2fSBarry Smith for (i = 0; i < m; i++) { 1626f1af5d2fSBarry Smith y_l = m - i - 1.0; 1627f1af5d2fSBarry Smith y_r = y_l + 1.0; 1628b05fc000SLisandro Dalcin color = PetscDrawRealToColor(PetscAbsScalar(v[j * m + i]), minv, maxv); 16299566063dSJacob Faibussowitsch PetscCall(PetscDrawRectangle(draw, x_l, y_l, x_r, y_r, color, color, color, color)); 1630f1af5d2fSBarry Smith } 1631f1af5d2fSBarry Smith } 1632d0609cedSBarry Smith PetscDrawCollectiveEnd(draw); 1633f1af5d2fSBarry Smith } 16349566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &v)); 1635f1af5d2fSBarry Smith PetscFunctionReturn(0); 1636f1af5d2fSBarry Smith } 1637f1af5d2fSBarry Smith 1638d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatView_SeqDense_Draw(Mat A, PetscViewer viewer) 1639d71ae5a4SJacob Faibussowitsch { 1640b0a32e0cSBarry Smith PetscDraw draw; 1641ace3abfcSBarry Smith PetscBool isnull; 1642329f5518SBarry Smith PetscReal xr, yr, xl, yl, h, w; 1643f1af5d2fSBarry Smith 1644f1af5d2fSBarry Smith PetscFunctionBegin; 16459566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawGetDraw(viewer, 0, &draw)); 16469566063dSJacob Faibussowitsch PetscCall(PetscDrawIsNull(draw, &isnull)); 1647abc0a331SBarry Smith if (isnull) PetscFunctionReturn(0); 1648f1af5d2fSBarry Smith 16499371c9d4SSatish Balay xr = A->cmap->n; 16509371c9d4SSatish Balay yr = A->rmap->n; 16519371c9d4SSatish Balay h = yr / 10.0; 16529371c9d4SSatish Balay w = xr / 10.0; 16539371c9d4SSatish Balay xr += w; 16549371c9d4SSatish Balay yr += h; 16559371c9d4SSatish Balay xl = -w; 16569371c9d4SSatish Balay yl = -h; 16579566063dSJacob Faibussowitsch PetscCall(PetscDrawSetCoordinates(draw, xl, yl, xr, yr)); 16589566063dSJacob Faibussowitsch PetscCall(PetscObjectCompose((PetscObject)A, "Zoomviewer", (PetscObject)viewer)); 16599566063dSJacob Faibussowitsch PetscCall(PetscDrawZoom(draw, MatView_SeqDense_Draw_Zoom, A)); 16609566063dSJacob Faibussowitsch PetscCall(PetscObjectCompose((PetscObject)A, "Zoomviewer", NULL)); 16619566063dSJacob Faibussowitsch PetscCall(PetscDrawSave(draw)); 1662f1af5d2fSBarry Smith PetscFunctionReturn(0); 1663f1af5d2fSBarry Smith } 1664f1af5d2fSBarry Smith 1665d71ae5a4SJacob Faibussowitsch PetscErrorCode MatView_SeqDense(Mat A, PetscViewer viewer) 1666d71ae5a4SJacob Faibussowitsch { 1667ace3abfcSBarry Smith PetscBool iascii, isbinary, isdraw; 1668932b0c3eSLois Curfman McInnes 16693a40ed3dSBarry Smith PetscFunctionBegin; 16709566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii)); 16719566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERBINARY, &isbinary)); 16729566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERDRAW, &isdraw)); 16731baa6e33SBarry Smith if (iascii) PetscCall(MatView_SeqDense_ASCII(A, viewer)); 16741baa6e33SBarry Smith else if (isbinary) PetscCall(MatView_Dense_Binary(A, viewer)); 16751baa6e33SBarry Smith else if (isdraw) PetscCall(MatView_SeqDense_Draw(A, viewer)); 16763a40ed3dSBarry Smith PetscFunctionReturn(0); 1677932b0c3eSLois Curfman McInnes } 1678289bc588SBarry Smith 1679d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatDensePlaceArray_SeqDense(Mat A, const PetscScalar *array) 1680d71ae5a4SJacob Faibussowitsch { 1681d3042a70SBarry Smith Mat_SeqDense *a = (Mat_SeqDense *)A->data; 1682d3042a70SBarry Smith 1683d3042a70SBarry Smith PetscFunctionBegin; 168428b400f6SJacob Faibussowitsch PetscCheck(!a->vecinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreColumnVec() first"); 168528b400f6SJacob Faibussowitsch PetscCheck(!a->matinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreSubMatrix() first"); 168628b400f6SJacob Faibussowitsch PetscCheck(!a->unplacedarray, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreArray() first"); 1687d3042a70SBarry Smith a->unplacedarray = a->v; 1688d3042a70SBarry Smith a->unplaced_user_alloc = a->user_alloc; 1689d3042a70SBarry Smith a->v = (PetscScalar *)array; 1690637a0070SStefano Zampini a->user_alloc = PETSC_TRUE; 169147d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) || defined(PETSC_HAVE_HIP) 1692c70f7ee4SJunchao Zhang A->offloadmask = PETSC_OFFLOAD_CPU; 1693ca15aa20SStefano Zampini #endif 1694d3042a70SBarry Smith PetscFunctionReturn(0); 1695d3042a70SBarry Smith } 1696d3042a70SBarry Smith 1697d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatDenseResetArray_SeqDense(Mat A) 1698d71ae5a4SJacob Faibussowitsch { 1699d3042a70SBarry Smith Mat_SeqDense *a = (Mat_SeqDense *)A->data; 1700d3042a70SBarry Smith 1701d3042a70SBarry Smith PetscFunctionBegin; 170228b400f6SJacob Faibussowitsch PetscCheck(!a->vecinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreColumnVec() first"); 170328b400f6SJacob Faibussowitsch PetscCheck(!a->matinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreSubMatrix() first"); 1704d3042a70SBarry Smith a->v = a->unplacedarray; 1705d3042a70SBarry Smith a->user_alloc = a->unplaced_user_alloc; 1706d3042a70SBarry Smith a->unplacedarray = NULL; 170747d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) || defined(PETSC_HAVE_HIP) 1708c70f7ee4SJunchao Zhang A->offloadmask = PETSC_OFFLOAD_CPU; 1709ca15aa20SStefano Zampini #endif 1710d3042a70SBarry Smith PetscFunctionReturn(0); 1711d3042a70SBarry Smith } 1712d3042a70SBarry Smith 1713d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatDenseReplaceArray_SeqDense(Mat A, const PetscScalar *array) 1714d71ae5a4SJacob Faibussowitsch { 1715d5ea218eSStefano Zampini Mat_SeqDense *a = (Mat_SeqDense *)A->data; 1716d5ea218eSStefano Zampini 1717d5ea218eSStefano Zampini PetscFunctionBegin; 171828b400f6SJacob Faibussowitsch PetscCheck(!a->vecinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreColumnVec() first"); 171928b400f6SJacob Faibussowitsch PetscCheck(!a->matinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreSubMatrix() first"); 17209566063dSJacob Faibussowitsch if (!a->user_alloc) PetscCall(PetscFree(a->v)); 1721d5ea218eSStefano Zampini a->v = (PetscScalar *)array; 1722d5ea218eSStefano Zampini a->user_alloc = PETSC_FALSE; 172347d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) || defined(PETSC_HAVE_HIP) 1724d5ea218eSStefano Zampini A->offloadmask = PETSC_OFFLOAD_CPU; 1725d5ea218eSStefano Zampini #endif 1726d5ea218eSStefano Zampini PetscFunctionReturn(0); 1727d5ea218eSStefano Zampini } 1728d5ea218eSStefano Zampini 1729d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDestroy_SeqDense(Mat mat) 1730d71ae5a4SJacob Faibussowitsch { 1731ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense *)mat->data; 173290f02eecSBarry Smith 17333a40ed3dSBarry Smith PetscFunctionBegin; 1734aa482453SBarry Smith #if defined(PETSC_USE_LOG) 1735c0aa6a63SJacob Faibussowitsch PetscLogObjectState((PetscObject)mat, "Rows %" PetscInt_FMT " Cols %" PetscInt_FMT, mat->rmap->n, mat->cmap->n); 1736a5a9c739SBarry Smith #endif 17379566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(l->qrrhs))); 17389566063dSJacob Faibussowitsch PetscCall(PetscFree(l->tau)); 17399566063dSJacob Faibussowitsch PetscCall(PetscFree(l->pivots)); 17409566063dSJacob Faibussowitsch PetscCall(PetscFree(l->fwork)); 17419566063dSJacob Faibussowitsch PetscCall(MatDestroy(&l->ptapwork)); 17429566063dSJacob Faibussowitsch if (!l->user_alloc) PetscCall(PetscFree(l->v)); 17439566063dSJacob Faibussowitsch if (!l->unplaced_user_alloc) PetscCall(PetscFree(l->unplacedarray)); 174428b400f6SJacob Faibussowitsch PetscCheck(!l->vecinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreColumnVec() first"); 174528b400f6SJacob Faibussowitsch PetscCheck(!l->matinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreSubMatrix() first"); 17469566063dSJacob Faibussowitsch PetscCall(VecDestroy(&l->cvec)); 17479566063dSJacob Faibussowitsch PetscCall(MatDestroy(&l->cmat)); 17489566063dSJacob Faibussowitsch PetscCall(PetscFree(mat->data)); 1749dbd8c25aSHong Zhang 17509566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)mat, NULL)); 17519566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatQRFactor_C", NULL)); 17522e956fe4SStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatQRFactorSymbolic_C", NULL)); 17532e956fe4SStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatQRFactorNumeric_C", NULL)); 17549566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseGetLDA_C", NULL)); 17559566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseSetLDA_C", NULL)); 17569566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseGetArray_C", NULL)); 17579566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseRestoreArray_C", NULL)); 17589566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDensePlaceArray_C", NULL)); 17599566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseResetArray_C", NULL)); 17609566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseReplaceArray_C", NULL)); 17619566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseGetArrayRead_C", NULL)); 17629566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseRestoreArrayRead_C", NULL)); 17639566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseGetArrayWrite_C", NULL)); 17649566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseRestoreArrayWrite_C", NULL)); 17659566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatConvert_seqdense_seqaij_C", NULL)); 17668baccfbdSHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 17679566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatConvert_seqdense_elemental_C", NULL)); 17688baccfbdSHong Zhang #endif 1769d24d4204SJose E. Roman #if defined(PETSC_HAVE_SCALAPACK) 17709566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatConvert_seqdense_scalapack_C", NULL)); 1771d24d4204SJose E. Roman #endif 17722bf066beSStefano Zampini #if defined(PETSC_HAVE_CUDA) 17739566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatConvert_seqdense_seqdensecuda_C", NULL)); 17749566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatProductSetFromOptions_seqdensecuda_seqdensecuda_C", NULL)); 17759566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatProductSetFromOptions_seqdensecuda_seqdense_C", NULL)); 17762e956fe4SStefano Zampini PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatProductSetFromOptions_seqdense_seqdensecuda_C", NULL)); 17772bf066beSStefano Zampini #endif 177847d993e7Ssuyashtn #if defined(PETSC_HAVE_HIP) 177947d993e7Ssuyashtn PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatConvert_seqdense_seqdensehip_C", NULL)); 178047d993e7Ssuyashtn PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatProductSetFromOptions_seqdensehip_seqdensehip_C", NULL)); 178147d993e7Ssuyashtn PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatProductSetFromOptions_seqdensehip_seqdense_C", NULL)); 178247d993e7Ssuyashtn PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatProductSetFromOptions_seqdense_seqdensehip_C", NULL)); 178347d993e7Ssuyashtn #endif 17849566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatSeqDenseSetPreallocation_C", NULL)); 17859566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatProductSetFromOptions_seqaij_seqdense_C", NULL)); 17869566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatProductSetFromOptions_seqdense_seqdense_C", NULL)); 17879566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatProductSetFromOptions_seqbaij_seqdense_C", NULL)); 17889566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatProductSetFromOptions_seqsbaij_seqdense_C", NULL)); 178952c5f739Sprj- 17909566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseGetColumn_C", NULL)); 17919566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseRestoreColumn_C", NULL)); 17929566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseGetColumnVec_C", NULL)); 17939566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseRestoreColumnVec_C", NULL)); 17949566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseGetColumnVecRead_C", NULL)); 17959566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseRestoreColumnVecRead_C", NULL)); 17969566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseGetColumnVecWrite_C", NULL)); 17979566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseRestoreColumnVecWrite_C", NULL)); 17989566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseGetSubMatrix_C", NULL)); 17999566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatDenseRestoreSubMatrix_C", NULL)); 18003a40ed3dSBarry Smith PetscFunctionReturn(0); 1801289bc588SBarry Smith } 1802289bc588SBarry Smith 1803d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatTranspose_SeqDense(Mat A, MatReuse reuse, Mat *matout) 1804d71ae5a4SJacob Faibussowitsch { 1805c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 18066536e3caSStefano Zampini PetscInt k, j, m = A->rmap->n, M = mat->lda, n = A->cmap->n; 180787828ca2SBarry Smith PetscScalar *v, tmp; 180848b35521SBarry Smith 18093a40ed3dSBarry Smith PetscFunctionBegin; 18107fb60732SBarry Smith if (reuse == MAT_REUSE_MATRIX) PetscCall(MatTransposeCheckNonzeroState_Private(A, *matout)); 18116536e3caSStefano Zampini if (reuse == MAT_INPLACE_MATRIX) { 18126536e3caSStefano Zampini if (m == n) { /* in place transpose */ 18139566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, &v)); 1814d3e5ee88SLois Curfman McInnes for (j = 0; j < m; j++) { 1815289bc588SBarry Smith for (k = 0; k < j; k++) { 18161b807ce4Svictorle tmp = v[j + k * M]; 18171b807ce4Svictorle v[j + k * M] = v[k + j * M]; 18181b807ce4Svictorle v[k + j * M] = tmp; 1819289bc588SBarry Smith } 1820289bc588SBarry Smith } 18219566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, &v)); 18226536e3caSStefano Zampini } else { /* reuse memory, temporary allocates new memory */ 18236536e3caSStefano Zampini PetscScalar *v2; 18246536e3caSStefano Zampini PetscLayout tmplayout; 18256536e3caSStefano Zampini 18269566063dSJacob Faibussowitsch PetscCall(PetscMalloc1((size_t)m * n, &v2)); 18279566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, &v)); 18286536e3caSStefano Zampini for (j = 0; j < n; j++) { 18296536e3caSStefano Zampini for (k = 0; k < m; k++) v2[j + (size_t)k * n] = v[k + (size_t)j * M]; 18306536e3caSStefano Zampini } 18319566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(v, v2, (size_t)m * n)); 18329566063dSJacob Faibussowitsch PetscCall(PetscFree(v2)); 18339566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, &v)); 18346536e3caSStefano Zampini /* cleanup size dependent quantities */ 18359566063dSJacob Faibussowitsch PetscCall(VecDestroy(&mat->cvec)); 18369566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mat->cmat)); 18379566063dSJacob Faibussowitsch PetscCall(PetscFree(mat->pivots)); 18389566063dSJacob Faibussowitsch PetscCall(PetscFree(mat->fwork)); 18399566063dSJacob Faibussowitsch PetscCall(MatDestroy(&mat->ptapwork)); 18406536e3caSStefano Zampini /* swap row/col layouts */ 18416536e3caSStefano Zampini mat->lda = n; 18426536e3caSStefano Zampini tmplayout = A->rmap; 18436536e3caSStefano Zampini A->rmap = A->cmap; 18446536e3caSStefano Zampini A->cmap = tmplayout; 18456536e3caSStefano Zampini } 18463a40ed3dSBarry Smith } else { /* out-of-place transpose */ 1847d3e5ee88SLois Curfman McInnes Mat tmat; 1848ec8511deSBarry Smith Mat_SeqDense *tmatd; 184987828ca2SBarry Smith PetscScalar *v2; 1850af36a384SStefano Zampini PetscInt M2; 1851ea709b57SSatish Balay 18526536e3caSStefano Zampini if (reuse == MAT_INITIAL_MATRIX) { 18539566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &tmat)); 18549566063dSJacob Faibussowitsch PetscCall(MatSetSizes(tmat, A->cmap->n, A->rmap->n, A->cmap->n, A->rmap->n)); 18559566063dSJacob Faibussowitsch PetscCall(MatSetType(tmat, ((PetscObject)A)->type_name)); 18569566063dSJacob Faibussowitsch PetscCall(MatSeqDenseSetPreallocation(tmat, NULL)); 1857ca15aa20SStefano Zampini } else tmat = *matout; 1858ca15aa20SStefano Zampini 18599566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, (const PetscScalar **)&v)); 18609566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(tmat, &v2)); 1861ec8511deSBarry Smith tmatd = (Mat_SeqDense *)tmat->data; 1862ca15aa20SStefano Zampini M2 = tmatd->lda; 1863d3e5ee88SLois Curfman McInnes for (j = 0; j < n; j++) { 1864af36a384SStefano Zampini for (k = 0; k < m; k++) v2[j + k * M2] = v[k + j * M]; 1865d3e5ee88SLois Curfman McInnes } 18669566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(tmat, &v2)); 18679566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, (const PetscScalar **)&v)); 18689566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(tmat, MAT_FINAL_ASSEMBLY)); 18699566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(tmat, MAT_FINAL_ASSEMBLY)); 18706536e3caSStefano Zampini *matout = tmat; 187148b35521SBarry Smith } 18723a40ed3dSBarry Smith PetscFunctionReturn(0); 1873289bc588SBarry Smith } 1874289bc588SBarry Smith 1875d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatEqual_SeqDense(Mat A1, Mat A2, PetscBool *flg) 1876d71ae5a4SJacob Faibussowitsch { 1877c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat1 = (Mat_SeqDense *)A1->data; 1878c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat2 = (Mat_SeqDense *)A2->data; 1879ca15aa20SStefano Zampini PetscInt i; 1880ca15aa20SStefano Zampini const PetscScalar *v1, *v2; 18819ea5d5aeSSatish Balay 18823a40ed3dSBarry Smith PetscFunctionBegin; 18839371c9d4SSatish Balay if (A1->rmap->n != A2->rmap->n) { 18849371c9d4SSatish Balay *flg = PETSC_FALSE; 18859371c9d4SSatish Balay PetscFunctionReturn(0); 18869371c9d4SSatish Balay } 18879371c9d4SSatish Balay if (A1->cmap->n != A2->cmap->n) { 18889371c9d4SSatish Balay *flg = PETSC_FALSE; 18899371c9d4SSatish Balay PetscFunctionReturn(0); 18909371c9d4SSatish Balay } 18919566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A1, &v1)); 18929566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A2, &v2)); 1893ca15aa20SStefano Zampini for (i = 0; i < A1->cmap->n; i++) { 18949566063dSJacob Faibussowitsch PetscCall(PetscArraycmp(v1, v2, A1->rmap->n, flg)); 1895ca15aa20SStefano Zampini if (*flg == PETSC_FALSE) PetscFunctionReturn(0); 1896ca15aa20SStefano Zampini v1 += mat1->lda; 1897ca15aa20SStefano Zampini v2 += mat2->lda; 18981b807ce4Svictorle } 18999566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A1, &v1)); 19009566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A2, &v2)); 190177c4ece6SBarry Smith *flg = PETSC_TRUE; 19023a40ed3dSBarry Smith PetscFunctionReturn(0); 1903289bc588SBarry Smith } 1904289bc588SBarry Smith 1905d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetDiagonal_SeqDense(Mat A, Vec v) 1906d71ae5a4SJacob Faibussowitsch { 1907c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 190813f74950SBarry Smith PetscInt i, n, len; 1909ca15aa20SStefano Zampini PetscScalar *x; 1910ca15aa20SStefano Zampini const PetscScalar *vv; 191144cd7ae7SLois Curfman McInnes 19123a40ed3dSBarry Smith PetscFunctionBegin; 19139566063dSJacob Faibussowitsch PetscCall(VecGetSize(v, &n)); 19149566063dSJacob Faibussowitsch PetscCall(VecGetArray(v, &x)); 1915d0f46423SBarry Smith len = PetscMin(A->rmap->n, A->cmap->n); 19169566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &vv)); 191708401ef6SPierre Jolivet PetscCheck(n == A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Nonconforming mat and vec"); 1918ad540459SPierre Jolivet for (i = 0; i < len; i++) x[i] = vv[i * mat->lda + i]; 19199566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &vv)); 19209566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(v, &x)); 19213a40ed3dSBarry Smith PetscFunctionReturn(0); 1922289bc588SBarry Smith } 1923289bc588SBarry Smith 1924d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatDiagonalScale_SeqDense(Mat A, Vec ll, Vec rr) 1925d71ae5a4SJacob Faibussowitsch { 1926c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 1927f1ceaac6SMatthew G. Knepley const PetscScalar *l, *r; 1928ca15aa20SStefano Zampini PetscScalar x, *v, *vv; 1929d0f46423SBarry Smith PetscInt i, j, m = A->rmap->n, n = A->cmap->n; 193055659b69SBarry Smith 19313a40ed3dSBarry Smith PetscFunctionBegin; 19329566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, &vv)); 193328988994SBarry Smith if (ll) { 19349566063dSJacob Faibussowitsch PetscCall(VecGetSize(ll, &m)); 19359566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(ll, &l)); 193608401ef6SPierre Jolivet PetscCheck(m == A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Left scaling vec wrong size"); 1937da3a660dSBarry Smith for (i = 0; i < m; i++) { 1938da3a660dSBarry Smith x = l[i]; 1939ca15aa20SStefano Zampini v = vv + i; 19409371c9d4SSatish Balay for (j = 0; j < n; j++) { 19419371c9d4SSatish Balay (*v) *= x; 19429371c9d4SSatish Balay v += mat->lda; 19439371c9d4SSatish Balay } 1944da3a660dSBarry Smith } 19459566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(ll, &l)); 19469566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(1.0 * n * m)); 1947da3a660dSBarry Smith } 194828988994SBarry Smith if (rr) { 19499566063dSJacob Faibussowitsch PetscCall(VecGetSize(rr, &n)); 19509566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(rr, &r)); 195108401ef6SPierre Jolivet PetscCheck(n == A->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Right scaling vec wrong size"); 1952da3a660dSBarry Smith for (i = 0; i < n; i++) { 1953da3a660dSBarry Smith x = r[i]; 1954ca15aa20SStefano Zampini v = vv + i * mat->lda; 19552205254eSKarl Rupp for (j = 0; j < m; j++) (*v++) *= x; 1956da3a660dSBarry Smith } 19579566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(rr, &r)); 19589566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(1.0 * n * m)); 1959da3a660dSBarry Smith } 19609566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, &vv)); 19613a40ed3dSBarry Smith PetscFunctionReturn(0); 1962289bc588SBarry Smith } 1963289bc588SBarry Smith 1964d71ae5a4SJacob Faibussowitsch PetscErrorCode MatNorm_SeqDense(Mat A, NormType type, PetscReal *nrm) 1965d71ae5a4SJacob Faibussowitsch { 1966c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 1967ca15aa20SStefano Zampini PetscScalar *v, *vv; 1968329f5518SBarry Smith PetscReal sum = 0.0; 196975f6d85dSStefano Zampini PetscInt lda, m = A->rmap->n, i, j; 197055659b69SBarry Smith 19713a40ed3dSBarry Smith PetscFunctionBegin; 19729566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, (const PetscScalar **)&vv)); 19739566063dSJacob Faibussowitsch PetscCall(MatDenseGetLDA(A, &lda)); 1974ca15aa20SStefano Zampini v = vv; 1975289bc588SBarry Smith if (type == NORM_FROBENIUS) { 1976a5ce6ee0Svictorle if (lda > m) { 1977d0f46423SBarry Smith for (j = 0; j < A->cmap->n; j++) { 1978ca15aa20SStefano Zampini v = vv + j * lda; 1979a5ce6ee0Svictorle for (i = 0; i < m; i++) { 19809371c9d4SSatish Balay sum += PetscRealPart(PetscConj(*v) * (*v)); 19819371c9d4SSatish Balay v++; 1982a5ce6ee0Svictorle } 1983a5ce6ee0Svictorle } 1984a5ce6ee0Svictorle } else { 1985570b7f6dSBarry Smith #if defined(PETSC_USE_REAL___FP16) 1986570b7f6dSBarry Smith PetscBLASInt one = 1, cnt = A->cmap->n * A->rmap->n; 1987792fecdfSBarry Smith PetscCallBLAS("BLASnrm2", *nrm = BLASnrm2_(&cnt, v, &one)); 1988570b7f6dSBarry Smith } 1989570b7f6dSBarry Smith #else 1990d0f46423SBarry Smith for (i = 0; i < A->cmap->n * A->rmap->n; i++) { 19919371c9d4SSatish Balay sum += PetscRealPart(PetscConj(*v) * (*v)); 19929371c9d4SSatish Balay v++; 1993289bc588SBarry Smith } 1994a5ce6ee0Svictorle } 19958f1a2a5eSBarry Smith *nrm = PetscSqrtReal(sum); 1996570b7f6dSBarry Smith #endif 19979566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(2.0 * A->cmap->n * A->rmap->n)); 19983a40ed3dSBarry Smith } else if (type == NORM_1) { 1999064f8208SBarry Smith *nrm = 0.0; 2000d0f46423SBarry Smith for (j = 0; j < A->cmap->n; j++) { 2001ca15aa20SStefano Zampini v = vv + j * mat->lda; 2002289bc588SBarry Smith sum = 0.0; 2003d0f46423SBarry Smith for (i = 0; i < A->rmap->n; i++) { 20049371c9d4SSatish Balay sum += PetscAbsScalar(*v); 20059371c9d4SSatish Balay v++; 2006289bc588SBarry Smith } 2007064f8208SBarry Smith if (sum > *nrm) *nrm = sum; 2008289bc588SBarry Smith } 20099566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(1.0 * A->cmap->n * A->rmap->n)); 20103a40ed3dSBarry Smith } else if (type == NORM_INFINITY) { 2011064f8208SBarry Smith *nrm = 0.0; 2012d0f46423SBarry Smith for (j = 0; j < A->rmap->n; j++) { 2013ca15aa20SStefano Zampini v = vv + j; 2014289bc588SBarry Smith sum = 0.0; 2015d0f46423SBarry Smith for (i = 0; i < A->cmap->n; i++) { 20169371c9d4SSatish Balay sum += PetscAbsScalar(*v); 20179371c9d4SSatish Balay v += mat->lda; 2018289bc588SBarry Smith } 2019064f8208SBarry Smith if (sum > *nrm) *nrm = sum; 2020289bc588SBarry Smith } 20219566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(1.0 * A->cmap->n * A->rmap->n)); 2022e7e72b3dSBarry Smith } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "No two norm"); 20239566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, (const PetscScalar **)&vv)); 20243a40ed3dSBarry Smith PetscFunctionReturn(0); 2025289bc588SBarry Smith } 2026289bc588SBarry Smith 2027d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatSetOption_SeqDense(Mat A, MatOption op, PetscBool flg) 2028d71ae5a4SJacob Faibussowitsch { 2029c0bbcb79SLois Curfman McInnes Mat_SeqDense *aij = (Mat_SeqDense *)A->data; 203067e560aaSBarry Smith 20313a40ed3dSBarry Smith PetscFunctionBegin; 2032b5a2b587SKris Buschelman switch (op) { 2033d71ae5a4SJacob Faibussowitsch case MAT_ROW_ORIENTED: 2034d71ae5a4SJacob Faibussowitsch aij->roworiented = flg; 2035d71ae5a4SJacob Faibussowitsch break; 2036512a5fc5SBarry Smith case MAT_NEW_NONZERO_LOCATIONS: 2037b5a2b587SKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 20383971808eSMatthew Knepley case MAT_NEW_NONZERO_ALLOCATION_ERR: 20398c78258cSHong Zhang case MAT_FORCE_DIAGONAL_ENTRIES: 204013fa8e87SLisandro Dalcin case MAT_KEEP_NONZERO_PATTERN: 2041b5a2b587SKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 2042b5a2b587SKris Buschelman case MAT_USE_HASH_TABLE: 20430f8fb01aSBarry Smith case MAT_IGNORE_ZERO_ENTRIES: 20445021d80fSJed Brown case MAT_IGNORE_LOWER_TRIANGULAR: 2045d71ae5a4SJacob Faibussowitsch case MAT_SORTED_FULL: 2046d71ae5a4SJacob Faibussowitsch PetscCall(PetscInfo(A, "Option %s ignored\n", MatOptions[op])); 2047d71ae5a4SJacob Faibussowitsch break; 20485021d80fSJed Brown case MAT_SPD: 204977e54ba9SKris Buschelman case MAT_SYMMETRIC: 205077e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 20519a4540c5SBarry Smith case MAT_HERMITIAN: 20529a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 2053b94d7dedSBarry Smith case MAT_STRUCTURAL_SYMMETRY_ETERNAL: 2054d71ae5a4SJacob Faibussowitsch case MAT_SPD_ETERNAL: 2055d71ae5a4SJacob Faibussowitsch break; 2056d71ae5a4SJacob Faibussowitsch default: 2057d71ae5a4SJacob Faibussowitsch SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "unknown option %s", MatOptions[op]); 20583a40ed3dSBarry Smith } 20593a40ed3dSBarry Smith PetscFunctionReturn(0); 2060289bc588SBarry Smith } 2061289bc588SBarry Smith 2062d71ae5a4SJacob Faibussowitsch PetscErrorCode MatZeroEntries_SeqDense(Mat A) 2063d71ae5a4SJacob Faibussowitsch { 2064ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense *)A->data; 20653d8925e7SStefano Zampini PetscInt lda = l->lda, m = A->rmap->n, n = A->cmap->n, j; 2066ca15aa20SStefano Zampini PetscScalar *v; 20673a40ed3dSBarry Smith 20683a40ed3dSBarry Smith PetscFunctionBegin; 20699566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayWrite(A, &v)); 2070a5ce6ee0Svictorle if (lda > m) { 207148a46eb9SPierre Jolivet for (j = 0; j < n; j++) PetscCall(PetscArrayzero(v + j * lda, m)); 2072a5ce6ee0Svictorle } else { 20739566063dSJacob Faibussowitsch PetscCall(PetscArrayzero(v, PetscInt64Mult(m, n))); 2074a5ce6ee0Svictorle } 20759566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayWrite(A, &v)); 20763a40ed3dSBarry Smith PetscFunctionReturn(0); 20776f0a148fSBarry Smith } 20786f0a148fSBarry Smith 2079d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatZeroRows_SeqDense(Mat A, PetscInt N, const PetscInt rows[], PetscScalar diag, Vec x, Vec b) 2080d71ae5a4SJacob Faibussowitsch { 2081ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense *)A->data; 2082b9679d65SBarry Smith PetscInt m = l->lda, n = A->cmap->n, i, j; 2083ca15aa20SStefano Zampini PetscScalar *slot, *bb, *v; 208497b48c8fSBarry Smith const PetscScalar *xx; 208555659b69SBarry Smith 20863a40ed3dSBarry Smith PetscFunctionBegin; 208776bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 2088b9679d65SBarry Smith for (i = 0; i < N; i++) { 208908401ef6SPierre Jolivet PetscCheck(rows[i] >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Negative row requested to be zeroed"); 209008401ef6SPierre Jolivet PetscCheck(rows[i] < A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Row %" PetscInt_FMT " requested to be zeroed greater than or equal number of rows %" PetscInt_FMT, rows[i], A->rmap->n); 2091b9679d65SBarry Smith } 209276bd3646SJed Brown } 2093ca15aa20SStefano Zampini if (!N) PetscFunctionReturn(0); 2094b9679d65SBarry Smith 209597b48c8fSBarry Smith /* fix right hand side if needed */ 209697b48c8fSBarry Smith if (x && b) { 20979566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(x, &xx)); 20989566063dSJacob Faibussowitsch PetscCall(VecGetArray(b, &bb)); 20992205254eSKarl Rupp for (i = 0; i < N; i++) bb[rows[i]] = diag * xx[rows[i]]; 21009566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(x, &xx)); 21019566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(b, &bb)); 210297b48c8fSBarry Smith } 210397b48c8fSBarry Smith 21049566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, &v)); 21056f0a148fSBarry Smith for (i = 0; i < N; i++) { 2106ca15aa20SStefano Zampini slot = v + rows[i]; 21079371c9d4SSatish Balay for (j = 0; j < n; j++) { 21089371c9d4SSatish Balay *slot = 0.0; 21099371c9d4SSatish Balay slot += m; 21109371c9d4SSatish Balay } 21116f0a148fSBarry Smith } 2112f4df32b1SMatthew Knepley if (diag != 0.0) { 211308401ef6SPierre Jolivet PetscCheck(A->rmap->n == A->cmap->n, PETSC_COMM_SELF, PETSC_ERR_SUP, "Only coded for square matrices"); 21146f0a148fSBarry Smith for (i = 0; i < N; i++) { 2115ca15aa20SStefano Zampini slot = v + (m + 1) * rows[i]; 2116f4df32b1SMatthew Knepley *slot = diag; 21176f0a148fSBarry Smith } 21186f0a148fSBarry Smith } 21199566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, &v)); 21203a40ed3dSBarry Smith PetscFunctionReturn(0); 21216f0a148fSBarry Smith } 2122557bce09SLois Curfman McInnes 2123d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatDenseGetLDA_SeqDense(Mat A, PetscInt *lda) 2124d71ae5a4SJacob Faibussowitsch { 212549a6ff4bSBarry Smith Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 212649a6ff4bSBarry Smith 212749a6ff4bSBarry Smith PetscFunctionBegin; 212849a6ff4bSBarry Smith *lda = mat->lda; 212949a6ff4bSBarry Smith PetscFunctionReturn(0); 213049a6ff4bSBarry Smith } 213149a6ff4bSBarry Smith 2132d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetArray_SeqDense(Mat A, PetscScalar **array) 2133d71ae5a4SJacob Faibussowitsch { 2134c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 21353a40ed3dSBarry Smith 21363a40ed3dSBarry Smith PetscFunctionBegin; 213728b400f6SJacob Faibussowitsch PetscCheck(!mat->matinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreSubMatrix() first"); 213864e87e97SBarry Smith *array = mat->v; 21393a40ed3dSBarry Smith PetscFunctionReturn(0); 214064e87e97SBarry Smith } 21410754003eSLois Curfman McInnes 2142d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreArray_SeqDense(Mat A, PetscScalar **array) 2143d71ae5a4SJacob Faibussowitsch { 21443a40ed3dSBarry Smith PetscFunctionBegin; 214575f6d85dSStefano Zampini if (array) *array = NULL; 21463a40ed3dSBarry Smith PetscFunctionReturn(0); 2147ff14e315SSatish Balay } 21480754003eSLois Curfman McInnes 21490f74d2c1SSatish Balay /*@ 215011a5261eSBarry Smith MatDenseGetLDA - gets the leading dimension of the array returned from `MatDenseGetArray()` 215149a6ff4bSBarry Smith 2152ad16ce7aSStefano Zampini Not collective 215349a6ff4bSBarry Smith 215449a6ff4bSBarry Smith Input Parameter: 215511a5261eSBarry Smith . mat - a `MATDENSE` or `MATDENSECUDA` matrix 215649a6ff4bSBarry Smith 215749a6ff4bSBarry Smith Output Parameter: 215849a6ff4bSBarry Smith . lda - the leading dimension 215949a6ff4bSBarry Smith 216049a6ff4bSBarry Smith Level: intermediate 216149a6ff4bSBarry Smith 216211a5261eSBarry Smith .seealso: `MATDENSE`, `MATDENSECUDA`, `MatDenseGetArray()`, `MatDenseRestoreArray()`, `MatDenseGetArrayRead()`, `MatDenseRestoreArrayRead()`, `MatDenseSetLDA()` 216349a6ff4bSBarry Smith @*/ 2164d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetLDA(Mat A, PetscInt *lda) 2165d71ae5a4SJacob Faibussowitsch { 216649a6ff4bSBarry Smith PetscFunctionBegin; 2167d5ea218eSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2168dadcf809SJacob Faibussowitsch PetscValidIntPointer(lda, 2); 216975f6d85dSStefano Zampini MatCheckPreallocated(A, 1); 2170cac4c232SBarry Smith PetscUseMethod(A, "MatDenseGetLDA_C", (Mat, PetscInt *), (A, lda)); 217149a6ff4bSBarry Smith PetscFunctionReturn(0); 217249a6ff4bSBarry Smith } 217349a6ff4bSBarry Smith 21740f74d2c1SSatish Balay /*@ 217511a5261eSBarry Smith MatDenseSetLDA - Sets the leading dimension of the array used by the `MATDENSE` matrix 2176ad16ce7aSStefano Zampini 2177ad16ce7aSStefano Zampini Not collective 2178ad16ce7aSStefano Zampini 2179d8d19677SJose E. Roman Input Parameters: 218011a5261eSBarry Smith + mat - a `MATDENSE` or `MATDENSECUDA` matrix 2181ad16ce7aSStefano Zampini - lda - the leading dimension 2182ad16ce7aSStefano Zampini 2183ad16ce7aSStefano Zampini Level: intermediate 2184ad16ce7aSStefano Zampini 218511a5261eSBarry Smith .seealso: `MATDENSE`, `MATDENSECUDA`, `MatDenseGetArray()`, `MatDenseRestoreArray()`, `MatDenseGetArrayRead()`, `MatDenseRestoreArrayRead()`, `MatDenseGetLDA()` 2186ad16ce7aSStefano Zampini @*/ 2187d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseSetLDA(Mat A, PetscInt lda) 2188d71ae5a4SJacob Faibussowitsch { 2189ad16ce7aSStefano Zampini PetscFunctionBegin; 2190ad16ce7aSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2191cac4c232SBarry Smith PetscTryMethod(A, "MatDenseSetLDA_C", (Mat, PetscInt), (A, lda)); 2192ad16ce7aSStefano Zampini PetscFunctionReturn(0); 2193ad16ce7aSStefano Zampini } 2194ad16ce7aSStefano Zampini 2195ad16ce7aSStefano Zampini /*@C 219611a5261eSBarry Smith MatDenseGetArray - gives read-write access to the array where the data for a `MATDENSE` matrix is stored 219773a71a0fSBarry Smith 2198c3339decSBarry Smith Logically Collective 219973a71a0fSBarry Smith 220073a71a0fSBarry Smith Input Parameter: 22016947451fSStefano Zampini . mat - a dense matrix 220273a71a0fSBarry Smith 220373a71a0fSBarry Smith Output Parameter: 220473a71a0fSBarry Smith . array - pointer to the data 220573a71a0fSBarry Smith 220673a71a0fSBarry Smith Level: intermediate 220773a71a0fSBarry Smith 22080ab4885dSBarry Smith Fortran Note: 22090ab4885dSBarry Smith `MatDenseGetArray()` Fortran binding is deprecated (since PETSc 3.19), use `MatDenseGetArrayF90()` 22100ab4885dSBarry Smith 221111a5261eSBarry Smith .seealso: `MATDENSE`, `MatDenseRestoreArray()`, `MatDenseGetArrayRead()`, `MatDenseRestoreArrayRead()`, `MatDenseGetArrayWrite()`, `MatDenseRestoreArrayWrite()` 221273a71a0fSBarry Smith @*/ 2213d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetArray(Mat A, PetscScalar **array) 2214d71ae5a4SJacob Faibussowitsch { 221573a71a0fSBarry Smith PetscFunctionBegin; 2216d5ea218eSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2217d5ea218eSStefano Zampini PetscValidPointer(array, 2); 2218cac4c232SBarry Smith PetscUseMethod(A, "MatDenseGetArray_C", (Mat, PetscScalar **), (A, array)); 221973a71a0fSBarry Smith PetscFunctionReturn(0); 222073a71a0fSBarry Smith } 222173a71a0fSBarry Smith 2222dec5eb66SMatthew G Knepley /*@C 222311a5261eSBarry Smith MatDenseRestoreArray - returns access to the array where the data for a `MATDENSE` matrix is stored obtained by `MatDenseGetArray()` 222473a71a0fSBarry Smith 2225c3339decSBarry Smith Logically Collective 22268572280aSBarry Smith 22278572280aSBarry Smith Input Parameters: 22286947451fSStefano Zampini + mat - a dense matrix 2229742765d3SMatthew Knepley - array - pointer to the data (may be NULL) 22308572280aSBarry Smith 22318572280aSBarry Smith Level: intermediate 22328572280aSBarry Smith 22330ab4885dSBarry Smith Fortran Note: 22340ab4885dSBarry Smith `MatDenseRestoreArray()` Fortran binding is deprecated (since PETSc 3.19), use `MatDenseRestoreArrayF90()` 22350ab4885dSBarry Smith 223611a5261eSBarry Smith .seealso: `MATDENSE`, `MatDenseGetArray()`, `MatDenseGetArrayRead()`, `MatDenseRestoreArrayRead()`, `MatDenseGetArrayWrite()`, `MatDenseRestoreArrayWrite()` 22378572280aSBarry Smith @*/ 2238d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreArray(Mat A, PetscScalar **array) 2239d71ae5a4SJacob Faibussowitsch { 22408572280aSBarry Smith PetscFunctionBegin; 2241d5ea218eSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2242d5ea218eSStefano Zampini PetscValidPointer(array, 2); 2243cac4c232SBarry Smith PetscUseMethod(A, "MatDenseRestoreArray_C", (Mat, PetscScalar **), (A, array)); 22449566063dSJacob Faibussowitsch PetscCall(PetscObjectStateIncrease((PetscObject)A)); 224547d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) || defined(PETSC_HAVE_HIP) 2246637a0070SStefano Zampini A->offloadmask = PETSC_OFFLOAD_CPU; 2247637a0070SStefano Zampini #endif 22488572280aSBarry Smith PetscFunctionReturn(0); 22498572280aSBarry Smith } 22508572280aSBarry Smith 22518572280aSBarry Smith /*@C 225211a5261eSBarry Smith MatDenseGetArrayRead - gives read-only access to the array where the data for a `MATDENSE` matrix is stored 22538572280aSBarry Smith 22540ab4885dSBarry Smith Not Collective; No Fortran Support 22558572280aSBarry Smith 22568572280aSBarry Smith Input Parameter: 22576947451fSStefano Zampini . mat - a dense matrix 22588572280aSBarry Smith 22598572280aSBarry Smith Output Parameter: 22608572280aSBarry Smith . array - pointer to the data 22618572280aSBarry Smith 22628572280aSBarry Smith Level: intermediate 22638572280aSBarry Smith 226411a5261eSBarry Smith .seealso: `MATDENSE`, `MatDenseRestoreArrayRead()`, `MatDenseGetArray()`, `MatDenseRestoreArray()`, `MatDenseGetArrayWrite()`, `MatDenseRestoreArrayWrite()` 22658572280aSBarry Smith @*/ 2266d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetArrayRead(Mat A, const PetscScalar **array) 2267d71ae5a4SJacob Faibussowitsch { 22688572280aSBarry Smith PetscFunctionBegin; 2269d5ea218eSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2270d5ea218eSStefano Zampini PetscValidPointer(array, 2); 2271cac4c232SBarry Smith PetscUseMethod(A, "MatDenseGetArrayRead_C", (Mat, const PetscScalar **), (A, array)); 22728572280aSBarry Smith PetscFunctionReturn(0); 22738572280aSBarry Smith } 22748572280aSBarry Smith 22758572280aSBarry Smith /*@C 227611a5261eSBarry Smith MatDenseRestoreArrayRead - returns access to the array where the data for a `MATDENSE` matrix is stored obtained by `MatDenseGetArrayRead()` 22778572280aSBarry Smith 22780ab4885dSBarry Smith Not Collective; No Fortran Support 227973a71a0fSBarry Smith 228073a71a0fSBarry Smith Input Parameters: 22816947451fSStefano Zampini + mat - a dense matrix 2282742765d3SMatthew Knepley - array - pointer to the data (may be NULL) 228373a71a0fSBarry Smith 228473a71a0fSBarry Smith Level: intermediate 228573a71a0fSBarry Smith 228611a5261eSBarry Smith .seealso: `MATDENSE`, `MatDenseGetArrayRead()`, `MatDenseGetArray()`, `MatDenseRestoreArray()`, `MatDenseGetArrayWrite()`, `MatDenseRestoreArrayWrite()` 228773a71a0fSBarry Smith @*/ 2288d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreArrayRead(Mat A, const PetscScalar **array) 2289d71ae5a4SJacob Faibussowitsch { 229073a71a0fSBarry Smith PetscFunctionBegin; 2291d5ea218eSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2292d5ea218eSStefano Zampini PetscValidPointer(array, 2); 2293cac4c232SBarry Smith PetscUseMethod(A, "MatDenseRestoreArrayRead_C", (Mat, const PetscScalar **), (A, array)); 229473a71a0fSBarry Smith PetscFunctionReturn(0); 229573a71a0fSBarry Smith } 229673a71a0fSBarry Smith 22976947451fSStefano Zampini /*@C 229811a5261eSBarry Smith MatDenseGetArrayWrite - gives write-only access to the array where the data for a `MATDENSE` matrix is stored 22996947451fSStefano Zampini 23000ab4885dSBarry Smith Not Collective; No Fortran Support 23016947451fSStefano Zampini 23026947451fSStefano Zampini Input Parameter: 23036947451fSStefano Zampini . mat - a dense matrix 23046947451fSStefano Zampini 23056947451fSStefano Zampini Output Parameter: 23066947451fSStefano Zampini . array - pointer to the data 23076947451fSStefano Zampini 23086947451fSStefano Zampini Level: intermediate 23096947451fSStefano Zampini 231011a5261eSBarry Smith .seealso: `MATDENSE`, `MatDenseRestoreArrayWrite()`, `MatDenseGetArray()`, `MatDenseRestoreArray()`, `MatDenseGetArrayRead()`, `MatDenseRestoreArrayRead()` 23116947451fSStefano Zampini @*/ 2312d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetArrayWrite(Mat A, PetscScalar **array) 2313d71ae5a4SJacob Faibussowitsch { 23146947451fSStefano Zampini PetscFunctionBegin; 2315d5ea218eSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2316d5ea218eSStefano Zampini PetscValidPointer(array, 2); 2317cac4c232SBarry Smith PetscUseMethod(A, "MatDenseGetArrayWrite_C", (Mat, PetscScalar **), (A, array)); 23186947451fSStefano Zampini PetscFunctionReturn(0); 23196947451fSStefano Zampini } 23206947451fSStefano Zampini 23216947451fSStefano Zampini /*@C 232211a5261eSBarry Smith MatDenseRestoreArrayWrite - returns access to the array where the data for a `MATDENSE` matrix is stored obtained by `MatDenseGetArrayWrite()` 23236947451fSStefano Zampini 23240ab4885dSBarry Smith Not Collective; No Fortran Support 23256947451fSStefano Zampini 23266947451fSStefano Zampini Input Parameters: 23276947451fSStefano Zampini + mat - a dense matrix 2328742765d3SMatthew Knepley - array - pointer to the data (may be NULL) 23296947451fSStefano Zampini 23306947451fSStefano Zampini Level: intermediate 23316947451fSStefano Zampini 233211a5261eSBarry Smith .seealso: `MATDENSE`, `MatDenseGetArrayWrite()`, `MatDenseGetArray()`, `MatDenseRestoreArray()`, `MatDenseGetArrayRead()`, `MatDenseRestoreArrayRead()` 23336947451fSStefano Zampini @*/ 2334d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreArrayWrite(Mat A, PetscScalar **array) 2335d71ae5a4SJacob Faibussowitsch { 23366947451fSStefano Zampini PetscFunctionBegin; 2337d5ea218eSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2338d5ea218eSStefano Zampini PetscValidPointer(array, 2); 2339cac4c232SBarry Smith PetscUseMethod(A, "MatDenseRestoreArrayWrite_C", (Mat, PetscScalar **), (A, array)); 23409566063dSJacob Faibussowitsch PetscCall(PetscObjectStateIncrease((PetscObject)A)); 234147d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) || defined(PETSC_HAVE_HIP) 23426947451fSStefano Zampini A->offloadmask = PETSC_OFFLOAD_CPU; 23436947451fSStefano Zampini #endif 23446947451fSStefano Zampini PetscFunctionReturn(0); 23456947451fSStefano Zampini } 23466947451fSStefano Zampini 2347cd3f9d89SJunchao Zhang /*@C 2348cd3f9d89SJunchao Zhang MatDenseGetArrayAndMemType - gives read-write access to the array where the data for a `MATDENSE` matrix is stored 2349cd3f9d89SJunchao Zhang 2350cd3f9d89SJunchao Zhang Logically Collective 2351cd3f9d89SJunchao Zhang 2352cd3f9d89SJunchao Zhang Input Parameter: 2353cd3f9d89SJunchao Zhang . mat - a dense matrix 2354cd3f9d89SJunchao Zhang 2355cd3f9d89SJunchao Zhang Output Parameters: 2356cd3f9d89SJunchao Zhang + array - pointer to the data 2357cd3f9d89SJunchao Zhang - mtype - memory type of the returned pointer 2358cd3f9d89SJunchao Zhang 2359cd3f9d89SJunchao Zhang Notes: 2360*e865de01SJunchao Zhang If the matrix is of a device type such as MATDENSECUDA, MATDENSEHIP, etc., 2361cd3f9d89SJunchao Zhang an array on device is always returned and is guaranteed to contain the matrix's latest data. 2362cd3f9d89SJunchao Zhang 2363cd3f9d89SJunchao Zhang Level: intermediate 2364cd3f9d89SJunchao Zhang 2365cd3f9d89SJunchao Zhang .seealso: `MATDENSE`, `MatDenseRestoreArrayAndMemType()`, `MatDenseGetArrayReadAndMemType()`, `MatDenseGetArrayWriteAndMemType()`, `MatDenseGetArrayRead()`, 2366cd3f9d89SJunchao Zhang `MatDenseRestoreArrayRead()`, `MatDenseGetArrayWrite()`, `MatDenseRestoreArrayWrite()`, `MatSeqAIJGetCSRAndMemType()` 2367cd3f9d89SJunchao Zhang @*/ 2368cd3f9d89SJunchao Zhang PetscErrorCode MatDenseGetArrayAndMemType(Mat A, PetscScalar **array, PetscMemType *mtype) 2369cd3f9d89SJunchao Zhang { 2370cd3f9d89SJunchao Zhang PetscBool isMPI; 2371cd3f9d89SJunchao Zhang 2372cd3f9d89SJunchao Zhang PetscFunctionBegin; 2373cd3f9d89SJunchao Zhang PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2374cd3f9d89SJunchao Zhang PetscValidPointer(array, 2); 2375*e865de01SJunchao Zhang PetscCall(MatBindToCPU(A, PETSC_FALSE)); /* We want device matrices to always return device arrays, so we unbind the matrix if it is bound to CPU */ 2376cd3f9d89SJunchao Zhang PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATMPIDENSE, &isMPI)); 2377cd3f9d89SJunchao Zhang if (isMPI) { 2378cd3f9d89SJunchao Zhang /* Dispatch here so that the code can be reused for all subclasses of MATDENSE */ 2379cd3f9d89SJunchao Zhang PetscCall(MatDenseGetArrayAndMemType(((Mat_MPIDense *)A->data)->A, array, mtype)); 2380cd3f9d89SJunchao Zhang } else { 2381cd3f9d89SJunchao Zhang PetscErrorCode (*fptr)(Mat, PetscScalar **, PetscMemType *); 2382cd3f9d89SJunchao Zhang PetscObjectQueryFunction((PetscObject)A, "MatDenseGetArrayAndMemType_C", &fptr); 2383cd3f9d89SJunchao Zhang if (fptr) { 2384cd3f9d89SJunchao Zhang PetscCall((*fptr)(A, array, mtype)); 2385cd3f9d89SJunchao Zhang } else { 2386cd3f9d89SJunchao Zhang PetscUseMethod(A, "MatDenseGetArray_C", (Mat, PetscScalar **), (A, array)); 2387cd3f9d89SJunchao Zhang if (mtype) *mtype = PETSC_MEMTYPE_HOST; 2388cd3f9d89SJunchao Zhang } 2389cd3f9d89SJunchao Zhang } 2390cd3f9d89SJunchao Zhang PetscFunctionReturn(0); 2391cd3f9d89SJunchao Zhang } 2392cd3f9d89SJunchao Zhang 2393cd3f9d89SJunchao Zhang /*@C 2394cd3f9d89SJunchao Zhang MatDenseRestoreArrayAndMemType - returns access to the array that is obtained by `MatDenseGetArrayAndMemType()` 2395cd3f9d89SJunchao Zhang 2396cd3f9d89SJunchao Zhang Logically Collective 2397cd3f9d89SJunchao Zhang 2398cd3f9d89SJunchao Zhang Input Parameters: 2399cd3f9d89SJunchao Zhang + mat - a dense matrix 2400cd3f9d89SJunchao Zhang - array - pointer to the data 2401cd3f9d89SJunchao Zhang 2402cd3f9d89SJunchao Zhang Level: intermediate 2403cd3f9d89SJunchao Zhang 2404cd3f9d89SJunchao Zhang .seealso: `MATDENSE`, `MatDenseGetArrayAndMemType()`, `MatDenseGetArray()`, `MatDenseGetArrayRead()`, `MatDenseRestoreArrayRead()`, `MatDenseGetArrayWrite()`, `MatDenseRestoreArrayWrite()` 2405cd3f9d89SJunchao Zhang @*/ 2406cd3f9d89SJunchao Zhang PetscErrorCode MatDenseRestoreArrayAndMemType(Mat A, PetscScalar **array) 2407cd3f9d89SJunchao Zhang { 2408cd3f9d89SJunchao Zhang PetscBool isMPI; 2409cd3f9d89SJunchao Zhang 2410cd3f9d89SJunchao Zhang PetscFunctionBegin; 2411cd3f9d89SJunchao Zhang PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2412cd3f9d89SJunchao Zhang PetscValidPointer(array, 2); 2413cd3f9d89SJunchao Zhang PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATMPIDENSE, &isMPI)); 2414cd3f9d89SJunchao Zhang if (isMPI) { 2415cd3f9d89SJunchao Zhang PetscCall(MatDenseRestoreArrayAndMemType(((Mat_MPIDense *)A->data)->A, array)); 2416cd3f9d89SJunchao Zhang } else { 2417cd3f9d89SJunchao Zhang PetscErrorCode (*fptr)(Mat, PetscScalar **); 2418cd3f9d89SJunchao Zhang PetscObjectQueryFunction((PetscObject)A, "MatDenseRestoreArrayAndMemType_C", &fptr); 2419cd3f9d89SJunchao Zhang if (fptr) { 2420cd3f9d89SJunchao Zhang PetscCall((*fptr)(A, array)); 2421cd3f9d89SJunchao Zhang } else { 2422cd3f9d89SJunchao Zhang PetscUseMethod(A, "MatDenseRestoreArray_C", (Mat, PetscScalar **), (A, array)); 2423cd3f9d89SJunchao Zhang } 2424cd3f9d89SJunchao Zhang *array = NULL; 2425cd3f9d89SJunchao Zhang } 2426cd3f9d89SJunchao Zhang PetscCall(PetscObjectStateIncrease((PetscObject)A)); 2427cd3f9d89SJunchao Zhang PetscFunctionReturn(0); 2428cd3f9d89SJunchao Zhang } 2429cd3f9d89SJunchao Zhang 2430cd3f9d89SJunchao Zhang /*@C 2431cd3f9d89SJunchao Zhang MatDenseGetArrayReadAndMemType - gives read-only access to the array where the data for a `MATDENSE` matrix is stored 2432cd3f9d89SJunchao Zhang 2433cd3f9d89SJunchao Zhang Logically Collective 2434cd3f9d89SJunchao Zhang 2435cd3f9d89SJunchao Zhang Input Parameter: 2436cd3f9d89SJunchao Zhang . mat - a dense matrix 2437cd3f9d89SJunchao Zhang 2438cd3f9d89SJunchao Zhang Output Parameters: 2439cd3f9d89SJunchao Zhang + array - pointer to the data 2440cd3f9d89SJunchao Zhang - mtype - memory type of the returned pointer 2441cd3f9d89SJunchao Zhang 2442cd3f9d89SJunchao Zhang Notes: 2443*e865de01SJunchao Zhang If the matrix is of a device type such as MATDENSECUDA, MATDENSEHIP, etc., 2444cd3f9d89SJunchao Zhang an array on device is always returned and is guaranteed to contain the matrix's latest data. 2445cd3f9d89SJunchao Zhang 2446cd3f9d89SJunchao Zhang Level: intermediate 2447cd3f9d89SJunchao Zhang 2448cd3f9d89SJunchao Zhang .seealso: `MATDENSE`, `MatDenseRestoreArrayReadAndMemType()`, `MatDenseGetArrayWriteAndMemType()`, 2449cd3f9d89SJunchao Zhang `MatDenseGetArrayRead()`, `MatDenseRestoreArrayRead()`, `MatDenseGetArrayWrite()`, `MatDenseRestoreArrayWrite()`, `MatSeqAIJGetCSRAndMemType()` 2450cd3f9d89SJunchao Zhang @*/ 2451cd3f9d89SJunchao Zhang PetscErrorCode MatDenseGetArrayReadAndMemType(Mat A, const PetscScalar **array, PetscMemType *mtype) 2452cd3f9d89SJunchao Zhang { 2453cd3f9d89SJunchao Zhang PetscBool isMPI; 2454cd3f9d89SJunchao Zhang 2455cd3f9d89SJunchao Zhang PetscFunctionBegin; 2456cd3f9d89SJunchao Zhang PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2457cd3f9d89SJunchao Zhang PetscValidPointer(array, 2); 2458*e865de01SJunchao Zhang PetscCall(MatBindToCPU(A, PETSC_FALSE)); /* We want device matrices to always return device arrays, so we unbind the matrix if it is bound to CPU */ 2459cd3f9d89SJunchao Zhang PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATMPIDENSE, &isMPI)); 2460cd3f9d89SJunchao Zhang if (isMPI) { /* Dispatch here so that the code can be reused for all subclasses of MATDENSE */ 2461cd3f9d89SJunchao Zhang PetscCall(MatDenseGetArrayReadAndMemType(((Mat_MPIDense *)A->data)->A, array, mtype)); 2462cd3f9d89SJunchao Zhang } else { 2463cd3f9d89SJunchao Zhang PetscErrorCode (*fptr)(Mat, const PetscScalar **, PetscMemType *); 2464cd3f9d89SJunchao Zhang PetscObjectQueryFunction((PetscObject)A, "MatDenseGetArrayReadAndMemType_C", &fptr); 2465cd3f9d89SJunchao Zhang if (fptr) { 2466cd3f9d89SJunchao Zhang PetscCall((*fptr)(A, array, mtype)); 2467cd3f9d89SJunchao Zhang } else { 2468cd3f9d89SJunchao Zhang PetscUseMethod(A, "MatDenseGetArrayRead_C", (Mat, const PetscScalar **), (A, array)); 2469cd3f9d89SJunchao Zhang if (mtype) *mtype = PETSC_MEMTYPE_HOST; 2470cd3f9d89SJunchao Zhang } 2471cd3f9d89SJunchao Zhang } 2472cd3f9d89SJunchao Zhang PetscFunctionReturn(0); 2473cd3f9d89SJunchao Zhang } 2474cd3f9d89SJunchao Zhang 2475cd3f9d89SJunchao Zhang /*@C 2476cd3f9d89SJunchao Zhang MatDenseRestoreArrayReadAndMemType - returns access to the array that is obtained by `MatDenseGetArrayReadAndMemType()` 2477cd3f9d89SJunchao Zhang 2478cd3f9d89SJunchao Zhang Logically Collective 2479cd3f9d89SJunchao Zhang 2480cd3f9d89SJunchao Zhang Input Parameters: 2481cd3f9d89SJunchao Zhang + mat - a dense matrix 2482cd3f9d89SJunchao Zhang - array - pointer to the data 2483cd3f9d89SJunchao Zhang 2484cd3f9d89SJunchao Zhang Level: intermediate 2485cd3f9d89SJunchao Zhang 2486cd3f9d89SJunchao Zhang .seealso: `MATDENSE`, `MatDenseGetArrayReadAndMemType()`, `MatDenseGetArray()`, `MatDenseGetArrayRead()`, `MatDenseRestoreArrayRead()`, `MatDenseGetArrayWrite()`, `MatDenseRestoreArrayWrite()` 2487cd3f9d89SJunchao Zhang @*/ 2488cd3f9d89SJunchao Zhang PetscErrorCode MatDenseRestoreArrayReadAndMemType(Mat A, const PetscScalar **array) 2489cd3f9d89SJunchao Zhang { 2490cd3f9d89SJunchao Zhang PetscBool isMPI; 2491cd3f9d89SJunchao Zhang 2492cd3f9d89SJunchao Zhang PetscFunctionBegin; 2493cd3f9d89SJunchao Zhang PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2494cd3f9d89SJunchao Zhang PetscValidPointer(array, 2); 2495cd3f9d89SJunchao Zhang PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATMPIDENSE, &isMPI)); 2496cd3f9d89SJunchao Zhang if (isMPI) { 2497cd3f9d89SJunchao Zhang PetscCall(MatDenseRestoreArrayReadAndMemType(((Mat_MPIDense *)A->data)->A, array)); 2498cd3f9d89SJunchao Zhang } else { 2499cd3f9d89SJunchao Zhang PetscErrorCode (*fptr)(Mat, const PetscScalar **); 2500cd3f9d89SJunchao Zhang PetscObjectQueryFunction((PetscObject)A, "MatDenseRestoreArrayReadAndMemType_C", &fptr); 2501cd3f9d89SJunchao Zhang if (fptr) { 2502cd3f9d89SJunchao Zhang PetscCall((*fptr)(A, array)); 2503cd3f9d89SJunchao Zhang } else { 2504cd3f9d89SJunchao Zhang PetscUseMethod(A, "MatDenseRestoreArrayRead_C", (Mat, const PetscScalar **), (A, array)); 2505cd3f9d89SJunchao Zhang } 2506cd3f9d89SJunchao Zhang *array = NULL; 2507cd3f9d89SJunchao Zhang } 2508cd3f9d89SJunchao Zhang PetscFunctionReturn(0); 2509cd3f9d89SJunchao Zhang } 2510cd3f9d89SJunchao Zhang 2511cd3f9d89SJunchao Zhang /*@C 2512cd3f9d89SJunchao Zhang MatDenseGetArrayWriteAndMemType - gives write-only access to the array where the data for a `MATDENSE` matrix is stored 2513cd3f9d89SJunchao Zhang 2514cd3f9d89SJunchao Zhang Logically Collective 2515cd3f9d89SJunchao Zhang 2516cd3f9d89SJunchao Zhang Input Parameter: 2517cd3f9d89SJunchao Zhang . mat - a dense matrix 2518cd3f9d89SJunchao Zhang 2519cd3f9d89SJunchao Zhang Output Parameters: 2520cd3f9d89SJunchao Zhang + array - pointer to the data 2521cd3f9d89SJunchao Zhang - mtype - memory type of the returned pointer 2522cd3f9d89SJunchao Zhang 2523cd3f9d89SJunchao Zhang Notes: 2524*e865de01SJunchao Zhang If the matrix is of a device type such as MATDENSECUDA, MATDENSEHIP, etc., 2525cd3f9d89SJunchao Zhang an array on device is always returned and is guaranteed to contain the matrix's latest data. 2526cd3f9d89SJunchao Zhang 2527cd3f9d89SJunchao Zhang Level: intermediate 2528cd3f9d89SJunchao Zhang 2529cd3f9d89SJunchao Zhang .seealso: `MATDENSE`, `MatDenseRestoreArrayWriteAndMemType()`, `MatDenseGetArrayReadAndMemType()`, `MatDenseGetArrayRead()`, 2530cd3f9d89SJunchao Zhang `MatDenseRestoreArrayRead()`, `MatDenseGetArrayWrite()`, `MatDenseRestoreArrayWrite()`, `MatSeqAIJGetCSRAndMemType()` 2531cd3f9d89SJunchao Zhang @*/ 2532cd3f9d89SJunchao Zhang PetscErrorCode MatDenseGetArrayWriteAndMemType(Mat A, PetscScalar **array, PetscMemType *mtype) 2533cd3f9d89SJunchao Zhang { 2534cd3f9d89SJunchao Zhang PetscBool isMPI; 2535cd3f9d89SJunchao Zhang 2536cd3f9d89SJunchao Zhang PetscFunctionBegin; 2537cd3f9d89SJunchao Zhang PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2538cd3f9d89SJunchao Zhang PetscValidPointer(array, 2); 2539*e865de01SJunchao Zhang PetscCall(MatBindToCPU(A, PETSC_FALSE)); /* We want device matrices to always return device arrays, so we unbind the matrix if it is bound to CPU */ 2540cd3f9d89SJunchao Zhang PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATMPIDENSE, &isMPI)); 2541cd3f9d89SJunchao Zhang if (isMPI) { 2542cd3f9d89SJunchao Zhang PetscCall(MatDenseGetArrayWriteAndMemType(((Mat_MPIDense *)A->data)->A, array, mtype)); 2543cd3f9d89SJunchao Zhang } else { 2544cd3f9d89SJunchao Zhang PetscErrorCode (*fptr)(Mat, PetscScalar **, PetscMemType *); 2545cd3f9d89SJunchao Zhang PetscObjectQueryFunction((PetscObject)A, "MatDenseGetArrayWriteAndMemType_C", &fptr); 2546cd3f9d89SJunchao Zhang if (fptr) { 2547cd3f9d89SJunchao Zhang PetscCall((*fptr)(A, array, mtype)); 2548cd3f9d89SJunchao Zhang } else { 2549cd3f9d89SJunchao Zhang PetscUseMethod(A, "MatDenseGetArrayWrite_C", (Mat, PetscScalar **), (A, array)); 2550cd3f9d89SJunchao Zhang if (mtype) *mtype = PETSC_MEMTYPE_HOST; 2551cd3f9d89SJunchao Zhang } 2552cd3f9d89SJunchao Zhang } 2553cd3f9d89SJunchao Zhang PetscFunctionReturn(0); 2554cd3f9d89SJunchao Zhang } 2555cd3f9d89SJunchao Zhang 2556cd3f9d89SJunchao Zhang /*@C 2557cd3f9d89SJunchao Zhang MatDenseRestoreArrayWriteAndMemType - returns access to the array that is obtained by `MatDenseGetArrayReadAndMemType()` 2558cd3f9d89SJunchao Zhang 2559cd3f9d89SJunchao Zhang Logically Collective 2560cd3f9d89SJunchao Zhang 2561cd3f9d89SJunchao Zhang Input Parameters: 2562cd3f9d89SJunchao Zhang + mat - a dense matrix 2563cd3f9d89SJunchao Zhang - array - pointer to the data 2564cd3f9d89SJunchao Zhang 2565cd3f9d89SJunchao Zhang Level: intermediate 2566cd3f9d89SJunchao Zhang 2567cd3f9d89SJunchao Zhang .seealso: `MATDENSE`, `MatDenseGetArrayWriteAndMemType()`, `MatDenseGetArray()`, `MatDenseGetArrayRead()`, `MatDenseRestoreArrayRead()`, `MatDenseGetArrayWrite()`, `MatDenseRestoreArrayWrite()` 2568cd3f9d89SJunchao Zhang @*/ 2569cd3f9d89SJunchao Zhang PetscErrorCode MatDenseRestoreArrayWriteAndMemType(Mat A, PetscScalar **array) 2570cd3f9d89SJunchao Zhang { 2571cd3f9d89SJunchao Zhang PetscBool isMPI; 2572cd3f9d89SJunchao Zhang 2573cd3f9d89SJunchao Zhang PetscFunctionBegin; 2574cd3f9d89SJunchao Zhang PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 2575cd3f9d89SJunchao Zhang PetscValidPointer(array, 2); 2576cd3f9d89SJunchao Zhang PetscCall(PetscObjectBaseTypeCompare((PetscObject)A, MATMPIDENSE, &isMPI)); 2577cd3f9d89SJunchao Zhang if (isMPI) { 2578cd3f9d89SJunchao Zhang PetscCall(MatDenseRestoreArrayWriteAndMemType(((Mat_MPIDense *)A->data)->A, array)); 2579cd3f9d89SJunchao Zhang } else { 2580cd3f9d89SJunchao Zhang PetscErrorCode (*fptr)(Mat, PetscScalar **); 2581cd3f9d89SJunchao Zhang PetscObjectQueryFunction((PetscObject)A, "MatDenseRestoreArrayWriteAndMemType_C", &fptr); 2582cd3f9d89SJunchao Zhang if (fptr) { 2583cd3f9d89SJunchao Zhang PetscCall((*fptr)(A, array)); 2584cd3f9d89SJunchao Zhang } else { 2585cd3f9d89SJunchao Zhang PetscUseMethod(A, "MatDenseRestoreArrayWrite_C", (Mat, PetscScalar **), (A, array)); 2586cd3f9d89SJunchao Zhang } 2587cd3f9d89SJunchao Zhang *array = NULL; 2588cd3f9d89SJunchao Zhang } 2589cd3f9d89SJunchao Zhang PetscCall(PetscObjectStateIncrease((PetscObject)A)); 2590cd3f9d89SJunchao Zhang PetscFunctionReturn(0); 2591cd3f9d89SJunchao Zhang } 2592cd3f9d89SJunchao Zhang 2593d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatCreateSubMatrix_SeqDense(Mat A, IS isrow, IS iscol, MatReuse scall, Mat *B) 2594d71ae5a4SJacob Faibussowitsch { 2595c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 2596bf5a80bcSToby Isaac PetscInt i, j, nrows, ncols, ldb; 25975d0c19d7SBarry Smith const PetscInt *irow, *icol; 259887828ca2SBarry Smith PetscScalar *av, *bv, *v = mat->v; 25990754003eSLois Curfman McInnes Mat newmat; 26000754003eSLois Curfman McInnes 26013a40ed3dSBarry Smith PetscFunctionBegin; 26029566063dSJacob Faibussowitsch PetscCall(ISGetIndices(isrow, &irow)); 26039566063dSJacob Faibussowitsch PetscCall(ISGetIndices(iscol, &icol)); 26049566063dSJacob Faibussowitsch PetscCall(ISGetLocalSize(isrow, &nrows)); 26059566063dSJacob Faibussowitsch PetscCall(ISGetLocalSize(iscol, &ncols)); 26060754003eSLois Curfman McInnes 2607182d2002SSatish Balay /* Check submatrixcall */ 2608182d2002SSatish Balay if (scall == MAT_REUSE_MATRIX) { 260913f74950SBarry Smith PetscInt n_cols, n_rows; 26109566063dSJacob Faibussowitsch PetscCall(MatGetSize(*B, &n_rows, &n_cols)); 261121a2c019SBarry Smith if (n_rows != nrows || n_cols != ncols) { 2612f746d493SDmitry Karpeev /* resize the result matrix to match number of requested rows/columns */ 26139566063dSJacob Faibussowitsch PetscCall(MatSetSizes(*B, nrows, ncols, nrows, ncols)); 261421a2c019SBarry Smith } 2615182d2002SSatish Balay newmat = *B; 2616182d2002SSatish Balay } else { 26170754003eSLois Curfman McInnes /* Create and fill new matrix */ 26189566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &newmat)); 26199566063dSJacob Faibussowitsch PetscCall(MatSetSizes(newmat, nrows, ncols, nrows, ncols)); 26209566063dSJacob Faibussowitsch PetscCall(MatSetType(newmat, ((PetscObject)A)->type_name)); 26219566063dSJacob Faibussowitsch PetscCall(MatSeqDenseSetPreallocation(newmat, NULL)); 2622182d2002SSatish Balay } 2623182d2002SSatish Balay 2624182d2002SSatish Balay /* Now extract the data pointers and do the copy,column at a time */ 26259566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(newmat, &bv)); 26269566063dSJacob Faibussowitsch PetscCall(MatDenseGetLDA(newmat, &ldb)); 2627182d2002SSatish Balay for (i = 0; i < ncols; i++) { 26286de62eeeSBarry Smith av = v + mat->lda * icol[i]; 2629ca15aa20SStefano Zampini for (j = 0; j < nrows; j++) bv[j] = av[irow[j]]; 2630bf5a80bcSToby Isaac bv += ldb; 26310754003eSLois Curfman McInnes } 26329566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(newmat, &bv)); 2633182d2002SSatish Balay 2634182d2002SSatish Balay /* Assemble the matrices so that the correct flags are set */ 26359566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(newmat, MAT_FINAL_ASSEMBLY)); 26369566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(newmat, MAT_FINAL_ASSEMBLY)); 26370754003eSLois Curfman McInnes 26380754003eSLois Curfman McInnes /* Free work space */ 26399566063dSJacob Faibussowitsch PetscCall(ISRestoreIndices(isrow, &irow)); 26409566063dSJacob Faibussowitsch PetscCall(ISRestoreIndices(iscol, &icol)); 2641182d2002SSatish Balay *B = newmat; 26423a40ed3dSBarry Smith PetscFunctionReturn(0); 26430754003eSLois Curfman McInnes } 26440754003eSLois Curfman McInnes 2645d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatCreateSubMatrices_SeqDense(Mat A, PetscInt n, const IS irow[], const IS icol[], MatReuse scall, Mat *B[]) 2646d71ae5a4SJacob Faibussowitsch { 264713f74950SBarry Smith PetscInt i; 2648905e6a2fSBarry Smith 26493a40ed3dSBarry Smith PetscFunctionBegin; 265048a46eb9SPierre Jolivet if (scall == MAT_INITIAL_MATRIX) PetscCall(PetscCalloc1(n, B)); 2651905e6a2fSBarry Smith 265248a46eb9SPierre Jolivet for (i = 0; i < n; i++) PetscCall(MatCreateSubMatrix_SeqDense(A, irow[i], icol[i], scall, &(*B)[i])); 26533a40ed3dSBarry Smith PetscFunctionReturn(0); 2654905e6a2fSBarry Smith } 2655905e6a2fSBarry Smith 2656d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatAssemblyBegin_SeqDense(Mat mat, MatAssemblyType mode) 2657d71ae5a4SJacob Faibussowitsch { 2658c0aa2d19SHong Zhang PetscFunctionBegin; 2659c0aa2d19SHong Zhang PetscFunctionReturn(0); 2660c0aa2d19SHong Zhang } 2661c0aa2d19SHong Zhang 2662d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatAssemblyEnd_SeqDense(Mat mat, MatAssemblyType mode) 2663d71ae5a4SJacob Faibussowitsch { 2664c0aa2d19SHong Zhang PetscFunctionBegin; 2665c0aa2d19SHong Zhang PetscFunctionReturn(0); 2666c0aa2d19SHong Zhang } 2667c0aa2d19SHong Zhang 2668d71ae5a4SJacob Faibussowitsch PetscErrorCode MatCopy_SeqDense(Mat A, Mat B, MatStructure str) 2669d71ae5a4SJacob Faibussowitsch { 26704b0e389bSBarry Smith Mat_SeqDense *a = (Mat_SeqDense *)A->data, *b = (Mat_SeqDense *)B->data; 2671ca15aa20SStefano Zampini const PetscScalar *va; 2672ca15aa20SStefano Zampini PetscScalar *vb; 2673d0f46423SBarry Smith PetscInt lda1 = a->lda, lda2 = b->lda, m = A->rmap->n, n = A->cmap->n, j; 26743a40ed3dSBarry Smith 26753a40ed3dSBarry Smith PetscFunctionBegin; 267633f4a19fSKris Buschelman /* If the two matrices don't have the same copy implementation, they aren't compatible for fast copy. */ 267733f4a19fSKris Buschelman if (A->ops->copy != B->ops->copy) { 26789566063dSJacob Faibussowitsch PetscCall(MatCopy_Basic(A, B, str)); 26793a40ed3dSBarry Smith PetscFunctionReturn(0); 26803a40ed3dSBarry Smith } 2681aed4548fSBarry Smith PetscCheck(m == B->rmap->n && n == B->cmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "size(B) != size(A)"); 26829566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &va)); 26839566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(B, &vb)); 2684a5ce6ee0Svictorle if (lda1 > m || lda2 > m) { 268548a46eb9SPierre Jolivet for (j = 0; j < n; j++) PetscCall(PetscArraycpy(vb + j * lda2, va + j * lda1, m)); 2686a5ce6ee0Svictorle } else { 26879566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(vb, va, A->rmap->n * A->cmap->n)); 2688a5ce6ee0Svictorle } 26899566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(B, &vb)); 26909566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &va)); 26919566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(B, MAT_FINAL_ASSEMBLY)); 26929566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(B, MAT_FINAL_ASSEMBLY)); 2693273d9f13SBarry Smith PetscFunctionReturn(0); 2694273d9f13SBarry Smith } 2695273d9f13SBarry Smith 2696d71ae5a4SJacob Faibussowitsch PetscErrorCode MatSetUp_SeqDense(Mat A) 2697d71ae5a4SJacob Faibussowitsch { 2698273d9f13SBarry Smith PetscFunctionBegin; 26999566063dSJacob Faibussowitsch PetscCall(PetscLayoutSetUp(A->rmap)); 27009566063dSJacob Faibussowitsch PetscCall(PetscLayoutSetUp(A->cmap)); 270148a46eb9SPierre Jolivet if (!A->preallocated) PetscCall(MatSeqDenseSetPreallocation(A, NULL)); 27023a40ed3dSBarry Smith PetscFunctionReturn(0); 27034b0e389bSBarry Smith } 27044b0e389bSBarry Smith 2705d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatConjugate_SeqDense(Mat A) 2706d71ae5a4SJacob Faibussowitsch { 27074396437dSToby Isaac Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 270806c5243aSJose E. Roman PetscInt i, j; 27094396437dSToby Isaac PetscInt min = PetscMin(A->rmap->n, A->cmap->n); 2710ca15aa20SStefano Zampini PetscScalar *aa; 2711ba337c44SJed Brown 2712ba337c44SJed Brown PetscFunctionBegin; 27139566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, &aa)); 271406c5243aSJose E. Roman for (j = 0; j < A->cmap->n; j++) { 271506c5243aSJose E. Roman for (i = 0; i < A->rmap->n; i++) aa[i + j * mat->lda] = PetscConj(aa[i + j * mat->lda]); 271606c5243aSJose E. Roman } 27179566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, &aa)); 27189371c9d4SSatish Balay if (mat->tau) 27199371c9d4SSatish Balay for (i = 0; i < min; i++) mat->tau[i] = PetscConj(mat->tau[i]); 2720ba337c44SJed Brown PetscFunctionReturn(0); 2721ba337c44SJed Brown } 2722ba337c44SJed Brown 2723d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatRealPart_SeqDense(Mat A) 2724d71ae5a4SJacob Faibussowitsch { 272506c5243aSJose E. Roman Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 272606c5243aSJose E. Roman PetscInt i, j; 2727ca15aa20SStefano Zampini PetscScalar *aa; 2728ba337c44SJed Brown 2729ba337c44SJed Brown PetscFunctionBegin; 27309566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, &aa)); 273106c5243aSJose E. Roman for (j = 0; j < A->cmap->n; j++) { 273206c5243aSJose E. Roman for (i = 0; i < A->rmap->n; i++) aa[i + j * mat->lda] = PetscRealPart(aa[i + j * mat->lda]); 273306c5243aSJose E. Roman } 27349566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, &aa)); 2735ba337c44SJed Brown PetscFunctionReturn(0); 2736ba337c44SJed Brown } 2737ba337c44SJed Brown 2738d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatImaginaryPart_SeqDense(Mat A) 2739d71ae5a4SJacob Faibussowitsch { 274006c5243aSJose E. Roman Mat_SeqDense *mat = (Mat_SeqDense *)A->data; 274106c5243aSJose E. Roman PetscInt i, j; 2742ca15aa20SStefano Zampini PetscScalar *aa; 2743ba337c44SJed Brown 2744ba337c44SJed Brown PetscFunctionBegin; 27459566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, &aa)); 274606c5243aSJose E. Roman for (j = 0; j < A->cmap->n; j++) { 274706c5243aSJose E. Roman for (i = 0; i < A->rmap->n; i++) aa[i + j * mat->lda] = PetscImaginaryPart(aa[i + j * mat->lda]); 274806c5243aSJose E. Roman } 27499566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, &aa)); 2750ba337c44SJed Brown PetscFunctionReturn(0); 2751ba337c44SJed Brown } 2752284134d9SBarry Smith 2753a9fe9ddaSSatish Balay /* ----------------------------------------------------------------*/ 2754d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMatMultSymbolic_SeqDense_SeqDense(Mat A, Mat B, PetscReal fill, Mat C) 2755d71ae5a4SJacob Faibussowitsch { 2756d0f46423SBarry Smith PetscInt m = A->rmap->n, n = B->cmap->n; 275747d993e7Ssuyashtn PetscBool cisdense = PETSC_FALSE; 2758a9fe9ddaSSatish Balay 2759ee16a9a1SHong Zhang PetscFunctionBegin; 27609566063dSJacob Faibussowitsch PetscCall(MatSetSizes(C, m, n, m, n)); 276147d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) 27629566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompareAny((PetscObject)C, &cisdense, MATSEQDENSE, MATSEQDENSECUDA, "")); 276347d993e7Ssuyashtn #endif 276447d993e7Ssuyashtn #if defined(PETSC_HAVE_HIP) 276547d993e7Ssuyashtn PetscCall(PetscObjectTypeCompareAny((PetscObject)C, &cisdense, MATSEQDENSE, MATSEQDENSEHIP, "")); 276647d993e7Ssuyashtn #endif 27677a3c3d58SStefano Zampini if (!cisdense) { 27687a3c3d58SStefano Zampini PetscBool flg; 27697a3c3d58SStefano Zampini 27709566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)B, ((PetscObject)A)->type_name, &flg)); 27719566063dSJacob Faibussowitsch PetscCall(MatSetType(C, flg ? ((PetscObject)A)->type_name : MATDENSE)); 27727a3c3d58SStefano Zampini } 27739566063dSJacob Faibussowitsch PetscCall(MatSetUp(C)); 2774ee16a9a1SHong Zhang PetscFunctionReturn(0); 2775ee16a9a1SHong Zhang } 2776a9fe9ddaSSatish Balay 2777d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMatMultNumeric_SeqDense_SeqDense(Mat A, Mat B, Mat C) 2778d71ae5a4SJacob Faibussowitsch { 27796718818eSStefano Zampini Mat_SeqDense *a = (Mat_SeqDense *)A->data, *b = (Mat_SeqDense *)B->data, *c = (Mat_SeqDense *)C->data; 27800805154bSBarry Smith PetscBLASInt m, n, k; 2781ca15aa20SStefano Zampini const PetscScalar *av, *bv; 2782ca15aa20SStefano Zampini PetscScalar *cv; 2783a9fe9ddaSSatish Balay PetscScalar _DOne = 1.0, _DZero = 0.0; 2784a9fe9ddaSSatish Balay 2785a9fe9ddaSSatish Balay PetscFunctionBegin; 27869566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(C->rmap->n, &m)); 27879566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(C->cmap->n, &n)); 27889566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->cmap->n, &k)); 278949d0e964SStefano Zampini if (!m || !n || !k) PetscFunctionReturn(0); 27909566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &av)); 27919566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(B, &bv)); 27929566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayWrite(C, &cv)); 2793792fecdfSBarry Smith PetscCallBLAS("BLASgemm", BLASgemm_("N", "N", &m, &n, &k, &_DOne, av, &a->lda, bv, &b->lda, &_DZero, cv, &c->lda)); 27949566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(1.0 * m * n * k + 1.0 * m * n * (k - 1))); 27959566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &av)); 27969566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(B, &bv)); 27979566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayWrite(C, &cv)); 2798a9fe9ddaSSatish Balay PetscFunctionReturn(0); 2799a9fe9ddaSSatish Balay } 2800a9fe9ddaSSatish Balay 2801d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMatTransposeMultSymbolic_SeqDense_SeqDense(Mat A, Mat B, PetscReal fill, Mat C) 2802d71ae5a4SJacob Faibussowitsch { 280369f65d41SStefano Zampini PetscInt m = A->rmap->n, n = B->rmap->n; 280447d993e7Ssuyashtn PetscBool cisdense = PETSC_FALSE; 280569f65d41SStefano Zampini 280669f65d41SStefano Zampini PetscFunctionBegin; 28079566063dSJacob Faibussowitsch PetscCall(MatSetSizes(C, m, n, m, n)); 280847d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) 28099566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompareAny((PetscObject)C, &cisdense, MATSEQDENSE, MATSEQDENSECUDA, "")); 281047d993e7Ssuyashtn #endif 281147d993e7Ssuyashtn #if defined(PETSC_HAVE_HIP) 281247d993e7Ssuyashtn PetscCall(PetscObjectTypeCompareAny((PetscObject)C, &cisdense, MATSEQDENSE, MATSEQDENSEHIP, "")); 281347d993e7Ssuyashtn #endif 28147a3c3d58SStefano Zampini if (!cisdense) { 28157a3c3d58SStefano Zampini PetscBool flg; 28167a3c3d58SStefano Zampini 28179566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)B, ((PetscObject)A)->type_name, &flg)); 28189566063dSJacob Faibussowitsch PetscCall(MatSetType(C, flg ? ((PetscObject)A)->type_name : MATDENSE)); 28197a3c3d58SStefano Zampini } 28209566063dSJacob Faibussowitsch PetscCall(MatSetUp(C)); 282169f65d41SStefano Zampini PetscFunctionReturn(0); 282269f65d41SStefano Zampini } 282369f65d41SStefano Zampini 2824d71ae5a4SJacob Faibussowitsch PetscErrorCode MatMatTransposeMultNumeric_SeqDense_SeqDense(Mat A, Mat B, Mat C) 2825d71ae5a4SJacob Faibussowitsch { 282669f65d41SStefano Zampini Mat_SeqDense *a = (Mat_SeqDense *)A->data; 282769f65d41SStefano Zampini Mat_SeqDense *b = (Mat_SeqDense *)B->data; 282869f65d41SStefano Zampini Mat_SeqDense *c = (Mat_SeqDense *)C->data; 28296718818eSStefano Zampini const PetscScalar *av, *bv; 28306718818eSStefano Zampini PetscScalar *cv; 283169f65d41SStefano Zampini PetscBLASInt m, n, k; 283269f65d41SStefano Zampini PetscScalar _DOne = 1.0, _DZero = 0.0; 283369f65d41SStefano Zampini 283469f65d41SStefano Zampini PetscFunctionBegin; 28359566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(C->rmap->n, &m)); 28369566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(C->cmap->n, &n)); 28379566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->cmap->n, &k)); 283849d0e964SStefano Zampini if (!m || !n || !k) PetscFunctionReturn(0); 28399566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &av)); 28409566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(B, &bv)); 28419566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayWrite(C, &cv)); 2842792fecdfSBarry Smith PetscCallBLAS("BLASgemm", BLASgemm_("N", "T", &m, &n, &k, &_DOne, av, &a->lda, bv, &b->lda, &_DZero, cv, &c->lda)); 28439566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &av)); 28449566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(B, &bv)); 28459566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayWrite(C, &cv)); 28469566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(1.0 * m * n * k + 1.0 * m * n * (k - 1))); 284769f65d41SStefano Zampini PetscFunctionReturn(0); 284869f65d41SStefano Zampini } 284969f65d41SStefano Zampini 2850d71ae5a4SJacob Faibussowitsch PetscErrorCode MatTransposeMatMultSymbolic_SeqDense_SeqDense(Mat A, Mat B, PetscReal fill, Mat C) 2851d71ae5a4SJacob Faibussowitsch { 2852d0f46423SBarry Smith PetscInt m = A->cmap->n, n = B->cmap->n; 285347d993e7Ssuyashtn PetscBool cisdense = PETSC_FALSE; 2854a9fe9ddaSSatish Balay 2855ee16a9a1SHong Zhang PetscFunctionBegin; 28569566063dSJacob Faibussowitsch PetscCall(MatSetSizes(C, m, n, m, n)); 285747d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) 28589566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompareAny((PetscObject)C, &cisdense, MATSEQDENSE, MATSEQDENSECUDA, "")); 285947d993e7Ssuyashtn #endif 286047d993e7Ssuyashtn #if defined(PETSC_HAVE_HIP) 286147d993e7Ssuyashtn PetscCall(PetscObjectTypeCompareAny((PetscObject)C, &cisdense, MATSEQDENSE, MATSEQDENSEHIP, "")); 286247d993e7Ssuyashtn #endif 28637a3c3d58SStefano Zampini if (!cisdense) { 28647a3c3d58SStefano Zampini PetscBool flg; 28657a3c3d58SStefano Zampini 28669566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)B, ((PetscObject)A)->type_name, &flg)); 28679566063dSJacob Faibussowitsch PetscCall(MatSetType(C, flg ? ((PetscObject)A)->type_name : MATDENSE)); 28687a3c3d58SStefano Zampini } 28699566063dSJacob Faibussowitsch PetscCall(MatSetUp(C)); 2870ee16a9a1SHong Zhang PetscFunctionReturn(0); 2871ee16a9a1SHong Zhang } 2872a9fe9ddaSSatish Balay 2873d71ae5a4SJacob Faibussowitsch PetscErrorCode MatTransposeMatMultNumeric_SeqDense_SeqDense(Mat A, Mat B, Mat C) 2874d71ae5a4SJacob Faibussowitsch { 2875a9fe9ddaSSatish Balay Mat_SeqDense *a = (Mat_SeqDense *)A->data; 2876a9fe9ddaSSatish Balay Mat_SeqDense *b = (Mat_SeqDense *)B->data; 2877a9fe9ddaSSatish Balay Mat_SeqDense *c = (Mat_SeqDense *)C->data; 28786718818eSStefano Zampini const PetscScalar *av, *bv; 28796718818eSStefano Zampini PetscScalar *cv; 28800805154bSBarry Smith PetscBLASInt m, n, k; 2881a9fe9ddaSSatish Balay PetscScalar _DOne = 1.0, _DZero = 0.0; 2882a9fe9ddaSSatish Balay 2883a9fe9ddaSSatish Balay PetscFunctionBegin; 28849566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(C->rmap->n, &m)); 28859566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(C->cmap->n, &n)); 28869566063dSJacob Faibussowitsch PetscCall(PetscBLASIntCast(A->rmap->n, &k)); 288749d0e964SStefano Zampini if (!m || !n || !k) PetscFunctionReturn(0); 28889566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &av)); 28899566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(B, &bv)); 28909566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayWrite(C, &cv)); 2891792fecdfSBarry Smith PetscCallBLAS("BLASgemm", BLASgemm_("T", "N", &m, &n, &k, &_DOne, av, &a->lda, bv, &b->lda, &_DZero, cv, &c->lda)); 28929566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &av)); 28939566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(B, &bv)); 28949566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayWrite(C, &cv)); 28959566063dSJacob Faibussowitsch PetscCall(PetscLogFlops(1.0 * m * n * k + 1.0 * m * n * (k - 1))); 2896a9fe9ddaSSatish Balay PetscFunctionReturn(0); 2897a9fe9ddaSSatish Balay } 2898985db425SBarry Smith 28994222ddf1SHong Zhang /* ----------------------------------------------- */ 2900d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatProductSetFromOptions_SeqDense_AB(Mat C) 2901d71ae5a4SJacob Faibussowitsch { 29024222ddf1SHong Zhang PetscFunctionBegin; 29034222ddf1SHong Zhang C->ops->matmultsymbolic = MatMatMultSymbolic_SeqDense_SeqDense; 29044222ddf1SHong Zhang C->ops->productsymbolic = MatProductSymbolic_AB; 29054222ddf1SHong Zhang PetscFunctionReturn(0); 29064222ddf1SHong Zhang } 29074222ddf1SHong Zhang 2908d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatProductSetFromOptions_SeqDense_AtB(Mat C) 2909d71ae5a4SJacob Faibussowitsch { 29104222ddf1SHong Zhang PetscFunctionBegin; 29114222ddf1SHong Zhang C->ops->transposematmultsymbolic = MatTransposeMatMultSymbolic_SeqDense_SeqDense; 29124222ddf1SHong Zhang C->ops->productsymbolic = MatProductSymbolic_AtB; 29134222ddf1SHong Zhang PetscFunctionReturn(0); 29144222ddf1SHong Zhang } 29154222ddf1SHong Zhang 2916d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatProductSetFromOptions_SeqDense_ABt(Mat C) 2917d71ae5a4SJacob Faibussowitsch { 29184222ddf1SHong Zhang PetscFunctionBegin; 29194222ddf1SHong Zhang C->ops->mattransposemultsymbolic = MatMatTransposeMultSymbolic_SeqDense_SeqDense; 29204222ddf1SHong Zhang C->ops->productsymbolic = MatProductSymbolic_ABt; 29214222ddf1SHong Zhang PetscFunctionReturn(0); 29224222ddf1SHong Zhang } 29234222ddf1SHong Zhang 2924d71ae5a4SJacob Faibussowitsch PETSC_INTERN PetscErrorCode MatProductSetFromOptions_SeqDense(Mat C) 2925d71ae5a4SJacob Faibussowitsch { 29264222ddf1SHong Zhang Mat_Product *product = C->product; 29274222ddf1SHong Zhang 29284222ddf1SHong Zhang PetscFunctionBegin; 29294222ddf1SHong Zhang switch (product->type) { 2930d71ae5a4SJacob Faibussowitsch case MATPRODUCT_AB: 2931d71ae5a4SJacob Faibussowitsch PetscCall(MatProductSetFromOptions_SeqDense_AB(C)); 2932d71ae5a4SJacob Faibussowitsch break; 2933d71ae5a4SJacob Faibussowitsch case MATPRODUCT_AtB: 2934d71ae5a4SJacob Faibussowitsch PetscCall(MatProductSetFromOptions_SeqDense_AtB(C)); 2935d71ae5a4SJacob Faibussowitsch break; 2936d71ae5a4SJacob Faibussowitsch case MATPRODUCT_ABt: 2937d71ae5a4SJacob Faibussowitsch PetscCall(MatProductSetFromOptions_SeqDense_ABt(C)); 2938d71ae5a4SJacob Faibussowitsch break; 2939d71ae5a4SJacob Faibussowitsch default: 2940d71ae5a4SJacob Faibussowitsch break; 29414222ddf1SHong Zhang } 29424222ddf1SHong Zhang PetscFunctionReturn(0); 29434222ddf1SHong Zhang } 29444222ddf1SHong Zhang /* ----------------------------------------------- */ 29454222ddf1SHong Zhang 2946d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetRowMax_SeqDense(Mat A, Vec v, PetscInt idx[]) 2947d71ae5a4SJacob Faibussowitsch { 2948985db425SBarry Smith Mat_SeqDense *a = (Mat_SeqDense *)A->data; 2949d0f46423SBarry Smith PetscInt i, j, m = A->rmap->n, n = A->cmap->n, p; 2950985db425SBarry Smith PetscScalar *x; 2951ca15aa20SStefano Zampini const PetscScalar *aa; 2952985db425SBarry Smith 2953985db425SBarry Smith PetscFunctionBegin; 295428b400f6SJacob Faibussowitsch PetscCheck(!A->factortype, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "Not for factored matrix"); 29559566063dSJacob Faibussowitsch PetscCall(VecGetArray(v, &x)); 29569566063dSJacob Faibussowitsch PetscCall(VecGetLocalSize(v, &p)); 29579566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &aa)); 295808401ef6SPierre Jolivet PetscCheck(p == A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Nonconforming matrix and vector"); 2959985db425SBarry Smith for (i = 0; i < m; i++) { 29609371c9d4SSatish Balay x[i] = aa[i]; 29619371c9d4SSatish Balay if (idx) idx[i] = 0; 2962985db425SBarry Smith for (j = 1; j < n; j++) { 29639371c9d4SSatish Balay if (PetscRealPart(x[i]) < PetscRealPart(aa[i + a->lda * j])) { 29649371c9d4SSatish Balay x[i] = aa[i + a->lda * j]; 29659371c9d4SSatish Balay if (idx) idx[i] = j; 29669371c9d4SSatish Balay } 2967985db425SBarry Smith } 2968985db425SBarry Smith } 29699566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &aa)); 29709566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(v, &x)); 2971985db425SBarry Smith PetscFunctionReturn(0); 2972985db425SBarry Smith } 2973985db425SBarry Smith 2974d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetRowMaxAbs_SeqDense(Mat A, Vec v, PetscInt idx[]) 2975d71ae5a4SJacob Faibussowitsch { 2976985db425SBarry Smith Mat_SeqDense *a = (Mat_SeqDense *)A->data; 2977d0f46423SBarry Smith PetscInt i, j, m = A->rmap->n, n = A->cmap->n, p; 2978985db425SBarry Smith PetscScalar *x; 2979985db425SBarry Smith PetscReal atmp; 2980ca15aa20SStefano Zampini const PetscScalar *aa; 2981985db425SBarry Smith 2982985db425SBarry Smith PetscFunctionBegin; 298328b400f6SJacob Faibussowitsch PetscCheck(!A->factortype, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "Not for factored matrix"); 29849566063dSJacob Faibussowitsch PetscCall(VecGetArray(v, &x)); 29859566063dSJacob Faibussowitsch PetscCall(VecGetLocalSize(v, &p)); 29869566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &aa)); 298708401ef6SPierre Jolivet PetscCheck(p == A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Nonconforming matrix and vector"); 2988985db425SBarry Smith for (i = 0; i < m; i++) { 29899189402eSHong Zhang x[i] = PetscAbsScalar(aa[i]); 2990985db425SBarry Smith for (j = 1; j < n; j++) { 2991ca15aa20SStefano Zampini atmp = PetscAbsScalar(aa[i + a->lda * j]); 29929371c9d4SSatish Balay if (PetscAbsScalar(x[i]) < atmp) { 29939371c9d4SSatish Balay x[i] = atmp; 29949371c9d4SSatish Balay if (idx) idx[i] = j; 29959371c9d4SSatish Balay } 2996985db425SBarry Smith } 2997985db425SBarry Smith } 29989566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &aa)); 29999566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(v, &x)); 3000985db425SBarry Smith PetscFunctionReturn(0); 3001985db425SBarry Smith } 3002985db425SBarry Smith 3003d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatGetRowMin_SeqDense(Mat A, Vec v, PetscInt idx[]) 3004d71ae5a4SJacob Faibussowitsch { 3005985db425SBarry Smith Mat_SeqDense *a = (Mat_SeqDense *)A->data; 3006d0f46423SBarry Smith PetscInt i, j, m = A->rmap->n, n = A->cmap->n, p; 3007985db425SBarry Smith PetscScalar *x; 3008ca15aa20SStefano Zampini const PetscScalar *aa; 3009985db425SBarry Smith 3010985db425SBarry Smith PetscFunctionBegin; 301128b400f6SJacob Faibussowitsch PetscCheck(!A->factortype, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "Not for factored matrix"); 30129566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &aa)); 30139566063dSJacob Faibussowitsch PetscCall(VecGetArray(v, &x)); 30149566063dSJacob Faibussowitsch PetscCall(VecGetLocalSize(v, &p)); 301508401ef6SPierre Jolivet PetscCheck(p == A->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Nonconforming matrix and vector"); 3016985db425SBarry Smith for (i = 0; i < m; i++) { 30179371c9d4SSatish Balay x[i] = aa[i]; 30189371c9d4SSatish Balay if (idx) idx[i] = 0; 3019985db425SBarry Smith for (j = 1; j < n; j++) { 30209371c9d4SSatish Balay if (PetscRealPart(x[i]) > PetscRealPart(aa[i + a->lda * j])) { 30219371c9d4SSatish Balay x[i] = aa[i + a->lda * j]; 30229371c9d4SSatish Balay if (idx) idx[i] = j; 30239371c9d4SSatish Balay } 3024985db425SBarry Smith } 3025985db425SBarry Smith } 30269566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(v, &x)); 30279566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &aa)); 3028985db425SBarry Smith PetscFunctionReturn(0); 3029985db425SBarry Smith } 3030985db425SBarry Smith 3031d71ae5a4SJacob Faibussowitsch PetscErrorCode MatGetColumnVector_SeqDense(Mat A, Vec v, PetscInt col) 3032d71ae5a4SJacob Faibussowitsch { 30338d0534beSBarry Smith Mat_SeqDense *a = (Mat_SeqDense *)A->data; 30348d0534beSBarry Smith PetscScalar *x; 3035ca15aa20SStefano Zampini const PetscScalar *aa; 30368d0534beSBarry Smith 30378d0534beSBarry Smith PetscFunctionBegin; 303828b400f6SJacob Faibussowitsch PetscCheck(!A->factortype, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "Not for factored matrix"); 30399566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &aa)); 30409566063dSJacob Faibussowitsch PetscCall(VecGetArray(v, &x)); 30419566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(x, aa + col * a->lda, A->rmap->n)); 30429566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(v, &x)); 30439566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &aa)); 30448d0534beSBarry Smith PetscFunctionReturn(0); 30458d0534beSBarry Smith } 30468d0534beSBarry Smith 3047d71ae5a4SJacob Faibussowitsch PETSC_INTERN PetscErrorCode MatGetColumnReductions_SeqDense(Mat A, PetscInt type, PetscReal *reductions) 3048d71ae5a4SJacob Faibussowitsch { 30490716a85fSBarry Smith PetscInt i, j, m, n; 30501683a169SBarry Smith const PetscScalar *a; 30510716a85fSBarry Smith 30520716a85fSBarry Smith PetscFunctionBegin; 30539566063dSJacob Faibussowitsch PetscCall(MatGetSize(A, &m, &n)); 30549566063dSJacob Faibussowitsch PetscCall(PetscArrayzero(reductions, n)); 30559566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &a)); 3056857cbf51SRichard Tran Mills if (type == NORM_2) { 30570716a85fSBarry Smith for (i = 0; i < n; i++) { 3058ad540459SPierre Jolivet for (j = 0; j < m; j++) reductions[i] += PetscAbsScalar(a[j] * a[j]); 30590716a85fSBarry Smith a += m; 30600716a85fSBarry Smith } 3061857cbf51SRichard Tran Mills } else if (type == NORM_1) { 30620716a85fSBarry Smith for (i = 0; i < n; i++) { 3063ad540459SPierre Jolivet for (j = 0; j < m; j++) reductions[i] += PetscAbsScalar(a[j]); 30640716a85fSBarry Smith a += m; 30650716a85fSBarry Smith } 3066857cbf51SRichard Tran Mills } else if (type == NORM_INFINITY) { 30670716a85fSBarry Smith for (i = 0; i < n; i++) { 3068ad540459SPierre Jolivet for (j = 0; j < m; j++) reductions[i] = PetscMax(PetscAbsScalar(a[j]), reductions[i]); 30690716a85fSBarry Smith a += m; 30700716a85fSBarry Smith } 3071857cbf51SRichard Tran Mills } else if (type == REDUCTION_SUM_REALPART || type == REDUCTION_MEAN_REALPART) { 3072a873a8cdSSam Reynolds for (i = 0; i < n; i++) { 3073ad540459SPierre Jolivet for (j = 0; j < m; j++) reductions[i] += PetscRealPart(a[j]); 3074a873a8cdSSam Reynolds a += m; 3075a873a8cdSSam Reynolds } 3076857cbf51SRichard Tran Mills } else if (type == REDUCTION_SUM_IMAGINARYPART || type == REDUCTION_MEAN_IMAGINARYPART) { 3077857cbf51SRichard Tran Mills for (i = 0; i < n; i++) { 3078ad540459SPierre Jolivet for (j = 0; j < m; j++) reductions[i] += PetscImaginaryPart(a[j]); 3079857cbf51SRichard Tran Mills a += m; 3080857cbf51SRichard Tran Mills } 3081857cbf51SRichard Tran Mills } else SETERRQ(PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONG, "Unknown reduction type"); 30829566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &a)); 3083857cbf51SRichard Tran Mills if (type == NORM_2) { 3084a873a8cdSSam Reynolds for (i = 0; i < n; i++) reductions[i] = PetscSqrtReal(reductions[i]); 3085857cbf51SRichard Tran Mills } else if (type == REDUCTION_MEAN_REALPART || type == REDUCTION_MEAN_IMAGINARYPART) { 3086a873a8cdSSam Reynolds for (i = 0; i < n; i++) reductions[i] /= m; 30870716a85fSBarry Smith } 30880716a85fSBarry Smith PetscFunctionReturn(0); 30890716a85fSBarry Smith } 30900716a85fSBarry Smith 3091d71ae5a4SJacob Faibussowitsch PetscErrorCode MatSetRandom_SeqDense(Mat x, PetscRandom rctx) 3092d71ae5a4SJacob Faibussowitsch { 309373a71a0fSBarry Smith PetscScalar *a; 3094637a0070SStefano Zampini PetscInt lda, m, n, i, j; 309573a71a0fSBarry Smith 309673a71a0fSBarry Smith PetscFunctionBegin; 30979566063dSJacob Faibussowitsch PetscCall(MatGetSize(x, &m, &n)); 30989566063dSJacob Faibussowitsch PetscCall(MatDenseGetLDA(x, &lda)); 30993faff063SStefano Zampini PetscCall(MatDenseGetArrayWrite(x, &a)); 3100637a0070SStefano Zampini for (j = 0; j < n; j++) { 310148a46eb9SPierre Jolivet for (i = 0; i < m; i++) PetscCall(PetscRandomGetValue(rctx, a + j * lda + i)); 310273a71a0fSBarry Smith } 31033faff063SStefano Zampini PetscCall(MatDenseRestoreArrayWrite(x, &a)); 310473a71a0fSBarry Smith PetscFunctionReturn(0); 310573a71a0fSBarry Smith } 310673a71a0fSBarry Smith 3107d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatMissingDiagonal_SeqDense(Mat A, PetscBool *missing, PetscInt *d) 3108d71ae5a4SJacob Faibussowitsch { 31093b49f96aSBarry Smith PetscFunctionBegin; 31103b49f96aSBarry Smith *missing = PETSC_FALSE; 31113b49f96aSBarry Smith PetscFunctionReturn(0); 31123b49f96aSBarry Smith } 311373a71a0fSBarry Smith 3114ca15aa20SStefano Zampini /* vals is not const */ 3115d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatDenseGetColumn_SeqDense(Mat A, PetscInt col, PetscScalar **vals) 3116d71ae5a4SJacob Faibussowitsch { 311786aefd0dSHong Zhang Mat_SeqDense *a = (Mat_SeqDense *)A->data; 3118ca15aa20SStefano Zampini PetscScalar *v; 311986aefd0dSHong Zhang 312086aefd0dSHong Zhang PetscFunctionBegin; 312128b400f6SJacob Faibussowitsch PetscCheck(!A->factortype, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "Not for factored matrix"); 31229566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, &v)); 3123ca15aa20SStefano Zampini *vals = v + col * a->lda; 31249566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, &v)); 312586aefd0dSHong Zhang PetscFunctionReturn(0); 312686aefd0dSHong Zhang } 312786aefd0dSHong Zhang 3128d71ae5a4SJacob Faibussowitsch static PetscErrorCode MatDenseRestoreColumn_SeqDense(Mat A, PetscScalar **vals) 3129d71ae5a4SJacob Faibussowitsch { 313086aefd0dSHong Zhang PetscFunctionBegin; 3131742765d3SMatthew Knepley if (vals) *vals = NULL; /* user cannot accidentally use the array later */ 313286aefd0dSHong Zhang PetscFunctionReturn(0); 313386aefd0dSHong Zhang } 3134abc3b08eSStefano Zampini 3135289bc588SBarry Smith /* -------------------------------------------------------------------*/ 3136a5ae1ecdSBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_SeqDense, 3137905e6a2fSBarry Smith MatGetRow_SeqDense, 3138905e6a2fSBarry Smith MatRestoreRow_SeqDense, 3139905e6a2fSBarry Smith MatMult_SeqDense, 314097304618SKris Buschelman /* 4*/ MatMultAdd_SeqDense, 31417c922b88SBarry Smith MatMultTranspose_SeqDense, 31427c922b88SBarry Smith MatMultTransposeAdd_SeqDense, 3143f4259b30SLisandro Dalcin NULL, 3144f4259b30SLisandro Dalcin NULL, 3145f4259b30SLisandro Dalcin NULL, 3146f4259b30SLisandro Dalcin /* 10*/ NULL, 3147905e6a2fSBarry Smith MatLUFactor_SeqDense, 3148905e6a2fSBarry Smith MatCholeskyFactor_SeqDense, 314941f059aeSBarry Smith MatSOR_SeqDense, 3150ec8511deSBarry Smith MatTranspose_SeqDense, 315197304618SKris Buschelman /* 15*/ MatGetInfo_SeqDense, 3152905e6a2fSBarry Smith MatEqual_SeqDense, 3153905e6a2fSBarry Smith MatGetDiagonal_SeqDense, 3154905e6a2fSBarry Smith MatDiagonalScale_SeqDense, 3155905e6a2fSBarry Smith MatNorm_SeqDense, 3156c0aa2d19SHong Zhang /* 20*/ MatAssemblyBegin_SeqDense, 3157c0aa2d19SHong Zhang MatAssemblyEnd_SeqDense, 3158905e6a2fSBarry Smith MatSetOption_SeqDense, 3159905e6a2fSBarry Smith MatZeroEntries_SeqDense, 3160d519adbfSMatthew Knepley /* 24*/ MatZeroRows_SeqDense, 3161f4259b30SLisandro Dalcin NULL, 3162f4259b30SLisandro Dalcin NULL, 3163f4259b30SLisandro Dalcin NULL, 3164f4259b30SLisandro Dalcin NULL, 31654994cf47SJed Brown /* 29*/ MatSetUp_SeqDense, 3166f4259b30SLisandro Dalcin NULL, 3167f4259b30SLisandro Dalcin NULL, 3168f4259b30SLisandro Dalcin NULL, 3169f4259b30SLisandro Dalcin NULL, 3170d519adbfSMatthew Knepley /* 34*/ MatDuplicate_SeqDense, 3171f4259b30SLisandro Dalcin NULL, 3172f4259b30SLisandro Dalcin NULL, 3173f4259b30SLisandro Dalcin NULL, 3174f4259b30SLisandro Dalcin NULL, 3175d519adbfSMatthew Knepley /* 39*/ MatAXPY_SeqDense, 31767dae84e0SHong Zhang MatCreateSubMatrices_SeqDense, 3177f4259b30SLisandro Dalcin NULL, 31784b0e389bSBarry Smith MatGetValues_SeqDense, 3179a5ae1ecdSBarry Smith MatCopy_SeqDense, 3180d519adbfSMatthew Knepley /* 44*/ MatGetRowMax_SeqDense, 3181a5ae1ecdSBarry Smith MatScale_SeqDense, 31822f605a99SJose E. Roman MatShift_SeqDense, 3183f4259b30SLisandro Dalcin NULL, 31843f49a652SStefano Zampini MatZeroRowsColumns_SeqDense, 318573a71a0fSBarry Smith /* 49*/ MatSetRandom_SeqDense, 3186f4259b30SLisandro Dalcin NULL, 3187f4259b30SLisandro Dalcin NULL, 3188f4259b30SLisandro Dalcin NULL, 3189f4259b30SLisandro Dalcin NULL, 3190f4259b30SLisandro Dalcin /* 54*/ NULL, 3191f4259b30SLisandro Dalcin NULL, 3192f4259b30SLisandro Dalcin NULL, 3193f4259b30SLisandro Dalcin NULL, 3194f4259b30SLisandro Dalcin NULL, 3195023c16fcSToby Isaac /* 59*/ MatCreateSubMatrix_SeqDense, 3196e03a110bSBarry Smith MatDestroy_SeqDense, 3197e03a110bSBarry Smith MatView_SeqDense, 3198f4259b30SLisandro Dalcin NULL, 3199f4259b30SLisandro Dalcin NULL, 3200f4259b30SLisandro Dalcin /* 64*/ NULL, 3201f4259b30SLisandro Dalcin NULL, 3202f4259b30SLisandro Dalcin NULL, 3203f4259b30SLisandro Dalcin NULL, 3204f4259b30SLisandro Dalcin NULL, 3205d519adbfSMatthew Knepley /* 69*/ MatGetRowMaxAbs_SeqDense, 3206f4259b30SLisandro Dalcin NULL, 3207f4259b30SLisandro Dalcin NULL, 3208f4259b30SLisandro Dalcin NULL, 3209f4259b30SLisandro Dalcin NULL, 3210f4259b30SLisandro Dalcin /* 74*/ NULL, 3211f4259b30SLisandro Dalcin NULL, 3212f4259b30SLisandro Dalcin NULL, 3213f4259b30SLisandro Dalcin NULL, 3214f4259b30SLisandro Dalcin NULL, 3215f4259b30SLisandro Dalcin /* 79*/ NULL, 3216f4259b30SLisandro Dalcin NULL, 3217f4259b30SLisandro Dalcin NULL, 3218f4259b30SLisandro Dalcin NULL, 32195bba2384SShri Abhyankar /* 83*/ MatLoad_SeqDense, 3220637a0070SStefano Zampini MatIsSymmetric_SeqDense, 32211cbb95d3SBarry Smith MatIsHermitian_SeqDense, 3222f4259b30SLisandro Dalcin NULL, 3223f4259b30SLisandro Dalcin NULL, 3224f4259b30SLisandro Dalcin NULL, 3225f4259b30SLisandro Dalcin /* 89*/ NULL, 3226f4259b30SLisandro Dalcin NULL, 3227a9fe9ddaSSatish Balay MatMatMultNumeric_SeqDense_SeqDense, 3228f4259b30SLisandro Dalcin NULL, 3229f4259b30SLisandro Dalcin NULL, 3230f4259b30SLisandro Dalcin /* 94*/ NULL, 3231f4259b30SLisandro Dalcin NULL, 3232f4259b30SLisandro Dalcin NULL, 323369f65d41SStefano Zampini MatMatTransposeMultNumeric_SeqDense_SeqDense, 3234f4259b30SLisandro Dalcin NULL, 32354222ddf1SHong Zhang /* 99*/ MatProductSetFromOptions_SeqDense, 3236f4259b30SLisandro Dalcin NULL, 3237f4259b30SLisandro Dalcin NULL, 3238ba337c44SJed Brown MatConjugate_SeqDense, 3239f4259b30SLisandro Dalcin NULL, 3240f4259b30SLisandro Dalcin /*104*/ NULL, 3241ba337c44SJed Brown MatRealPart_SeqDense, 3242ba337c44SJed Brown MatImaginaryPart_SeqDense, 3243f4259b30SLisandro Dalcin NULL, 3244f4259b30SLisandro Dalcin NULL, 3245f4259b30SLisandro Dalcin /*109*/ NULL, 3246f4259b30SLisandro Dalcin NULL, 32478d0534beSBarry Smith MatGetRowMin_SeqDense, 3248aabbc4fbSShri Abhyankar MatGetColumnVector_SeqDense, 32493b49f96aSBarry Smith MatMissingDiagonal_SeqDense, 3250f4259b30SLisandro Dalcin /*114*/ NULL, 3251f4259b30SLisandro Dalcin NULL, 3252f4259b30SLisandro Dalcin NULL, 3253f4259b30SLisandro Dalcin NULL, 3254f4259b30SLisandro Dalcin NULL, 3255f4259b30SLisandro Dalcin /*119*/ NULL, 3256f4259b30SLisandro Dalcin NULL, 3257f4259b30SLisandro Dalcin NULL, 3258f4259b30SLisandro Dalcin NULL, 3259f4259b30SLisandro Dalcin NULL, 3260f4259b30SLisandro Dalcin /*124*/ NULL, 3261a873a8cdSSam Reynolds MatGetColumnReductions_SeqDense, 3262f4259b30SLisandro Dalcin NULL, 3263f4259b30SLisandro Dalcin NULL, 3264f4259b30SLisandro Dalcin NULL, 3265f4259b30SLisandro Dalcin /*129*/ NULL, 3266f4259b30SLisandro Dalcin NULL, 3267f4259b30SLisandro Dalcin NULL, 326875648e8dSHong Zhang MatTransposeMatMultNumeric_SeqDense_SeqDense, 3269f4259b30SLisandro Dalcin NULL, 3270f4259b30SLisandro Dalcin /*134*/ NULL, 3271f4259b30SLisandro Dalcin NULL, 3272f4259b30SLisandro Dalcin NULL, 3273f4259b30SLisandro Dalcin NULL, 3274f4259b30SLisandro Dalcin NULL, 3275f4259b30SLisandro Dalcin /*139*/ NULL, 3276f4259b30SLisandro Dalcin NULL, 3277f4259b30SLisandro Dalcin NULL, 3278f4259b30SLisandro Dalcin NULL, 3279f4259b30SLisandro Dalcin NULL, 32804222ddf1SHong Zhang MatCreateMPIMatConcatenateSeqMat_SeqDense, 3281f4259b30SLisandro Dalcin /*145*/ NULL, 3282f4259b30SLisandro Dalcin NULL, 328399a7f59eSMark Adams NULL, 328499a7f59eSMark Adams NULL, 32857fb60732SBarry Smith NULL, 3286dec0b466SHong Zhang /*150*/ NULL, 3287dec0b466SHong Zhang NULL}; 328890ace30eSBarry Smith 32894b828684SBarry Smith /*@C 329011a5261eSBarry Smith MatCreateSeqDense - Creates a `MATSEQDENSE` that 3291d65003e9SLois Curfman McInnes is stored in column major order (the usual Fortran 77 manner). Many 3292d65003e9SLois Curfman McInnes of the matrix operations use the BLAS and LAPACK routines. 3293289bc588SBarry Smith 3294d083f849SBarry Smith Collective 3295db81eaa0SLois Curfman McInnes 329620563c6bSBarry Smith Input Parameters: 329711a5261eSBarry Smith + comm - MPI communicator, set to `PETSC_COMM_SELF` 32980c775827SLois Curfman McInnes . m - number of rows 329918f449edSLois Curfman McInnes . n - number of columns 33000298fd71SBarry Smith - data - optional location of matrix data in column major order. Set data=NULL for PETSc 3301dfc5480cSLois Curfman McInnes to control all matrix memory allocation. 330220563c6bSBarry Smith 330320563c6bSBarry Smith Output Parameter: 330444cd7ae7SLois Curfman McInnes . A - the matrix 330520563c6bSBarry Smith 330611a5261eSBarry Smith Note: 330718f449edSLois Curfman McInnes The data input variable is intended primarily for Fortran programmers 330818f449edSLois Curfman McInnes who wish to allocate their own matrix memory space. Most users should 33090298fd71SBarry Smith set data=NULL. 331018f449edSLois Curfman McInnes 3311027ccd11SLois Curfman McInnes Level: intermediate 3312027ccd11SLois Curfman McInnes 331311a5261eSBarry Smith .seealso: `MATSEQDENSE`, `MatCreate()`, `MatCreateDense()`, `MatSetValues()` 331420563c6bSBarry Smith @*/ 3315d71ae5a4SJacob Faibussowitsch PetscErrorCode MatCreateSeqDense(MPI_Comm comm, PetscInt m, PetscInt n, PetscScalar *data, Mat *A) 3316d71ae5a4SJacob Faibussowitsch { 33173a40ed3dSBarry Smith PetscFunctionBegin; 33189566063dSJacob Faibussowitsch PetscCall(MatCreate(comm, A)); 33199566063dSJacob Faibussowitsch PetscCall(MatSetSizes(*A, m, n, m, n)); 33209566063dSJacob Faibussowitsch PetscCall(MatSetType(*A, MATSEQDENSE)); 33219566063dSJacob Faibussowitsch PetscCall(MatSeqDenseSetPreallocation(*A, data)); 3322273d9f13SBarry Smith PetscFunctionReturn(0); 3323273d9f13SBarry Smith } 3324273d9f13SBarry Smith 3325273d9f13SBarry Smith /*@C 332611a5261eSBarry Smith MatSeqDenseSetPreallocation - Sets the array used for storing the matrix elements of a `MATSEQDENSE` matrix 3327273d9f13SBarry Smith 3328d083f849SBarry Smith Collective 3329273d9f13SBarry Smith 3330273d9f13SBarry Smith Input Parameters: 33311c4f3114SJed Brown + B - the matrix 33320298fd71SBarry Smith - data - the array (or NULL) 3333273d9f13SBarry Smith 333411a5261eSBarry Smith Note: 3335273d9f13SBarry Smith The data input variable is intended primarily for Fortran programmers 3336273d9f13SBarry Smith who wish to allocate their own matrix memory space. Most users should 3337284134d9SBarry Smith need not call this routine. 3338273d9f13SBarry Smith 3339273d9f13SBarry Smith Level: intermediate 3340273d9f13SBarry Smith 334111a5261eSBarry Smith .seealso: `MATSEQDENSE`, `MatCreate()`, `MatCreateDense()`, `MatSetValues()`, `MatDenseSetLDA()` 3342273d9f13SBarry Smith @*/ 3343d71ae5a4SJacob Faibussowitsch PetscErrorCode MatSeqDenseSetPreallocation(Mat B, PetscScalar data[]) 3344d71ae5a4SJacob Faibussowitsch { 3345a23d5eceSKris Buschelman PetscFunctionBegin; 3346d5ea218eSStefano Zampini PetscValidHeaderSpecific(B, MAT_CLASSID, 1); 3347cac4c232SBarry Smith PetscTryMethod(B, "MatSeqDenseSetPreallocation_C", (Mat, PetscScalar[]), (B, data)); 3348a23d5eceSKris Buschelman PetscFunctionReturn(0); 3349a23d5eceSKris Buschelman } 3350a23d5eceSKris Buschelman 3351d71ae5a4SJacob Faibussowitsch PetscErrorCode MatSeqDenseSetPreallocation_SeqDense(Mat B, PetscScalar *data) 3352d71ae5a4SJacob Faibussowitsch { 3353ad16ce7aSStefano Zampini Mat_SeqDense *b = (Mat_SeqDense *)B->data; 3354273d9f13SBarry Smith 3355273d9f13SBarry Smith PetscFunctionBegin; 335628b400f6SJacob Faibussowitsch PetscCheck(!b->matinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreSubMatrix() first"); 3357273d9f13SBarry Smith B->preallocated = PETSC_TRUE; 3358a868139aSShri Abhyankar 33599566063dSJacob Faibussowitsch PetscCall(PetscLayoutSetUp(B->rmap)); 33609566063dSJacob Faibussowitsch PetscCall(PetscLayoutSetUp(B->cmap)); 336134ef9618SShri Abhyankar 3362ad16ce7aSStefano Zampini if (b->lda <= 0) b->lda = B->rmap->n; 336386d161a7SShri Abhyankar 33649e8f95c4SLisandro Dalcin if (!data) { /* petsc-allocated storage */ 33659566063dSJacob Faibussowitsch if (!b->user_alloc) PetscCall(PetscFree(b->v)); 33669566063dSJacob Faibussowitsch PetscCall(PetscCalloc1((size_t)b->lda * B->cmap->n, &b->v)); 33672205254eSKarl Rupp 33689e8f95c4SLisandro Dalcin b->user_alloc = PETSC_FALSE; 3369273d9f13SBarry Smith } else { /* user-allocated storage */ 33709566063dSJacob Faibussowitsch if (!b->user_alloc) PetscCall(PetscFree(b->v)); 3371273d9f13SBarry Smith b->v = data; 3372273d9f13SBarry Smith b->user_alloc = PETSC_TRUE; 3373273d9f13SBarry Smith } 33740450473dSBarry Smith B->assembled = PETSC_TRUE; 3375273d9f13SBarry Smith PetscFunctionReturn(0); 3376273d9f13SBarry Smith } 3377273d9f13SBarry Smith 337865b80a83SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 3379d71ae5a4SJacob Faibussowitsch PETSC_INTERN PetscErrorCode MatConvert_SeqDense_Elemental(Mat A, MatType newtype, MatReuse reuse, Mat *newmat) 3380d71ae5a4SJacob Faibussowitsch { 3381d77f618aSHong Zhang Mat mat_elemental; 33821683a169SBarry Smith const PetscScalar *array; 33831683a169SBarry Smith PetscScalar *v_colwise; 3384d77f618aSHong Zhang PetscInt M = A->rmap->N, N = A->cmap->N, i, j, k, *rows, *cols; 3385d77f618aSHong Zhang 33868baccfbdSHong Zhang PetscFunctionBegin; 33879566063dSJacob Faibussowitsch PetscCall(PetscMalloc3(M * N, &v_colwise, M, &rows, N, &cols)); 33889566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &array)); 3389d77f618aSHong Zhang /* convert column-wise array into row-wise v_colwise, see MatSetValues_Elemental() */ 3390d77f618aSHong Zhang k = 0; 3391d77f618aSHong Zhang for (j = 0; j < N; j++) { 3392d77f618aSHong Zhang cols[j] = j; 3393ad540459SPierre Jolivet for (i = 0; i < M; i++) v_colwise[j * M + i] = array[k++]; 3394d77f618aSHong Zhang } 3395ad540459SPierre Jolivet for (i = 0; i < M; i++) rows[i] = i; 33969566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &array)); 3397d77f618aSHong Zhang 33989566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A), &mat_elemental)); 33999566063dSJacob Faibussowitsch PetscCall(MatSetSizes(mat_elemental, PETSC_DECIDE, PETSC_DECIDE, M, N)); 34009566063dSJacob Faibussowitsch PetscCall(MatSetType(mat_elemental, MATELEMENTAL)); 34019566063dSJacob Faibussowitsch PetscCall(MatSetUp(mat_elemental)); 3402d77f618aSHong Zhang 3403d77f618aSHong Zhang /* PETSc-Elemental interaface uses axpy for setting off-processor entries, only ADD_VALUES is allowed */ 34049566063dSJacob Faibussowitsch PetscCall(MatSetValues(mat_elemental, M, rows, N, cols, v_colwise, ADD_VALUES)); 34059566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(mat_elemental, MAT_FINAL_ASSEMBLY)); 34069566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(mat_elemental, MAT_FINAL_ASSEMBLY)); 34079566063dSJacob Faibussowitsch PetscCall(PetscFree3(v_colwise, rows, cols)); 3408d77f618aSHong Zhang 3409511c6705SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 34109566063dSJacob Faibussowitsch PetscCall(MatHeaderReplace(A, &mat_elemental)); 3411d77f618aSHong Zhang } else { 3412d77f618aSHong Zhang *newmat = mat_elemental; 3413d77f618aSHong Zhang } 34148baccfbdSHong Zhang PetscFunctionReturn(0); 34158baccfbdSHong Zhang } 341665b80a83SHong Zhang #endif 34178baccfbdSHong Zhang 3418d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseSetLDA_SeqDense(Mat B, PetscInt lda) 3419d71ae5a4SJacob Faibussowitsch { 34201b807ce4Svictorle Mat_SeqDense *b = (Mat_SeqDense *)B->data; 34217422da62SJose E. Roman PetscBool data; 342221a2c019SBarry Smith 34231b807ce4Svictorle PetscFunctionBegin; 34247422da62SJose E. Roman data = (PetscBool)((B->rmap->n > 0 && B->cmap->n > 0) ? (b->v ? PETSC_TRUE : PETSC_FALSE) : PETSC_FALSE); 3425aed4548fSBarry Smith PetscCheck(b->user_alloc || !data || b->lda == lda, PETSC_COMM_SELF, PETSC_ERR_ORDER, "LDA cannot be changed after allocation of internal storage"); 342608401ef6SPierre Jolivet PetscCheck(lda >= B->rmap->n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "LDA %" PetscInt_FMT " must be at least matrix dimension %" PetscInt_FMT, lda, B->rmap->n); 34271b807ce4Svictorle b->lda = lda; 34281b807ce4Svictorle PetscFunctionReturn(0); 34291b807ce4Svictorle } 34301b807ce4Svictorle 3431d71ae5a4SJacob Faibussowitsch PetscErrorCode MatCreateMPIMatConcatenateSeqMat_SeqDense(MPI_Comm comm, Mat inmat, PetscInt n, MatReuse scall, Mat *outmat) 3432d71ae5a4SJacob Faibussowitsch { 3433d528f656SJakub Kruzik PetscFunctionBegin; 34349566063dSJacob Faibussowitsch PetscCall(MatCreateMPIMatConcatenateSeqMat_MPIDense(comm, inmat, n, scall, outmat)); 3435d528f656SJakub Kruzik PetscFunctionReturn(0); 3436d528f656SJakub Kruzik } 3437d528f656SJakub Kruzik 3438d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetColumnVec_SeqDense(Mat A, PetscInt col, Vec *v) 3439d71ae5a4SJacob Faibussowitsch { 34406947451fSStefano Zampini Mat_SeqDense *a = (Mat_SeqDense *)A->data; 34416947451fSStefano Zampini 34426947451fSStefano Zampini PetscFunctionBegin; 344328b400f6SJacob Faibussowitsch PetscCheck(!a->vecinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreColumnVec() first"); 344428b400f6SJacob Faibussowitsch PetscCheck(!a->matinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreSubMatrix() first"); 34454dfa11a4SJacob Faibussowitsch if (!a->cvec) { PetscCall(VecCreateSeqWithArray(PetscObjectComm((PetscObject)A), A->rmap->bs, A->rmap->n, NULL, &a->cvec)); } 34466947451fSStefano Zampini a->vecinuse = col + 1; 34479566063dSJacob Faibussowitsch PetscCall(MatDenseGetArray(A, (PetscScalar **)&a->ptrinuse)); 34489566063dSJacob Faibussowitsch PetscCall(VecPlaceArray(a->cvec, a->ptrinuse + (size_t)col * (size_t)a->lda)); 34496947451fSStefano Zampini *v = a->cvec; 34506947451fSStefano Zampini PetscFunctionReturn(0); 34516947451fSStefano Zampini } 34526947451fSStefano Zampini 3453d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreColumnVec_SeqDense(Mat A, PetscInt col, Vec *v) 3454d71ae5a4SJacob Faibussowitsch { 34556947451fSStefano Zampini Mat_SeqDense *a = (Mat_SeqDense *)A->data; 34566947451fSStefano Zampini 34576947451fSStefano Zampini PetscFunctionBegin; 345828b400f6SJacob Faibussowitsch PetscCheck(a->vecinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseGetColumnVec() first"); 345928b400f6SJacob Faibussowitsch PetscCheck(a->cvec, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Missing internal column vector"); 34606947451fSStefano Zampini a->vecinuse = 0; 34619566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArray(A, (PetscScalar **)&a->ptrinuse)); 34629566063dSJacob Faibussowitsch PetscCall(VecResetArray(a->cvec)); 346375f6d85dSStefano Zampini if (v) *v = NULL; 34646947451fSStefano Zampini PetscFunctionReturn(0); 34656947451fSStefano Zampini } 34666947451fSStefano Zampini 3467d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetColumnVecRead_SeqDense(Mat A, PetscInt col, Vec *v) 3468d71ae5a4SJacob Faibussowitsch { 34696947451fSStefano Zampini Mat_SeqDense *a = (Mat_SeqDense *)A->data; 34706947451fSStefano Zampini 34716947451fSStefano Zampini PetscFunctionBegin; 347228b400f6SJacob Faibussowitsch PetscCheck(!a->vecinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreColumnVec() first"); 347328b400f6SJacob Faibussowitsch PetscCheck(!a->matinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreSubMatrix() first"); 34744dfa11a4SJacob Faibussowitsch if (!a->cvec) { PetscCall(VecCreateSeqWithArray(PetscObjectComm((PetscObject)A), A->rmap->bs, A->rmap->n, NULL, &a->cvec)); } 34756947451fSStefano Zampini a->vecinuse = col + 1; 34769566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayRead(A, &a->ptrinuse)); 34779566063dSJacob Faibussowitsch PetscCall(VecPlaceArray(a->cvec, a->ptrinuse + (size_t)col * (size_t)a->lda)); 34789566063dSJacob Faibussowitsch PetscCall(VecLockReadPush(a->cvec)); 34796947451fSStefano Zampini *v = a->cvec; 34806947451fSStefano Zampini PetscFunctionReturn(0); 34816947451fSStefano Zampini } 34826947451fSStefano Zampini 3483d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreColumnVecRead_SeqDense(Mat A, PetscInt col, Vec *v) 3484d71ae5a4SJacob Faibussowitsch { 34856947451fSStefano Zampini Mat_SeqDense *a = (Mat_SeqDense *)A->data; 34866947451fSStefano Zampini 34876947451fSStefano Zampini PetscFunctionBegin; 348828b400f6SJacob Faibussowitsch PetscCheck(a->vecinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseGetColumnVec() first"); 348928b400f6SJacob Faibussowitsch PetscCheck(a->cvec, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Missing internal column vector"); 34906947451fSStefano Zampini a->vecinuse = 0; 34919566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayRead(A, &a->ptrinuse)); 34929566063dSJacob Faibussowitsch PetscCall(VecLockReadPop(a->cvec)); 34939566063dSJacob Faibussowitsch PetscCall(VecResetArray(a->cvec)); 349475f6d85dSStefano Zampini if (v) *v = NULL; 34956947451fSStefano Zampini PetscFunctionReturn(0); 34966947451fSStefano Zampini } 34976947451fSStefano Zampini 3498d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetColumnVecWrite_SeqDense(Mat A, PetscInt col, Vec *v) 3499d71ae5a4SJacob Faibussowitsch { 35006947451fSStefano Zampini Mat_SeqDense *a = (Mat_SeqDense *)A->data; 35016947451fSStefano Zampini 35026947451fSStefano Zampini PetscFunctionBegin; 350328b400f6SJacob Faibussowitsch PetscCheck(!a->vecinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreColumnVec() first"); 350428b400f6SJacob Faibussowitsch PetscCheck(!a->matinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreSubMatrix() first"); 35054dfa11a4SJacob Faibussowitsch if (!a->cvec) { PetscCall(VecCreateSeqWithArray(PetscObjectComm((PetscObject)A), A->rmap->bs, A->rmap->n, NULL, &a->cvec)); } 35066947451fSStefano Zampini a->vecinuse = col + 1; 35079566063dSJacob Faibussowitsch PetscCall(MatDenseGetArrayWrite(A, (PetscScalar **)&a->ptrinuse)); 35089566063dSJacob Faibussowitsch PetscCall(VecPlaceArray(a->cvec, a->ptrinuse + (size_t)col * (size_t)a->lda)); 35096947451fSStefano Zampini *v = a->cvec; 35106947451fSStefano Zampini PetscFunctionReturn(0); 35116947451fSStefano Zampini } 35126947451fSStefano Zampini 3513d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreColumnVecWrite_SeqDense(Mat A, PetscInt col, Vec *v) 3514d71ae5a4SJacob Faibussowitsch { 35156947451fSStefano Zampini Mat_SeqDense *a = (Mat_SeqDense *)A->data; 35166947451fSStefano Zampini 35176947451fSStefano Zampini PetscFunctionBegin; 351828b400f6SJacob Faibussowitsch PetscCheck(a->vecinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseGetColumnVec() first"); 351928b400f6SJacob Faibussowitsch PetscCheck(a->cvec, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Missing internal column vector"); 35206947451fSStefano Zampini a->vecinuse = 0; 35219566063dSJacob Faibussowitsch PetscCall(MatDenseRestoreArrayWrite(A, (PetscScalar **)&a->ptrinuse)); 35229566063dSJacob Faibussowitsch PetscCall(VecResetArray(a->cvec)); 352375f6d85dSStefano Zampini if (v) *v = NULL; 35246947451fSStefano Zampini PetscFunctionReturn(0); 35256947451fSStefano Zampini } 35266947451fSStefano Zampini 3527d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetSubMatrix_SeqDense(Mat A, PetscInt rbegin, PetscInt rend, PetscInt cbegin, PetscInt cend, Mat *v) 3528d71ae5a4SJacob Faibussowitsch { 35295ea7661aSPierre Jolivet Mat_SeqDense *a = (Mat_SeqDense *)A->data; 35305ea7661aSPierre Jolivet 35315ea7661aSPierre Jolivet PetscFunctionBegin; 353228b400f6SJacob Faibussowitsch PetscCheck(!a->vecinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreColumnVec() first"); 353328b400f6SJacob Faibussowitsch PetscCheck(!a->matinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseRestoreSubMatrix() first"); 3534a2748737SPierre Jolivet if (a->cmat && (cend - cbegin != a->cmat->cmap->N || rend - rbegin != a->cmat->rmap->N)) PetscCall(MatDestroy(&a->cmat)); 35355ea7661aSPierre Jolivet if (!a->cmat) { 3536a2748737SPierre Jolivet PetscCall(MatCreateDense(PetscObjectComm((PetscObject)A), rend - rbegin, PETSC_DECIDE, rend - rbegin, cend - cbegin, a->v + rbegin + (size_t)cbegin * a->lda, &a->cmat)); 35375ea7661aSPierre Jolivet } else { 3538a2748737SPierre Jolivet PetscCall(MatDensePlaceArray(a->cmat, a->v + rbegin + (size_t)cbegin * a->lda)); 35395ea7661aSPierre Jolivet } 35409566063dSJacob Faibussowitsch PetscCall(MatDenseSetLDA(a->cmat, a->lda)); 35415ea7661aSPierre Jolivet a->matinuse = cbegin + 1; 35425ea7661aSPierre Jolivet *v = a->cmat; 354347d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) || defined(PETSC_HAVE_HIP) 354475f6d85dSStefano Zampini A->offloadmask = PETSC_OFFLOAD_CPU; 354575f6d85dSStefano Zampini #endif 35465ea7661aSPierre Jolivet PetscFunctionReturn(0); 35475ea7661aSPierre Jolivet } 35485ea7661aSPierre Jolivet 3549d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreSubMatrix_SeqDense(Mat A, Mat *v) 3550d71ae5a4SJacob Faibussowitsch { 35515ea7661aSPierre Jolivet Mat_SeqDense *a = (Mat_SeqDense *)A->data; 35525ea7661aSPierre Jolivet 35535ea7661aSPierre Jolivet PetscFunctionBegin; 355428b400f6SJacob Faibussowitsch PetscCheck(a->matinuse, PETSC_COMM_SELF, PETSC_ERR_ORDER, "Need to call MatDenseGetSubMatrix() first"); 355528b400f6SJacob Faibussowitsch PetscCheck(a->cmat, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Missing internal column matrix"); 355608401ef6SPierre Jolivet PetscCheck(*v == a->cmat, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Not the matrix obtained from MatDenseGetSubMatrix()"); 35575ea7661aSPierre Jolivet a->matinuse = 0; 35589566063dSJacob Faibussowitsch PetscCall(MatDenseResetArray(a->cmat)); 3559742765d3SMatthew Knepley if (v) *v = NULL; 356047d993e7Ssuyashtn #if defined(PETSC_HAVE_CUDA) || defined(PETSC_HAVE_HIP) 35613faff063SStefano Zampini A->offloadmask = PETSC_OFFLOAD_CPU; 35623faff063SStefano Zampini #endif 35635ea7661aSPierre Jolivet PetscFunctionReturn(0); 35645ea7661aSPierre Jolivet } 35655ea7661aSPierre Jolivet 35660bad9183SKris Buschelman /*MC 3567fafad747SKris Buschelman MATSEQDENSE - MATSEQDENSE = "seqdense" - A matrix type to be used for sequential dense matrices. 35680bad9183SKris Buschelman 35690bad9183SKris Buschelman Options Database Keys: 357011a5261eSBarry Smith . -mat_type seqdense - sets the matrix type to `MATSEQDENSE` during a call to `MatSetFromOptions()` 35710bad9183SKris Buschelman 35720bad9183SKris Buschelman Level: beginner 35730bad9183SKris Buschelman 357411a5261eSBarry Smith .seealso: `MATSEQDENSE`, `MatCreateSeqDense()` 35750bad9183SKris Buschelman M*/ 3576d71ae5a4SJacob Faibussowitsch PetscErrorCode MatCreate_SeqDense(Mat B) 3577d71ae5a4SJacob Faibussowitsch { 3578273d9f13SBarry Smith Mat_SeqDense *b; 35797c334f02SBarry Smith PetscMPIInt size; 3580273d9f13SBarry Smith 3581273d9f13SBarry Smith PetscFunctionBegin; 35829566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)B), &size)); 358308401ef6SPierre Jolivet PetscCheck(size <= 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Comm must be of size 1"); 358455659b69SBarry Smith 35854dfa11a4SJacob Faibussowitsch PetscCall(PetscNew(&b)); 35869566063dSJacob Faibussowitsch PetscCall(PetscMemcpy(B->ops, &MatOps_Values, sizeof(struct _MatOps))); 358744cd7ae7SLois Curfman McInnes B->data = (void *)b; 358818f449edSLois Curfman McInnes 3589273d9f13SBarry Smith b->roworiented = PETSC_TRUE; 35904e220ebcSLois Curfman McInnes 35919566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatQRFactor_C", MatQRFactor_SeqDense)); 35929566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseGetLDA_C", MatDenseGetLDA_SeqDense)); 35939566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseSetLDA_C", MatDenseSetLDA_SeqDense)); 35949566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseGetArray_C", MatDenseGetArray_SeqDense)); 35959566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseRestoreArray_C", MatDenseRestoreArray_SeqDense)); 35969566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDensePlaceArray_C", MatDensePlaceArray_SeqDense)); 35979566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseResetArray_C", MatDenseResetArray_SeqDense)); 35989566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseReplaceArray_C", MatDenseReplaceArray_SeqDense)); 35999566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseGetArrayRead_C", MatDenseGetArray_SeqDense)); 36009566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseRestoreArrayRead_C", MatDenseRestoreArray_SeqDense)); 36019566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseGetArrayWrite_C", MatDenseGetArray_SeqDense)); 36029566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseRestoreArrayWrite_C", MatDenseRestoreArray_SeqDense)); 36039566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatConvert_seqdense_seqaij_C", MatConvert_SeqDense_SeqAIJ)); 36048baccfbdSHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 36059566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatConvert_seqdense_elemental_C", MatConvert_SeqDense_Elemental)); 36068baccfbdSHong Zhang #endif 3607d24d4204SJose E. Roman #if defined(PETSC_HAVE_SCALAPACK) 36089566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatConvert_seqdense_scalapack_C", MatConvert_Dense_ScaLAPACK)); 3609d24d4204SJose E. Roman #endif 36102bf066beSStefano Zampini #if defined(PETSC_HAVE_CUDA) 36119566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatConvert_seqdense_seqdensecuda_C", MatConvert_SeqDense_SeqDenseCUDA)); 36129566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatProductSetFromOptions_seqdensecuda_seqdensecuda_C", MatProductSetFromOptions_SeqDense)); 36139566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatProductSetFromOptions_seqdensecuda_seqdense_C", MatProductSetFromOptions_SeqDense)); 36149566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatProductSetFromOptions_seqdense_seqdensecuda_C", MatProductSetFromOptions_SeqDense)); 36152bf066beSStefano Zampini #endif 361647d993e7Ssuyashtn #if defined(PETSC_HAVE_HIP) 361747d993e7Ssuyashtn PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatConvert_seqdense_seqdensehip_C", MatConvert_SeqDense_SeqDenseHIP)); 361847d993e7Ssuyashtn PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatProductSetFromOptions_seqdensehip_seqdensehip_C", MatProductSetFromOptions_SeqDense)); 361947d993e7Ssuyashtn PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatProductSetFromOptions_seqdensehip_seqdense_C", MatProductSetFromOptions_SeqDense)); 362047d993e7Ssuyashtn PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatProductSetFromOptions_seqdense_seqdensehip_C", MatProductSetFromOptions_SeqDense)); 362147d993e7Ssuyashtn #endif 36229566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatSeqDenseSetPreallocation_C", MatSeqDenseSetPreallocation_SeqDense)); 36239566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatProductSetFromOptions_seqaij_seqdense_C", MatProductSetFromOptions_SeqAIJ_SeqDense)); 36249566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatProductSetFromOptions_seqdense_seqdense_C", MatProductSetFromOptions_SeqDense)); 36259566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatProductSetFromOptions_seqbaij_seqdense_C", MatProductSetFromOptions_SeqXBAIJ_SeqDense)); 36269566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatProductSetFromOptions_seqsbaij_seqdense_C", MatProductSetFromOptions_SeqXBAIJ_SeqDense)); 362796e6d5c4SRichard Tran Mills 36289566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseGetColumn_C", MatDenseGetColumn_SeqDense)); 36299566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseRestoreColumn_C", MatDenseRestoreColumn_SeqDense)); 36309566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseGetColumnVec_C", MatDenseGetColumnVec_SeqDense)); 36319566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseRestoreColumnVec_C", MatDenseRestoreColumnVec_SeqDense)); 36329566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseGetColumnVecRead_C", MatDenseGetColumnVecRead_SeqDense)); 36339566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseRestoreColumnVecRead_C", MatDenseRestoreColumnVecRead_SeqDense)); 36349566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseGetColumnVecWrite_C", MatDenseGetColumnVecWrite_SeqDense)); 36359566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseRestoreColumnVecWrite_C", MatDenseRestoreColumnVecWrite_SeqDense)); 36369566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseGetSubMatrix_C", MatDenseGetSubMatrix_SeqDense)); 36379566063dSJacob Faibussowitsch PetscCall(PetscObjectComposeFunction((PetscObject)B, "MatDenseRestoreSubMatrix_C", MatDenseRestoreSubMatrix_SeqDense)); 36389566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)B, MATSEQDENSE)); 36393a40ed3dSBarry Smith PetscFunctionReturn(0); 3640289bc588SBarry Smith } 364186aefd0dSHong Zhang 364286aefd0dSHong Zhang /*@C 364311a5261eSBarry Smith MatDenseGetColumn - gives access to a column of a dense matrix. This is only the local part of the column. You MUST call `MatDenseRestoreColumn()` to avoid memory bleeding. 364486aefd0dSHong Zhang 364586aefd0dSHong Zhang Not Collective 364686aefd0dSHong Zhang 36475ea7661aSPierre Jolivet Input Parameters: 364811a5261eSBarry Smith + mat - a `MATSEQDENSE` or `MATMPIDENSE` matrix 364986aefd0dSHong Zhang - col - column index 365086aefd0dSHong Zhang 365186aefd0dSHong Zhang Output Parameter: 365286aefd0dSHong Zhang . vals - pointer to the data 365386aefd0dSHong Zhang 365486aefd0dSHong Zhang Level: intermediate 365586aefd0dSHong Zhang 365611a5261eSBarry Smith Note: 365711a5261eSBarry Smith Use `MatDenseGetColumnVec()` to get access to a column of a `MATDENSE` treated as a `Vec` 365811a5261eSBarry Smith 365911a5261eSBarry Smith .seealso: `MATDENSE`, `MatDenseRestoreColumn()`, `MatDenseGetColumnVec()` 366086aefd0dSHong Zhang @*/ 3661d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetColumn(Mat A, PetscInt col, PetscScalar **vals) 3662d71ae5a4SJacob Faibussowitsch { 366386aefd0dSHong Zhang PetscFunctionBegin; 3664d5ea218eSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 3665d5ea218eSStefano Zampini PetscValidLogicalCollectiveInt(A, col, 2); 3666d5ea218eSStefano Zampini PetscValidPointer(vals, 3); 3667cac4c232SBarry Smith PetscUseMethod(A, "MatDenseGetColumn_C", (Mat, PetscInt, PetscScalar **), (A, col, vals)); 366886aefd0dSHong Zhang PetscFunctionReturn(0); 366986aefd0dSHong Zhang } 367086aefd0dSHong Zhang 367186aefd0dSHong Zhang /*@C 367211a5261eSBarry Smith MatDenseRestoreColumn - returns access to a column of a `MATDENSE` matrix which is returned by `MatDenseGetColumn()`. 367386aefd0dSHong Zhang 367486aefd0dSHong Zhang Not Collective 367586aefd0dSHong Zhang 3676742765d3SMatthew Knepley Input Parameters: 367711a5261eSBarry Smith + mat - a `MATSEQDENSE` or `MATMPIDENSE` matrix 3678742765d3SMatthew Knepley - vals - pointer to the data (may be NULL) 367986aefd0dSHong Zhang 368086aefd0dSHong Zhang Level: intermediate 368186aefd0dSHong Zhang 368211a5261eSBarry Smith .seealso: `MATDENSE`, `MatDenseGetColumn()` 368386aefd0dSHong Zhang @*/ 3684d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreColumn(Mat A, PetscScalar **vals) 3685d71ae5a4SJacob Faibussowitsch { 368686aefd0dSHong Zhang PetscFunctionBegin; 3687d5ea218eSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 3688d5ea218eSStefano Zampini PetscValidPointer(vals, 2); 3689cac4c232SBarry Smith PetscUseMethod(A, "MatDenseRestoreColumn_C", (Mat, PetscScalar **), (A, vals)); 369086aefd0dSHong Zhang PetscFunctionReturn(0); 369186aefd0dSHong Zhang } 36926947451fSStefano Zampini 36930f74d2c1SSatish Balay /*@ 369411a5261eSBarry Smith MatDenseGetColumnVec - Gives read-write access to a column of a `MATDENSE` matrix, represented as a `Vec`. 36956947451fSStefano Zampini 36966947451fSStefano Zampini Collective 36976947451fSStefano Zampini 36985ea7661aSPierre Jolivet Input Parameters: 369911a5261eSBarry Smith + mat - the `Mat` object 37006947451fSStefano Zampini - col - the column index 37016947451fSStefano Zampini 37026947451fSStefano Zampini Output Parameter: 37036947451fSStefano Zampini . v - the vector 37046947451fSStefano Zampini 37056947451fSStefano Zampini Notes: 370611a5261eSBarry Smith The vector is owned by PETSc. Users need to call `MatDenseRestoreColumnVec()` when the vector is no longer needed. 370711a5261eSBarry Smith 370811a5261eSBarry Smith Use `MatDenseGetColumnVecRead()` to obtain read-only access or `MatDenseGetColumnVecWrite()` for write-only access. 37096947451fSStefano Zampini 37106947451fSStefano Zampini Level: intermediate 37116947451fSStefano Zampini 371247d993e7Ssuyashtn .seealso: `MATDENSE`, `MATDENSECUDA`, `MATDENSEHIP`, `MatDenseGetColumnVecRead()`, `MatDenseGetColumnVecWrite()`, `MatDenseRestoreColumnVec()`, `MatDenseRestoreColumnVecRead()`, `MatDenseRestoreColumnVecWrite()`, `MatDenseGetColumn()` 37136947451fSStefano Zampini @*/ 3714d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetColumnVec(Mat A, PetscInt col, Vec *v) 3715d71ae5a4SJacob Faibussowitsch { 37166947451fSStefano Zampini PetscFunctionBegin; 37176947451fSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 37186947451fSStefano Zampini PetscValidType(A, 1); 37196947451fSStefano Zampini PetscValidLogicalCollectiveInt(A, col, 2); 37206947451fSStefano Zampini PetscValidPointer(v, 3); 372128b400f6SJacob Faibussowitsch PetscCheck(A->preallocated, PetscObjectComm((PetscObject)A), PETSC_ERR_ORDER, "Matrix not preallocated"); 37222cf15c64SPierre Jolivet PetscCheck(col >= 0 && col < A->cmap->N, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONG, "Invalid col %" PetscInt_FMT ", should be in [0,%" PetscInt_FMT ")", col, A->cmap->N); 3723cac4c232SBarry Smith PetscUseMethod(A, "MatDenseGetColumnVec_C", (Mat, PetscInt, Vec *), (A, col, v)); 37246947451fSStefano Zampini PetscFunctionReturn(0); 37256947451fSStefano Zampini } 37266947451fSStefano Zampini 37270f74d2c1SSatish Balay /*@ 37286947451fSStefano Zampini MatDenseRestoreColumnVec - Returns access to a column of a dense matrix obtained from MatDenseGetColumnVec(). 37296947451fSStefano Zampini 37306947451fSStefano Zampini Collective 37316947451fSStefano Zampini 37325ea7661aSPierre Jolivet Input Parameters: 37336947451fSStefano Zampini + mat - the Mat object 37346947451fSStefano Zampini . col - the column index 3735742765d3SMatthew Knepley - v - the Vec object (may be NULL) 37366947451fSStefano Zampini 37376947451fSStefano Zampini Level: intermediate 37386947451fSStefano Zampini 373947d993e7Ssuyashtn .seealso: `MATDENSE`, `MATDENSECUDA`, `MATDENSEHIP`, `MatDenseGetColumnVec()`, `MatDenseGetColumnVecRead()`, `MatDenseGetColumnVecWrite()`, `MatDenseRestoreColumnVecRead()`, `MatDenseRestoreColumnVecWrite()` 37406947451fSStefano Zampini @*/ 3741d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreColumnVec(Mat A, PetscInt col, Vec *v) 3742d71ae5a4SJacob Faibussowitsch { 37436947451fSStefano Zampini PetscFunctionBegin; 37446947451fSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 37456947451fSStefano Zampini PetscValidType(A, 1); 37466947451fSStefano Zampini PetscValidLogicalCollectiveInt(A, col, 2); 374708401ef6SPierre Jolivet PetscCheck(A->preallocated, PetscObjectComm((PetscObject)A), PETSC_ERR_ORDER, "Matrix not preallocated"); 37482cf15c64SPierre Jolivet PetscCheck(col >= 0 && col < A->cmap->N, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONG, "Invalid col %" PetscInt_FMT ", should be in [0,%" PetscInt_FMT ")", col, A->cmap->N); 3749cac4c232SBarry Smith PetscUseMethod(A, "MatDenseRestoreColumnVec_C", (Mat, PetscInt, Vec *), (A, col, v)); 37506947451fSStefano Zampini PetscFunctionReturn(0); 37516947451fSStefano Zampini } 37526947451fSStefano Zampini 37530f74d2c1SSatish Balay /*@ 37546947451fSStefano Zampini MatDenseGetColumnVecRead - Gives read-only access to a column of a dense matrix, represented as a Vec. 37556947451fSStefano Zampini 37566947451fSStefano Zampini Collective 37576947451fSStefano Zampini 37585ea7661aSPierre Jolivet Input Parameters: 37596947451fSStefano Zampini + mat - the Mat object 37606947451fSStefano Zampini - col - the column index 37616947451fSStefano Zampini 37626947451fSStefano Zampini Output Parameter: 37636947451fSStefano Zampini . v - the vector 37646947451fSStefano Zampini 37656947451fSStefano Zampini Notes: 37666947451fSStefano Zampini The vector is owned by PETSc and users cannot modify it. 376711a5261eSBarry Smith 37686947451fSStefano Zampini Users need to call MatDenseRestoreColumnVecRead() when the vector is no longer needed. 376911a5261eSBarry Smith 37706947451fSStefano Zampini Use MatDenseGetColumnVec() to obtain read-write access or MatDenseGetColumnVecWrite() for write-only access. 37716947451fSStefano Zampini 37726947451fSStefano Zampini Level: intermediate 37736947451fSStefano Zampini 377447d993e7Ssuyashtn .seealso: `MATDENSE`, `MATDENSECUDA`, `MATDENSEHIP`, `MatDenseGetColumnVec()`, `MatDenseGetColumnVecWrite()`, `MatDenseRestoreColumnVec()`, `MatDenseRestoreColumnVecRead()`, `MatDenseRestoreColumnVecWrite()` 37756947451fSStefano Zampini @*/ 3776d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetColumnVecRead(Mat A, PetscInt col, Vec *v) 3777d71ae5a4SJacob Faibussowitsch { 37786947451fSStefano Zampini PetscFunctionBegin; 37796947451fSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 37806947451fSStefano Zampini PetscValidType(A, 1); 37816947451fSStefano Zampini PetscValidLogicalCollectiveInt(A, col, 2); 37826947451fSStefano Zampini PetscValidPointer(v, 3); 378328b400f6SJacob Faibussowitsch PetscCheck(A->preallocated, PetscObjectComm((PetscObject)A), PETSC_ERR_ORDER, "Matrix not preallocated"); 37842cf15c64SPierre Jolivet PetscCheck(col >= 0 && col < A->cmap->N, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONG, "Invalid col %" PetscInt_FMT ", should be in [0,%" PetscInt_FMT ")", col, A->cmap->N); 3785cac4c232SBarry Smith PetscUseMethod(A, "MatDenseGetColumnVecRead_C", (Mat, PetscInt, Vec *), (A, col, v)); 37866947451fSStefano Zampini PetscFunctionReturn(0); 37876947451fSStefano Zampini } 37886947451fSStefano Zampini 37890f74d2c1SSatish Balay /*@ 37906947451fSStefano Zampini MatDenseRestoreColumnVecRead - Returns access to a column of a dense matrix obtained from MatDenseGetColumnVecRead(). 37916947451fSStefano Zampini 37926947451fSStefano Zampini Collective 37936947451fSStefano Zampini 37945ea7661aSPierre Jolivet Input Parameters: 37956947451fSStefano Zampini + mat - the Mat object 37966947451fSStefano Zampini . col - the column index 3797742765d3SMatthew Knepley - v - the Vec object (may be NULL) 37986947451fSStefano Zampini 37996947451fSStefano Zampini Level: intermediate 38006947451fSStefano Zampini 380147d993e7Ssuyashtn .seealso: `MATDENSE`, `MATDENSECUDA`, `MATDENSEHIP`, `MatDenseGetColumnVec()`, `MatDenseGetColumnVecRead()`, `MatDenseGetColumnVecWrite()`, `MatDenseRestoreColumnVec()`, `MatDenseRestoreColumnVecWrite()` 38026947451fSStefano Zampini @*/ 3803d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreColumnVecRead(Mat A, PetscInt col, Vec *v) 3804d71ae5a4SJacob Faibussowitsch { 38056947451fSStefano Zampini PetscFunctionBegin; 38066947451fSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 38076947451fSStefano Zampini PetscValidType(A, 1); 38086947451fSStefano Zampini PetscValidLogicalCollectiveInt(A, col, 2); 380908401ef6SPierre Jolivet PetscCheck(A->preallocated, PetscObjectComm((PetscObject)A), PETSC_ERR_ORDER, "Matrix not preallocated"); 38102cf15c64SPierre Jolivet PetscCheck(col >= 0 && col < A->cmap->N, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONG, "Invalid col %" PetscInt_FMT ", should be in [0,%" PetscInt_FMT ")", col, A->cmap->N); 3811cac4c232SBarry Smith PetscUseMethod(A, "MatDenseRestoreColumnVecRead_C", (Mat, PetscInt, Vec *), (A, col, v)); 38126947451fSStefano Zampini PetscFunctionReturn(0); 38136947451fSStefano Zampini } 38146947451fSStefano Zampini 38150f74d2c1SSatish Balay /*@ 38166947451fSStefano Zampini MatDenseGetColumnVecWrite - Gives write-only access to a column of a dense matrix, represented as a Vec. 38176947451fSStefano Zampini 38186947451fSStefano Zampini Collective 38196947451fSStefano Zampini 38205ea7661aSPierre Jolivet Input Parameters: 38216947451fSStefano Zampini + mat - the Mat object 38226947451fSStefano Zampini - col - the column index 38236947451fSStefano Zampini 38246947451fSStefano Zampini Output Parameter: 38256947451fSStefano Zampini . v - the vector 38266947451fSStefano Zampini 38276947451fSStefano Zampini Notes: 38286947451fSStefano Zampini The vector is owned by PETSc. Users need to call MatDenseRestoreColumnVecWrite() when the vector is no longer needed. 382911a5261eSBarry Smith 38306947451fSStefano Zampini Use MatDenseGetColumnVec() to obtain read-write access or MatDenseGetColumnVecRead() for read-only access. 38316947451fSStefano Zampini 38326947451fSStefano Zampini Level: intermediate 38336947451fSStefano Zampini 383447d993e7Ssuyashtn .seealso: `MATDENSE`, `MATDENSECUDA`, `MATDENSEHIP`, `MatDenseGetColumnVec()`, `MatDenseGetColumnVecRead()`, `MatDenseRestoreColumnVec()`, `MatDenseRestoreColumnVecRead()`, `MatDenseRestoreColumnVecWrite()` 38356947451fSStefano Zampini @*/ 3836d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetColumnVecWrite(Mat A, PetscInt col, Vec *v) 3837d71ae5a4SJacob Faibussowitsch { 38386947451fSStefano Zampini PetscFunctionBegin; 38396947451fSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 38406947451fSStefano Zampini PetscValidType(A, 1); 38416947451fSStefano Zampini PetscValidLogicalCollectiveInt(A, col, 2); 38426947451fSStefano Zampini PetscValidPointer(v, 3); 384328b400f6SJacob Faibussowitsch PetscCheck(A->preallocated, PetscObjectComm((PetscObject)A), PETSC_ERR_ORDER, "Matrix not preallocated"); 3844aed4548fSBarry Smith PetscCheck(col >= 0 && col < A->cmap->N, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONG, "Invalid col %" PetscInt_FMT ", should be in [0,%" PetscInt_FMT ")", col, A->cmap->N); 3845cac4c232SBarry Smith PetscUseMethod(A, "MatDenseGetColumnVecWrite_C", (Mat, PetscInt, Vec *), (A, col, v)); 38466947451fSStefano Zampini PetscFunctionReturn(0); 38476947451fSStefano Zampini } 38486947451fSStefano Zampini 38490f74d2c1SSatish Balay /*@ 38506947451fSStefano Zampini MatDenseRestoreColumnVecWrite - Returns access to a column of a dense matrix obtained from MatDenseGetColumnVecWrite(). 38516947451fSStefano Zampini 38526947451fSStefano Zampini Collective 38536947451fSStefano Zampini 38545ea7661aSPierre Jolivet Input Parameters: 38556947451fSStefano Zampini + mat - the Mat object 38566947451fSStefano Zampini . col - the column index 3857742765d3SMatthew Knepley - v - the Vec object (may be NULL) 38586947451fSStefano Zampini 38596947451fSStefano Zampini Level: intermediate 38606947451fSStefano Zampini 386147d993e7Ssuyashtn .seealso: `MATDENSE`, `MATDENSECUDA`, `MATDENSEHIP`, `MatDenseGetColumnVec()`, `MatDenseGetColumnVecRead()`, `MatDenseGetColumnVecWrite()`, `MatDenseRestoreColumnVec()`, `MatDenseRestoreColumnVecRead()` 38626947451fSStefano Zampini @*/ 3863d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreColumnVecWrite(Mat A, PetscInt col, Vec *v) 3864d71ae5a4SJacob Faibussowitsch { 38656947451fSStefano Zampini PetscFunctionBegin; 38666947451fSStefano Zampini PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 38676947451fSStefano Zampini PetscValidType(A, 1); 38686947451fSStefano Zampini PetscValidLogicalCollectiveInt(A, col, 2); 386908401ef6SPierre Jolivet PetscCheck(A->preallocated, PetscObjectComm((PetscObject)A), PETSC_ERR_ORDER, "Matrix not preallocated"); 3870aed4548fSBarry Smith PetscCheck(col >= 0 && col < A->cmap->N, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONG, "Invalid col %" PetscInt_FMT ", should be in [0,%" PetscInt_FMT ")", col, A->cmap->N); 3871cac4c232SBarry Smith PetscUseMethod(A, "MatDenseRestoreColumnVecWrite_C", (Mat, PetscInt, Vec *), (A, col, v)); 38726947451fSStefano Zampini PetscFunctionReturn(0); 38736947451fSStefano Zampini } 38745ea7661aSPierre Jolivet 38750f74d2c1SSatish Balay /*@ 3876a2748737SPierre Jolivet MatDenseGetSubMatrix - Gives access to a block of rows and columns of a dense matrix, represented as a Mat. 38775ea7661aSPierre Jolivet 38785ea7661aSPierre Jolivet Collective 38795ea7661aSPierre Jolivet 38805ea7661aSPierre Jolivet Input Parameters: 38815ea7661aSPierre Jolivet + mat - the Mat object 3882a2748737SPierre Jolivet . rbegin - the first global row index in the block (if PETSC_DECIDE, is 0) 3883a2748737SPierre Jolivet . rend - the global row index past the last one in the block (if PETSC_DECIDE, is M) 3884a2748737SPierre Jolivet . cbegin - the first global column index in the block (if PETSC_DECIDE, is 0) 3885a2748737SPierre Jolivet - cend - the global column index past the last one in the block (if PETSC_DECIDE, is N) 38865ea7661aSPierre Jolivet 38875ea7661aSPierre Jolivet Output Parameter: 38885ea7661aSPierre Jolivet . v - the matrix 38895ea7661aSPierre Jolivet 38905ea7661aSPierre Jolivet Notes: 38915ea7661aSPierre Jolivet The matrix is owned by PETSc. Users need to call MatDenseRestoreSubMatrix() when the matrix is no longer needed. 389211a5261eSBarry Smith 3893a2748737SPierre Jolivet The output matrix is not redistributed by PETSc, so depending on the values of rbegin and rend, some processes may have no local rows. 38945ea7661aSPierre Jolivet 38955ea7661aSPierre Jolivet Level: intermediate 38965ea7661aSPierre Jolivet 389747d993e7Ssuyashtn .seealso: `MATDENSE`, `MATDENSECUDA`, `MATDENSEHIP`, `MatDenseGetColumnVec()`, `MatDenseRestoreColumnVec()`, `MatDenseRestoreSubMatrix()` 38985ea7661aSPierre Jolivet @*/ 3899d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseGetSubMatrix(Mat A, PetscInt rbegin, PetscInt rend, PetscInt cbegin, PetscInt cend, Mat *v) 3900d71ae5a4SJacob Faibussowitsch { 39015ea7661aSPierre Jolivet PetscFunctionBegin; 39025ea7661aSPierre Jolivet PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 39035ea7661aSPierre Jolivet PetscValidType(A, 1); 3904a2748737SPierre Jolivet PetscValidLogicalCollectiveInt(A, rbegin, 2); 3905a2748737SPierre Jolivet PetscValidLogicalCollectiveInt(A, rend, 3); 3906a2748737SPierre Jolivet PetscValidLogicalCollectiveInt(A, cbegin, 4); 3907a2748737SPierre Jolivet PetscValidLogicalCollectiveInt(A, cend, 5); 3908a2748737SPierre Jolivet PetscValidPointer(v, 6); 3909a2748737SPierre Jolivet if (rbegin == PETSC_DECIDE) rbegin = 0; 3910a2748737SPierre Jolivet if (rend == PETSC_DECIDE) rend = A->rmap->N; 3911a2748737SPierre Jolivet if (cbegin == PETSC_DECIDE) cbegin = 0; 3912a2748737SPierre Jolivet if (cend == PETSC_DECIDE) cend = A->cmap->N; 391328b400f6SJacob Faibussowitsch PetscCheck(A->preallocated, PetscObjectComm((PetscObject)A), PETSC_ERR_ORDER, "Matrix not preallocated"); 3914a2748737SPierre Jolivet PetscCheck(rbegin >= 0 && rbegin <= A->rmap->N, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONG, "Invalid rbegin %" PetscInt_FMT ", should be in [0,%" PetscInt_FMT "]", rbegin, A->rmap->N); 3915a2748737SPierre Jolivet PetscCheck(rend >= rbegin && rend <= A->rmap->N, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONG, "Invalid rend %" PetscInt_FMT ", should be in [%" PetscInt_FMT ",%" PetscInt_FMT "]", rend, rbegin, A->rmap->N); 3916a2748737SPierre Jolivet PetscCheck(cbegin >= 0 && cbegin <= A->cmap->N, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONG, "Invalid cbegin %" PetscInt_FMT ", should be in [0,%" PetscInt_FMT "]", cbegin, A->cmap->N); 3917a2748737SPierre Jolivet PetscCheck(cend >= cbegin && cend <= A->cmap->N, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONG, "Invalid cend %" PetscInt_FMT ", should be in [%" PetscInt_FMT ",%" PetscInt_FMT "]", cend, cbegin, A->cmap->N); 3918a2748737SPierre Jolivet PetscUseMethod(A, "MatDenseGetSubMatrix_C", (Mat, PetscInt, PetscInt, PetscInt, PetscInt, Mat *), (A, rbegin, rend, cbegin, cend, v)); 39195ea7661aSPierre Jolivet PetscFunctionReturn(0); 39205ea7661aSPierre Jolivet } 39215ea7661aSPierre Jolivet 39220f74d2c1SSatish Balay /*@ 39235ea7661aSPierre Jolivet MatDenseRestoreSubMatrix - Returns access to a block of columns of a dense matrix obtained from MatDenseGetSubMatrix(). 39245ea7661aSPierre Jolivet 39255ea7661aSPierre Jolivet Collective 39265ea7661aSPierre Jolivet 39275ea7661aSPierre Jolivet Input Parameters: 39285ea7661aSPierre Jolivet + mat - the Mat object 3929742765d3SMatthew Knepley - v - the Mat object (may be NULL) 39305ea7661aSPierre Jolivet 39315ea7661aSPierre Jolivet Level: intermediate 39325ea7661aSPierre Jolivet 393347d993e7Ssuyashtn .seealso: `MATDENSE`, `MATDENSECUDA`, `MATDENSEHIP`, `MatDenseGetColumnVec()`, `MatDenseRestoreColumnVec()`, `MatDenseGetSubMatrix()` 39345ea7661aSPierre Jolivet @*/ 3935d71ae5a4SJacob Faibussowitsch PetscErrorCode MatDenseRestoreSubMatrix(Mat A, Mat *v) 3936d71ae5a4SJacob Faibussowitsch { 39375ea7661aSPierre Jolivet PetscFunctionBegin; 39385ea7661aSPierre Jolivet PetscValidHeaderSpecific(A, MAT_CLASSID, 1); 39395ea7661aSPierre Jolivet PetscValidType(A, 1); 39405ea7661aSPierre Jolivet PetscValidPointer(v, 2); 3941cac4c232SBarry Smith PetscUseMethod(A, "MatDenseRestoreSubMatrix_C", (Mat, Mat *), (A, v)); 39425ea7661aSPierre Jolivet PetscFunctionReturn(0); 39435ea7661aSPierre Jolivet } 39448a9c020eSBarry Smith 39458a9c020eSBarry Smith #include <petscblaslapack.h> 39468a9c020eSBarry Smith #include <petsc/private/kernels/blockinvert.h> 39478a9c020eSBarry Smith 3948d71ae5a4SJacob Faibussowitsch PetscErrorCode MatSeqDenseInvert(Mat A) 3949d71ae5a4SJacob Faibussowitsch { 39508a9c020eSBarry Smith Mat_SeqDense *a = (Mat_SeqDense *)A->data; 39518a9c020eSBarry Smith PetscInt bs = A->rmap->n; 39528a9c020eSBarry Smith MatScalar *values = a->v; 39538a9c020eSBarry Smith const PetscReal shift = 0.0; 39548a9c020eSBarry Smith PetscBool allowzeropivot = PetscNot(A->erroriffailure), zeropivotdetected = PETSC_FALSE; 39558a9c020eSBarry Smith 39568a9c020eSBarry Smith PetscFunctionBegin; 39578a9c020eSBarry Smith /* factor and invert each block */ 39588a9c020eSBarry Smith switch (bs) { 3959d71ae5a4SJacob Faibussowitsch case 1: 3960d71ae5a4SJacob Faibussowitsch values[0] = (PetscScalar)1.0 / (values[0] + shift); 3961d71ae5a4SJacob Faibussowitsch break; 39628a9c020eSBarry Smith case 2: 39638a9c020eSBarry Smith PetscCall(PetscKernel_A_gets_inverse_A_2(values, shift, allowzeropivot, &zeropivotdetected)); 39648a9c020eSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 39658a9c020eSBarry Smith break; 39668a9c020eSBarry Smith case 3: 39678a9c020eSBarry Smith PetscCall(PetscKernel_A_gets_inverse_A_3(values, shift, allowzeropivot, &zeropivotdetected)); 39688a9c020eSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 39698a9c020eSBarry Smith break; 39708a9c020eSBarry Smith case 4: 39718a9c020eSBarry Smith PetscCall(PetscKernel_A_gets_inverse_A_4(values, shift, allowzeropivot, &zeropivotdetected)); 39728a9c020eSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 39738a9c020eSBarry Smith break; 39749371c9d4SSatish Balay case 5: { 39758a9c020eSBarry Smith PetscScalar work[25]; 39768a9c020eSBarry Smith PetscInt ipvt[5]; 39778a9c020eSBarry Smith 39788a9c020eSBarry Smith PetscCall(PetscKernel_A_gets_inverse_A_5(values, ipvt, work, shift, allowzeropivot, &zeropivotdetected)); 39798a9c020eSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 39809371c9d4SSatish Balay } break; 39818a9c020eSBarry Smith case 6: 39828a9c020eSBarry Smith PetscCall(PetscKernel_A_gets_inverse_A_6(values, shift, allowzeropivot, &zeropivotdetected)); 39838a9c020eSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 39848a9c020eSBarry Smith break; 39858a9c020eSBarry Smith case 7: 39868a9c020eSBarry Smith PetscCall(PetscKernel_A_gets_inverse_A_7(values, shift, allowzeropivot, &zeropivotdetected)); 39878a9c020eSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 39888a9c020eSBarry Smith break; 39899371c9d4SSatish Balay default: { 39908a9c020eSBarry Smith PetscInt *v_pivots, *IJ, j; 39918a9c020eSBarry Smith PetscScalar *v_work; 39928a9c020eSBarry Smith 39938a9c020eSBarry Smith PetscCall(PetscMalloc3(bs, &v_work, bs, &v_pivots, bs, &IJ)); 3994ad540459SPierre Jolivet for (j = 0; j < bs; j++) IJ[j] = j; 39958a9c020eSBarry Smith PetscCall(PetscKernel_A_gets_inverse_A(bs, values, v_pivots, v_work, allowzeropivot, &zeropivotdetected)); 39968a9c020eSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 39978a9c020eSBarry Smith PetscCall(PetscFree3(v_work, v_pivots, IJ)); 39988a9c020eSBarry Smith } 39998a9c020eSBarry Smith } 40008a9c020eSBarry Smith PetscFunctionReturn(0); 40018a9c020eSBarry Smith } 4002