| /petsc/src/mat/impls/hypre/ |
| H A D | mhypre.c | 2248 PetscBLASInt one = 1, bnz; local
2269 PetscCall(PetscBLASIntCast(xnnz, &bnz));
2270 …PetscCallBLAS("BLASaxpy", BLASaxpy_(&bnz, &a, (PetscScalar *)xarr, &one, (PetscScalar *)yarr, &one…
2288 PetscCall(PetscBLASIntCast(xnnz, &bnz));
2289 …PetscCallBLAS("BLASaxpy", BLASaxpy_(&bnz, &a, (PetscScalar *)xarr, &one, (PetscScalar *)yarr, &one…
|
| /petsc/src/mat/impls/sbaij/mpi/ |
| H A D | mpisbaij.c | 1655 PetscBLASInt bnz, one = 1; in MatAXPY_MPISBAIJ() local
1664 PetscCall(PetscBLASIntCast(xa->nz, &bnz)); in MatAXPY_MPISBAIJ()
1665 PetscCallBLAS("BLASaxpy", BLASaxpy_(&bnz, &alpha, xa->a, &one, ya->a, &one)); in MatAXPY_MPISBAIJ()
1668 PetscCall(PetscBLASIntCast(xb->nz, &bnz)); in MatAXPY_MPISBAIJ()
1669 PetscCallBLAS("BLASaxpy", BLASaxpy_(&bnz, &alpha, xb->a, &one, yb->a, &one)); in MatAXPY_MPISBAIJ()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 3323 PetscBLASInt one = 1, bnz = 1; in MatAXPY_SeqAIJHIPSPARSE() local
3328 PetscCall(PetscBLASIntCast(x->nz, &bnz)); in MatAXPY_SeqAIJHIPSPARSE()
3330 PetscCallHIPBLAS(hipblasXaxpy(hipblasv2handle, bnz, &a, ax, one, ay, one)); in MatAXPY_SeqAIJHIPSPARSE()
3331 PetscCall(PetscLogGpuFlops(2.0 * bnz)); in MatAXPY_SeqAIJHIPSPARSE()
3347 PetscBLASInt one = 1, bnz = 1; in MatScale_SeqAIJHIPSPARSE() local
3352 PetscCall(PetscBLASIntCast(y->nz, &bnz)); in MatScale_SeqAIJHIPSPARSE()
3354 PetscCallHIPBLAS(hipblasXscal(hipblasv2handle, bnz, &a, ay, one)); in MatScale_SeqAIJHIPSPARSE()
3355 PetscCall(PetscLogGpuFlops(bnz)); in MatScale_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/sbaij/seq/ |
| H A D | sbaij.c | 981 PetscBLASInt bnz; in MatAXPY_SeqSBAIJ() local
982 PetscCall(PetscBLASIntCast(x->nz * bs2, &bnz)); in MatAXPY_SeqSBAIJ()
983 PetscCallBLAS("BLASaxpy", BLASaxpy_(&bnz, &alpha, x->a, &one, y->a, &one)); in MatAXPY_SeqSBAIJ()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 3931 PetscBLASInt one = 1, bnz = 1; in MatAXPY_SeqAIJCUSPARSE() local
3936 PetscCall(PetscBLASIntCast(x->nz, &bnz)); in MatAXPY_SeqAIJCUSPARSE()
3938 PetscCallCUBLAS(cublasXaxpy(cublasv2handle, bnz, &a, ax, one, ay, one)); in MatAXPY_SeqAIJCUSPARSE()
3939 PetscCall(PetscLogGpuFlops(2.0 * bnz)); in MatAXPY_SeqAIJCUSPARSE()
3955 PetscBLASInt one = 1, bnz = 1; in MatScale_SeqAIJCUSPARSE() local
3960 PetscCall(PetscBLASIntCast(y->nz, &bnz)); in MatScale_SeqAIJCUSPARSE()
3962 PetscCallCUBLAS(cublasXscal(cublasv2handle, bnz, &a, ay, one)); in MatScale_SeqAIJCUSPARSE()
3963 PetscCall(PetscLogGpuFlops(bnz)); in MatScale_SeqAIJCUSPARSE()
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aij.c | 2623 PetscBLASInt one = 1, bnz; in MatScale_SeqAIJ() local
2627 PetscCall(PetscBLASIntCast(a->nz, &bnz)); in MatScale_SeqAIJ()
2628 PetscCallBLAS("BLASscal", BLASscal_(&bnz, &alpha, v, &one)); in MatScale_SeqAIJ()
2960 PetscBLASInt one = 1, bnz; in MatAXPY_SeqAIJ() local
2962 PetscCall(PetscBLASIntCast(x->nz, &bnz)); in MatAXPY_SeqAIJ()
2965 PetscCallBLAS("BLASaxpy", BLASaxpy_(&bnz, &alpha, xa, &one, ya, &one)); in MatAXPY_SeqAIJ()
2968 PetscCall(PetscLogFlops(2.0 * bnz)); in MatAXPY_SeqAIJ()
|
| /petsc/src/mat/impls/baij/mpi/ |
| H A D | mpibaij.c | 1811 PetscBLASInt bnz, one = 1; in MatAXPY_MPIBAIJ() local
1820 PetscCall(PetscBLASIntCast(x->nz * bs2, &bnz)); in MatAXPY_MPIBAIJ()
1821 PetscCallBLAS("BLASaxpy", BLASaxpy_(&bnz, &alpha, x->a, &one, y->a, &one)); in MatAXPY_MPIBAIJ()
1824 PetscCall(PetscBLASIntCast(x->nz * bs2, &bnz)); in MatAXPY_MPIBAIJ()
1825 PetscCallBLAS("BLASaxpy", BLASaxpy_(&bnz, &alpha, x->a, &one, y->a, &one)); in MatAXPY_MPIBAIJ()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpimatmatmult.c | 1603 …PetscInt cnz = 0, *bj_i, *bi, *bj, bnz, nextcj; /* bi,bj,ba: local array of C(mpi mat)… in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ() local
1726 bnz = bi[i + 1] - bi[i]; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1745 PetscCall(MatSetValues(C, 1, &row, bnz, bj_i, ba_i, INSERT_VALUES)); in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
|
| H A D | mpiaij.c | 1501 …PetscInt i, j, m, n, ng, anz, bnz, *dnnz, *onnz, *tdnnz, *tonnz, *rdest, *cdest, *work, *gc… in MatPermute_MPIAIJ() local
1545 PetscCall(MatGetRowIJ(aB, 0, PETSC_FALSE, PETSC_FALSE, &bnz, &bi, &bj, &done)); in MatPermute_MPIAIJ()
1591 PetscCall(MatRestoreRowIJ(aB, 0, PETSC_FALSE, PETSC_FALSE, &bnz, &bi, &bj, &done)); in MatPermute_MPIAIJ()
|
| /petsc/src/mat/impls/baij/seq/ |
| H A D | baij.c | 2749 PetscBLASInt bnz; in MatAXPY_SeqBAIJ() local
2750 PetscCall(PetscBLASIntCast(x->nz * bs2, &bnz)); in MatAXPY_SeqBAIJ()
2751 PetscCallBLAS("BLASaxpy", BLASaxpy_(&bnz, &alpha, x->a, &one, y->a, &one)); in MatAXPY_SeqBAIJ()
|