147   PetscScalar       *work, *values;  in MatGetDiagonal_Normal()  local
153   PetscCall(PetscMalloc1(A->cmap->N, &work));  in MatGetDiagonal_Normal()
154   PetscCall(PetscArrayzero(work, A->cmap->N));  in MatGetDiagonal_Normal()
158     for (j = 0; j < nnz; j++) work[cols[j]] += mvalues[j] * mvalues[j];  in MatGetDiagonal_Normal()
161 …PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, work, A->cmap->N, MPIU_SCALAR, MPIU_SUM, PetscObjectComm…  in MatGetDiagonal_Normal()
165   PetscCall(PetscArraycpy(values, work + rstart, rend - rstart));  in MatGetDiagonal_Normal()
167   PetscCall(PetscFree(work));  in MatGetDiagonal_Normal()
267   Mat work[2];  member
290   PetscCall(MatProductNumeric(contents->work[0]));  in MatProductNumeric_Normal_Dense()
292   PetscCall(MatDensePlaceArray(contents->work[1], array));  in MatProductNumeric_Normal_Dense()
293   PetscCall(MatProductNumeric(contents->work[1]));  in MatProductNumeric_Normal_Dense()
295   PetscCall(MatDenseResetArray(contents->work[1]));  in MatProductNumeric_Normal_Dense()
308   PetscCall(MatDestroy(contents->work));  in MatNormal_DenseDestroy()
309   PetscCall(MatDestroy(contents->work + 1));  in MatNormal_DenseDestroy()
344   if (right) PetscCall(MatProductCreate(a->A, C, NULL, contents->work));  in MatProductSymbolic_Normal_Dense()
345   else PetscCall(MatProductCreate(a->A, B, NULL, contents->work));  in MatProductSymbolic_Normal_Dense()
346   PetscCall(MatProductSetType(contents->work[0], MATPRODUCT_AB));  in MatProductSymbolic_Normal_Dense()
347   PetscCall(MatProductSetFromOptions(contents->work[0]));  in MatProductSymbolic_Normal_Dense()
348   PetscCall(MatProductSymbolic(contents->work[0]));  in MatProductSymbolic_Normal_Dense()
349   PetscCall(MatProductCreate(a->A, contents->work[0], NULL, contents->work + 1));  in MatProductSymbolic_Normal_Dense()
350   PetscCall(MatProductSetType(contents->work[1], MATPRODUCT_AtB));  in MatProductSymbolic_Normal_Dense()
351   PetscCall(MatProductSetFromOptions(contents->work[1]));  in MatProductSymbolic_Normal_Dense()
352   PetscCall(MatProductSymbolic(contents->work[1]));  in MatProductSymbolic_Normal_Dense()
354   PetscCall(MatSeqDenseSetPreallocation(contents->work[1], array));  in MatProductSymbolic_Normal_Dense()
355   PetscCall(MatMPIDenseSetPreallocation(contents->work[1], array));  in MatProductSymbolic_Normal_Dense()