| /petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/ |
| H A D | blas_cyclic.c | 250 PetscBLASInt b_one = 1, blda, bm; in TRSVCyclic_Private() local
255 PetscCall(PetscBLASIntCast(lda, &blda)); in TRSVCyclic_Private()
264 …PetscCallBLAS("BLAStrsv", BLAStrsv_("U", trans, "N", &bn, &A[i_oldest * (lda + 1)], &blda, &y[i_ol… in TRSVCyclic_Private()
272 … if (i_next > 0) PetscCallBLAS("BLAStrsv", BLAStrsv_("U", "N", "N", &i_next, A, &blda, y, &b_one)); in TRSVCyclic_Private()
273 …lBLAS("BLASgemv", BLASgemv_("N", &bn, &i_next, &minus_one, &A[i_oldest], &blda, y, &b_one, &one, &… in TRSVCyclic_Private()
274 …BLAS("BLAStrsv", BLAStrsv_("U", "N", "N", &bn, &A[i_oldest * (lda + 1)], &blda, &y[i_oldest], &b_o… in TRSVCyclic_Private()
276 …BLAS("BLAStrsv", BLAStrsv_("U", "C", "N", &bn, &A[i_oldest * (lda + 1)], &blda, &y[i_oldest], &b_o… in TRSVCyclic_Private()
277 …lBLAS("BLASgemv", BLASgemv_("C", &bn, &i_next, &minus_one, &A[i_oldest], &blda, &y[i_oldest], &b_o… in TRSVCyclic_Private()
278 … if (i_next > 0) PetscCallBLAS("BLAStrsv", BLAStrsv_("U", "C", "N", &i_next, A, &blda, y, &b_one)); in TRSVCyclic_Private()
345 PetscBLASInt b_one = 1, blda, bm; in HEMVCyclic_Private() local
[all …]
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | mattransposematmult.c | 64 PetscInt i, j, m = A->rmap->n, n = A->cmap->n, blda, clda; in MatTMatTMultNumeric_SeqAIJ_SeqDense() local
80 PetscCall(MatDenseGetLDA(B, &blda)); in MatTMatTMultNumeric_SeqAIJ_SeqDense()
89 for (i = 0; i < m; i++) btarray[i * mdof + j] = Barray[j * blda + i]; in MatTMatTMultNumeric_SeqAIJ_SeqDense()
106 if (blda == B->rmap->n) { in MatTMatTMultNumeric_SeqAIJ_SeqDense()
114 for (i = 0; i < bm; i++) ctarray[j * bm + i] = Barray[j * blda + i]; in MatTMatTMultNumeric_SeqAIJ_SeqDense()
120 if (blda == B->rmap->n) PetscCall(VecResetArray(ct)); in MatTMatTMultNumeric_SeqAIJ_SeqDense()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpimatmatmult.c | 395 PetscInt blda; member
416 PetscInt nz = aij->B->cmap->n, blda, m, M, n, N; in MatMatMultSymbolic_MPIAIJ_MPIDense() local
437 PetscCall(MatDenseGetLDA(B, &blda)); in MatMatMultSymbolic_MPIAIJ_MPIDense()
486 contents->blda = blda; in MatMatMultSymbolic_MPIAIJ_MPIDense()
493 PetscCallMPI(MPI_Type_create_resized(type1, 0, blda * sizeof(PetscScalar), &stype[i])); in MatMatMultSymbolic_MPIAIJ_MPIDense()
551 PetscInt blda; in MatMPIDenseScatter() local
570 PetscCall(MatDenseGetLDA(B, &blda)); in MatMPIDenseScatter()
571 …blda == contents->blda, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cannot reuse an input matrix with l… in MatMPIDenseScatter()
|
| /petsc/src/mat/impls/h2opus/cuda/ |
| H A D | math2opus.cu | 343 PetscInt blda, clda; in MatMultNKernel_H2OPUS() local
353 PetscCall(MatDenseGetLDA(B, &blda)); in MatMultNKernel_H2OPUS()
363 blda = n; in MatMultNKernel_H2OPUS()
383 … H2Opus_Trans : H2Opus_NoTrans, */ h2opus->s, *h2opus->dist_hmatrix, uxx, blda, 0.0, uyy, clda, B-… in MatMultNKernel_H2OPUS()
387 …hgemv(transA ? H2Opus_Trans : H2Opus_NoTrans, h2opus->s, *h2opus->hmatrix, uxx, blda, 0.0, uyy, cl… in MatMultNKernel_H2OPUS()
423 …pus_Trans : H2Opus_NoTrans, */ h2opus->s, *h2opus->dist_hmatrix_gpu, uxx, blda, 0.0, uyy, clda, B-… in MatMultNKernel_H2OPUS()
427 …hgemv(transA ? H2Opus_Trans : H2Opus_NoTrans, h2opus->s, *h2opus->hmatrix_gpu, uxx, blda, 0.0, uyy… in MatMultNKernel_H2OPUS()
|
| /petsc/src/mat/impls/dense/mpi/ |
| H A D | mpidense.c | 2273 PetscBLASInt cm, cn, ck, alda, blda = 0, clda; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic() local
2290 PetscCall(PetscBLASIntCast(i, &blda)); in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2295 if (blda == bn) { in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2300 for (j = 0; j < bn; j++, k++) sendbuf[k] = bv[i * blda + j]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2355 PetscInt blda, i, cK = A->cmap->N, k, j, bn; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv() local
2370 PetscCall(MatDenseGetLDA(b->A, &blda)); in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2378 for (i = 0; i < cK; i++, k++) sendbuf[k] = bv[i * blda + j]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 2651 PetscInt m, n, blda, clda; in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA() local
2712 PetscCall(MatDenseGetLDA(B, &blda)); in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2731 if (!mmdata->initialized || mmdata->Blda != blda || mmdata->Clda != clda) { in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2733 if (mmdata->initialized && mmdata->Blda != blda) { in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2738 …PetscCallCUSPARSE(cusparseCreateDnMat(&mmdata->matBDescr, B->rmap->n, B->cmap->n, blda, (void *)ba… in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2739 mmdata->Blda = blda; in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2794 …T, B->cmap->n, B->rmap->n, &PETSC_CUSPARSE_ONE, barray, blda, &PETSC_CUSPARSE_ZERO, barray, blda, … in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2796 blda = B->cmap->n; in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
2803 …, csrmat->column_indices->data().get(), mmdata->Bt ? mmdata->Bt : barray, blda, mat->beta_zero, ca… in MatProductNumeric_SeqAIJCUSPARSE_SeqDENSECUDA()
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 1324 PetscInt alda, blda, clda; in MatMatMult_Numeric_Dispatch() local
1327 PetscCall(MatDenseGetLDA(B, &blda)); in MatMatMult_Numeric_Dispatch()
1329 …_OP_T : CUPMBLAS_OP_N, m, n, k, &one, da.cupmdata(), alda, db.cupmdata(), blda, &zero, dc.cupmdata… in MatMatMult_Numeric_Dispatch()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 2218 PetscInt m, n, blda, clda; in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP() local
2277 PetscCall(MatDenseGetLDA(B, &blda)); in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2289 if (!mmdata->initialized || mmdata->Blda != blda || mmdata->Clda != clda) { in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2291 if (mmdata->initialized && mmdata->Blda != blda) { in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2296 …PetscCallHIPSPARSE(hipsparseCreateDnMat(&mmdata->matBDescr, B->rmap->n, B->cmap->n, blda, (void *)… in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
2297 mmdata->Blda = blda; in MatProductNumeric_SeqAIJHIPSPARSE_SeqDENSEHIP()
|
| /petsc/src/dm/dt/dualspace/impls/lagrange/ |
| H A D | dspacelagrange.c | 2575 PetscBLASInt bm, bn, bnrhs, blda, bldb, blwork, info; in PetscDualSpaceCreateInteriorSymmetryMatrix_Lagrange() local
2580 PetscCall(PetscBLASIntCast(bm, &blda)); in PetscDualSpaceCreateInteriorSymmetryMatrix_Lagrange()
2583 …PetscCallBLAS("LAPACKgels", LAPACKgels_(&transpose, &bm, &bn, &bnrhs, V, &blda, W, &bldb, work, &b… in PetscDualSpaceCreateInteriorSymmetryMatrix_Lagrange()
|