18 static PetscErrorCode AOView_MemoryScalable(AO ao, PetscViewer viewer)  in AOView_MemoryScalable()  argument
21   AO_MemoryScalable *aomems = (AO_MemoryScalable *)ao->data;  in AOView_MemoryScalable()
32   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)ao), &rank));  in AOView_MemoryScalable()
33   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)ao), &size));  in AOView_MemoryScalable()
35   PetscCall(PetscObjectGetNewTag((PetscObject)ao, &tag_app));  in AOView_MemoryScalable()
36   PetscCall(PetscObjectGetNewTag((PetscObject)ao, &tag_petsc));  in AOView_MemoryScalable()
39 …cCall(PetscViewerASCIIPrintf(viewer, "Number of elements in ordering %" PetscInt_FMT "\n", ao->N));  in AOView_MemoryScalable()
53 …PetscCallMPI(MPIU_Recv(app_loc, len, MPIU_INT, i, tag_app, PetscObjectComm((PetscObject)ao), &stat…  in AOView_MemoryScalable()
54 …lMPI(MPIU_Recv(petsc_loc, len, MPIU_INT, i, tag_petsc, PetscObjectComm((PetscObject)ao), &status));  in AOView_MemoryScalable()
62 …PIU_Send((void *)aomems->app_loc, map->n, MPIU_INT, 0, tag_app, PetscObjectComm((PetscObject)ao)));  in AOView_MemoryScalable()
63 …Send((void *)aomems->petsc_loc, map->n, MPIU_INT, 0, tag_petsc, PetscObjectComm((PetscObject)ao)));  in AOView_MemoryScalable()
69 static PetscErrorCode AODestroy_MemoryScalable(AO ao)  in AODestroy_MemoryScalable()  argument
71   AO_MemoryScalable *aomems = (AO_MemoryScalable *)ao->data;  in AODestroy_MemoryScalable()
90 static PetscErrorCode AOMap_MemoryScalable_private(AO ao, PetscInt n, PetscInt *ia, const PetscInt …  in AOMap_MemoryScalable_private()  argument
92   AO_MemoryScalable *aomems = (AO_MemoryScalable *)ao->data;  in AOMap_MemoryScalable_private()
107   PetscCall(PetscObjectGetComm((PetscObject)ao, &comm));  in AOMap_MemoryScalable_private()
119     if (ia[i] >= ao->N) owner[i] = -2; /* mark out of range entries with special negative value */  in AOMap_MemoryScalable_private()
143   PetscCall(PetscObjectGetNewTag((PetscObject)ao, &tag1));  in AOMap_MemoryScalable_private()
144   PetscCall(PetscObjectGetNewTag((PetscObject)ao, &tag2));  in AOMap_MemoryScalable_private()
235 static PetscErrorCode AOPetscToApplication_MemoryScalable(AO ao, PetscInt n, PetscInt *ia)  in AOPetscToApplication_MemoryScalable()  argument
237   AO_MemoryScalable *aomems  = (AO_MemoryScalable *)ao->data;  in AOPetscToApplication_MemoryScalable()
241   PetscCall(AOMap_MemoryScalable_private(ao, n, ia, app_loc));  in AOPetscToApplication_MemoryScalable()
245 static PetscErrorCode AOApplicationToPetsc_MemoryScalable(AO ao, PetscInt n, PetscInt *ia)  in AOApplicationToPetsc_MemoryScalable()  argument
247   AO_MemoryScalable *aomems    = (AO_MemoryScalable *)ao->data;  in AOApplicationToPetsc_MemoryScalable()
251   PetscCall(AOMap_MemoryScalable_private(ao, n, ia, petsc_loc));  in AOApplicationToPetsc_MemoryScalable()
266 … PetscInt napp, const PetscInt from_array[], const PetscInt to_array[], AO ao, PetscInt *aomap_loc)  in AOCreateMemoryScalable_private()  argument
268   AO_MemoryScalable *aomems  = (AO_MemoryScalable *)ao->data;  in AOCreateMemoryScalable_private()
317   PetscCall(PetscObjectGetNewTag((PetscObject)ao, &tag));  in AOCreateMemoryScalable_private()
385 PETSC_INTERN PetscErrorCode AOCreate_MemoryScalable(AO ao)  in AOCreate_MemoryScalable()  argument
387   IS                 isapp = ao->isapp, ispetsc = ao->ispetsc;  in AOCreate_MemoryScalable()
396 …PetscCheck(isapp, PetscObjectComm((PetscObject)ao), PETSC_ERR_ARG_WRONGSTATE, "AOSetIS() must be c…  in AOCreate_MemoryScalable()
399   ao->data   = (void *)aomems;  in AOCreate_MemoryScalable()
400   ao->ops[0] = AOOps_MemoryScalable;  in AOCreate_MemoryScalable()
401   PetscCall(PetscObjectChangeTypeName((PetscObject)ao, AOMEMORYSCALABLE));  in AOCreate_MemoryScalable()
437   ao->N       = N;  in AOCreate_MemoryScalable()
438   ao->n       = map->n;  in AOCreate_MemoryScalable()
446   PetscCall(AOCreateMemoryScalable_private(comm, napp, petsc, myapp, ao, aomems->app_loc));  in AOCreate_MemoryScalable()
447   PetscCall(AOCreateMemoryScalable_private(comm, napp, myapp, petsc, ao, aomems->petsc_loc));  in AOCreate_MemoryScalable()
527   AO       ao;  in AOCreateMemoryScalableIS()  local
531   PetscCall(AOCreate(comm, &ao));  in AOCreateMemoryScalableIS()
532   PetscCall(AOSetIS(ao, isapp, ispetsc));  in AOCreateMemoryScalableIS()
533   PetscCall(AOSetType(ao, AOMEMORYSCALABLE));  in AOCreateMemoryScalableIS()
534   PetscCall(AOViewFromOptions(ao, NULL, "-ao_view"));  in AOCreateMemoryScalableIS()
535   *aoout = ao;  in AOCreateMemoryScalableIS()