PetscErrorCode MatStashDestroy_Private(MatStash *stash)
{
PetscErrorCode ierr;
PetscFunctionBegin;
if (stash->space_head){
ierr = PetscMatStashSpaceDestroy(stash->space_head);CHKERRQ(ierr);
stash->space_head = 0;
stash->space = 0;
}
ierr = PetscFree(stash->flg_v);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*
MatStashDestroy_Private - Destroy the stash
*/
PetscErrorCode MatStashDestroy_Private(MatStash *stash)
{
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = PetscMatStashSpaceDestroy(&stash->space_head);CHKERRQ(ierr);
if (stash->ScatterDestroy) {ierr = (*stash->ScatterDestroy)(stash);CHKERRQ(ierr);}
stash->space = 0;
ierr = PetscFree(stash->flg_v);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode MatStashScatterEnd_Private(MatStash *stash)
{
PetscErrorCode ierr;
PetscInt nsends=stash->nsends,bs2,oldnmax,i;
MPI_Status *send_status;
PetscFunctionBegin;
for (i=0; i<2*stash->size; i++) stash->flg_v[i] = -1;
/* wait on sends */
if (nsends) {
ierr = PetscMalloc(2*nsends*sizeof(MPI_Status),&send_status);CHKERRQ(ierr);
ierr = MPI_Waitall(2*nsends,stash->send_waits,send_status);CHKERRQ(ierr);
ierr = PetscFree(send_status);CHKERRQ(ierr);
}
/* Now update nmaxold to be app 10% more than max n used, this way the
wastage of space is reduced the next time this stash is used.
Also update the oldmax, only if it increases */
if (stash->n) {
bs2 = stash->bs*stash->bs;
oldnmax = ((int)(stash->n * 1.1) + 5)*bs2;
if (oldnmax > stash->oldnmax) stash->oldnmax = oldnmax;
}
stash->nmax = 0;
stash->n = 0;
stash->reallocs = -1;
stash->nprocessed = 0;
if (stash->space_head){
ierr = PetscMatStashSpaceDestroy(stash->space_head);CHKERRQ(ierr);
stash->space_head = 0;
stash->space = 0;
}
ierr = PetscFree(stash->send_waits);CHKERRQ(ierr);
stash->send_waits = 0;
ierr = PetscFree(stash->recv_waits);CHKERRQ(ierr);
stash->recv_waits = 0;
ierr = PetscFree2(stash->svalues,stash->sindices);CHKERRQ(ierr);
stash->svalues = 0;
ierr = PetscFree(stash->rvalues[0]);CHKERRQ(ierr);
ierr = PetscFree(stash->rvalues);CHKERRQ(ierr);
stash->rvalues = 0;
ierr = PetscFree(stash->rindices[0]);CHKERRQ(ierr);
ierr = PetscFree(stash->rindices);CHKERRQ(ierr);
stash->rindices = 0;
PetscFunctionReturn(0);
}
