Date: 2025-03-28T21:11:14+00:00

Actual source code: memc.c

      1: /*
      2:     We define the memory operations here. The reason we just do not use
      3:   the standard memory routines in the PETSc code is that on some machines
      4:   they are broken.

      6: */
      7: #include <petsc/private/petscimpl.h>
      8: #include <petscviewer.h>
      9: #include <../src/sys/utils/ftn-kernels/fcopy.h>

     11: /*@
     12:   PetscMemcmp - Compares two byte streams in memory.

     14:   Not Collective

     16:   Input Parameters:
     17: + str1 - Pointer to the first byte stream
     18: . str2 - Pointer to the second byte stream
     19: - len  - The length of the byte stream
     20:          (both str1 and str2 are assumed to be of length len)

     22:   Output Parameter:
     23: . e - `PETSC_TRUE` if equal else `PETSC_FALSE`.

     25:   Level: intermediate

     27:   Notes:
     28:   `PetscArraycmp()` is preferred

     30:   This routine is analogous to `memcmp()` with additional error checking

     32: .seealso: `PetscMemcpy()`, `PetscArrayzero()`, `PetscMemzero()`, `PetscArraycmp()`, `PetscArraycpy()`, `PetscStrallocpy()`,
     33:           `PetscArraymove()`
     34: @*/
     35: PetscErrorCode PetscMemcmp(const void *str1, const void *str2, size_t len, PetscBool *e)
     36: {
     37:   if (!len) {
     38:     // if e is a bad ptr I guess we just die here then?
     39:     *e = PETSC_TRUE;
     40:     return PETSC_SUCCESS;
     41:   }

     43:   PetscFunctionBegin;
     44:   PetscAssertPointer(str1, 1);
     45:   PetscAssertPointer(str2, 2);
     46:   PetscAssertPointer(e, 4);
     47:   *e = memcmp((char *)str1, (char *)str2, len) ? PETSC_FALSE : PETSC_TRUE;
     48:   PetscFunctionReturn(PETSC_SUCCESS);
     49: }

     51: #if defined(PETSC_HAVE_HWLOC)
     52: #include <petsc/private/petscimpl.h>
     53:   #include <hwloc.h>
     54: #endif

     56: /*@
     57:   PetscProcessPlacementView - display the MPI rank placement by core

     59:   Input Parameter:
     60: . viewer - `PETSCVIEWERASCII` to display the results on

     62:   Level: intermediate

     64:   Note:
     65:   Requires that PETSc be installed with hwloc, for example using `--download-hwloc`

     67: .seealso: `PetscInitialize()`
     68: @*/
     69: PetscErrorCode PetscProcessPlacementView(PetscViewer viewer)
     70: {
     71: #if defined(PETSC_HAVE_HWLOC)
     72:   PetscBool        isascii;
     73:   PetscMPIInt      rank;
     74:   hwloc_bitmap_t   set;
     75:   hwloc_topology_t topology;

     77:   PetscFunctionBegin;
     79:   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
     80:   PetscCheck(isascii, PetscObjectComm((PetscObject)viewer), PETSC_ERR_SUP, "Only ASCII viewer is supported");

     82:   PetscCallMPI(MPI_Comm_rank(MPI_COMM_WORLD, &rank));
     83:   hwloc_topology_init(&topology);
     84:   hwloc_topology_load(topology);
     85:   set = hwloc_bitmap_alloc();

     87:   PetscCallExternal(hwloc_get_proc_cpubind, topology, getpid(), set, HWLOC_CPUBIND_PROCESS);
     88:   PetscCall(PetscViewerASCIIPushSynchronized(viewer));
     89:   PetscCall(PetscViewerASCIISynchronizedPrintf(viewer, "MPI rank %d Process id: %d coreid %d\n", rank, getpid(), hwloc_bitmap_first(set)));
     90:   PetscCall(PetscViewerFlush(viewer));
     91:   hwloc_bitmap_free(set);
     92:   hwloc_topology_destroy(topology);
     93: #else
     94:   PetscFunctionBegin;
     95:   SETERRQ(PetscObjectComm((PetscObject)viewer), PETSC_ERR_SUP, "Requires PETSc be configured with --with-hwloc or --download-hwloc");
     96: #endif
     97:   PetscFunctionReturn(PETSC_SUCCESS);
     98: }