File: numa.c

package info (click to toggle)
openmpi 5.0.7-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 202,312 kB
  • sloc: ansic: 612,441; makefile: 42,495; sh: 11,230; javascript: 9,244; f90: 7,052; java: 6,404; perl: 5,154; python: 1,856; lex: 740; fortran: 61; cpp: 20; tcl: 12
file content (92 lines) | stat: -rw-r--r-- 2,780 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#include <unistd.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <hwloc.h>
#include <hwloc/bitmap.h>

int main(int argc, char **argv)
{
    hwloc_topology_t topo;
#if HWLOC_API_VERSION >= 0x20000
    int ret;
#endif
    int num, weight, N;
    hwloc_obj_t obj;
    unsigned width, w;
    hwloc_bitmap_t numas[100], result;
    int m;
    char *tmp, cpus[2048];
    bool overlap = false;

    if (2 != argc) {
        fprintf(stderr, "Usage: %s xml-filename\n", argv[0]);
        return -1;
    }

    if (0 != hwloc_topology_init(&topo)) {
        return -1;
    }
    if (0 != hwloc_topology_set_xml(topo, argv[1])) {
        fprintf(stderr, "SETXML FAILED\n");
        return -1;
    }
    /* since we are loading this from an external source, we have to
     * explicitly set a flag so hwloc sets things up correctly
     */
#if HWLOC_API_VERSION >= 0x20000
    ret = hwloc_topology_set_io_types_filter(topo, HWLOC_TYPE_FILTER_KEEP_IMPORTANT);
    if (0 != ret) {
        fprintf(stderr, "IOTYPES FAILED\n");
        hwloc_topology_destroy(topo);
        return ret;
    }
#endif
    if (0 != hwloc_topology_set_flags(topo, HWLOC_TOPOLOGY_FLAG_IS_THISSYSTEM)) {
        fprintf(stderr, "SETFLAGS FAILED\n");
        hwloc_topology_destroy(topo);
        return -1;
    }
    /* now load the topology */
    if (0 != hwloc_topology_load(topo)) {
        fprintf(stderr, "LOAD FAILED\n");
        hwloc_topology_destroy(topo);
        return -1;
    }

    num = hwloc_get_nbobjs_by_type(topo, HWLOC_OBJ_NUMANODE);
    fprintf(stderr, "NUM NUMAS: %d\n", num);
    if (0 > num) {
        return -1;
    }
    width = (unsigned)num;

    N = 0;
    for (w = 0; w < width; w++) {
        /* get the object at this index */
        obj = hwloc_get_obj_by_type(topo, HWLOC_OBJ_NUMANODE, w);
        /* how many pu's are under it? */
        weight = hwloc_bitmap_weight(obj->cpuset);
        hwloc_bitmap_list_snprintf(cpus, 2048, obj->cpuset);
        fprintf(stderr, "NUMA: %u OSINDX: %u NPUS: %d PUS: %s\n", w, obj->os_index, weight, cpus);
        /* check for overlap with all preceding numas */
        for (m=0; m < N; m++) {
            if (hwloc_bitmap_intersects(obj->cpuset, numas[m])) {
                result = hwloc_bitmap_alloc();
                hwloc_bitmap_and(result, obj->cpuset, numas[m]);
                hwloc_bitmap_asprintf(&tmp, result);
                fprintf(stderr, "\tINTERSECTS NUMA %d: %s\n", m, tmp);
                hwloc_bitmap_free(result);
                free(tmp);
            }
        }
        /* cache this bitmap */
        numas[N] = hwloc_bitmap_alloc();
        hwloc_bitmap_copy(numas[N], obj->cpuset);
        ++N;
    }
    if (!overlap) {
        fprintf(stderr, "NO OVERLAP\n");
    }
}