Lines Matching refs:Aa
127 …nt Annz, const PetscCount Ajmap1[], const PetscCount Aperm1[], PetscScalar Aa[], PetscCount Bnnz, … in MatAddLocalCOOValues() argument
135 Aa[i] = (imode == INSERT_VALUES ? 0.0 : Aa[i]) + sum; in MatAddLocalCOOValues()
144 …imap2[], const PetscCount Ajmap2[], const PetscCount Aperm2[], PetscScalar Aa[], PetscCount Bnnz2,… in MatAddRemoteCOOValues() argument
150 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += kv[Aperm2[k]]; in MatAddRemoteCOOValues()
162 PetscScalar *Aa, *Ba; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
198 PetscCall(MatSeqAIJHIPSPARSEGetArrayWrite(A, &Aa)); /* write matrix values */ in MatSetValuesCOO_MPIAIJHIPSPARSE()
201 PetscCall(MatSeqAIJHIPSPARSEGetArray(A, &Aa)); /* read & write matrix values */ in MatSetValuesCOO_MPIAIJHIPSPARSE()
216 …56), dim3(256), 0, PetscDefaultHipStream, v1, imode, Annz, Ajmap1, Aperm1, Aa, Bnnz, Bjmap1, Bperm… in MatSetValuesCOO_MPIAIJHIPSPARSE()
223 …), dim3(256), 0, PetscDefaultHipStream, v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2, Bjma… in MatSetValuesCOO_MPIAIJHIPSPARSE()
229 PetscCall(MatSeqAIJHIPSPARSERestoreArrayWrite(A, &Aa)); in MatSetValuesCOO_MPIAIJHIPSPARSE()
232 PetscCall(MatSeqAIJHIPSPARSERestoreArray(A, &Aa)); in MatSetValuesCOO_MPIAIJHIPSPARSE()