Lines Matching refs:Ba
127 …alar Aa[], PetscCount Bnnz, const PetscCount Bjmap1[], const PetscCount Bperm1[], PetscScalar Ba[]) in MatAddLocalCOOValues() argument
139 Ba[i] = (imode == INSERT_VALUES ? 0.0 : Ba[i]) + sum; in MatAddLocalCOOValues()
144 … const PetscCount Bimap2[], const PetscCount Bjmap2[], const PetscCount Bperm2[], PetscScalar Ba[]) in MatAddRemoteCOOValues() argument
153 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += kv[Bperm2[k]]; in MatAddRemoteCOOValues()
162 PetscScalar *Aa, *Ba; in MatSetValuesCOO_MPIAIJCUSPARSE() local
199 PetscCall(MatSeqAIJCUSPARSEGetArrayWrite(B, &Ba)); in MatSetValuesCOO_MPIAIJCUSPARSE()
202 PetscCall(MatSeqAIJCUSPARSEGetArray(B, &Ba)); in MatSetValuesCOO_MPIAIJCUSPARSE()
216 …((Annz + Bnnz + 255) / 256), 256>>>(v1, imode, Annz, Ajmap1, Aperm1, Aa, Bnnz, Bjmap1, Bperm1, Ba); in MatSetValuesCOO_MPIAIJCUSPARSE()
223 …z2 + 255) / 256), 256>>>(v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2, Bjmap2, Bperm2, Ba); in MatSetValuesCOO_MPIAIJCUSPARSE()
230 PetscCall(MatSeqAIJCUSPARSERestoreArrayWrite(B, &Ba)); in MatSetValuesCOO_MPIAIJCUSPARSE()
233 PetscCall(MatSeqAIJCUSPARSERestoreArray(B, &Ba)); in MatSetValuesCOO_MPIAIJCUSPARSE()