-
Notifications
You must be signed in to change notification settings - Fork 168
/
levelzero.c
120 lines (97 loc) · 3.22 KB
/
levelzero.c
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/*
* Copyright © 2021 Inria. All rights reserved.
* See COPYING in top-level directory.
*/
#include <stdio.h>
#include <assert.h>
#include <level_zero/ze_api.h>
#include <level_zero/zes_api.h>
#include "hwloc.h"
#include "hwloc/levelzero.h"
/* check the RSMI helpers */
int main(void)
{
hwloc_topology_t topology;
ze_driver_handle_t *drh;
uint32_t nbdrivers, i, k;
ze_result_t res;
int err = 0;
putenv((char *) "ZES_ENABLE_SYSMAN=1");
res = zeInit(0);
if (res != ZE_RESULT_SUCCESS) {
fprintf(stderr, "Failed to initialize LevelZero in zeInit(): %d\n", (int)res);
return 0;
}
hwloc_topology_init(&topology);
hwloc_topology_set_io_types_filter(topology, HWLOC_TYPE_FILTER_KEEP_IMPORTANT);
hwloc_topology_load(topology);
nbdrivers = 0;
res = zeDriverGet(&nbdrivers, NULL);
if (res != ZE_RESULT_SUCCESS || !nbdrivers)
return 0;
drh = malloc(nbdrivers * sizeof(*drh));
if (!drh)
return 0;
res = zeDriverGet(&nbdrivers, drh);
if (res != ZE_RESULT_SUCCESS) {
free(drh);
return 0;
}
printf("found %u L0 drivers\n", nbdrivers);
k = 0;
for(i=0; i<nbdrivers; i++) {
uint32_t nbdevices, j;
ze_device_handle_t *dvh;
nbdevices = 0;
res = zeDeviceGet(drh[i], &nbdevices, NULL);
if (res != ZE_RESULT_SUCCESS || !nbdevices)
continue;
dvh = malloc(nbdevices * sizeof(*dvh));
if (!dvh)
continue;
res = zeDeviceGet(drh[i], &nbdevices, dvh);
if (res != ZE_RESULT_SUCCESS) {
free(dvh);
continue;
}
printf("found %u L0 devices in driver #%u\n", nbdevices, i);
for (j=0; j<nbdevices; j++, k++) {
hwloc_bitmap_t set;
hwloc_obj_t osdev, ancestor;
const char *value;
osdev = hwloc_levelzero_get_device_osdev(topology, dvh[j]);
assert(osdev);
ancestor = hwloc_get_non_io_ancestor_obj(topology, osdev);
printf("found OSDev %s\n", osdev->name);
err = strncmp(osdev->name, "ze", 2);
assert(!err);
assert(atoi(osdev->name+2) == (int) k);
assert(osdev->attr->osdev.type == HWLOC_OBJ_OSDEV_COPROC);
value = hwloc_obj_get_info_by_name(osdev, "Backend");
err = strcmp(value, "LevelZero");
assert(!err);
value = hwloc_obj_get_info_by_name(osdev, "LevelZeroDriverIndex");
assert(value);
assert(atoi(value) == (int) i);
value = hwloc_obj_get_info_by_name(osdev, "LevelZeroDriverDeviceIndex");
assert(value);
assert(atoi(value) == (int) j);
set = hwloc_bitmap_alloc();
err = hwloc_levelzero_get_device_cpuset(topology, dvh[j], set);
if (err < 0) {
printf("failed to get cpuset for driver #%u device #%u\n", i, j);
} else {
char *cpuset_string = NULL;
hwloc_bitmap_asprintf(&cpuset_string, set);
printf("got cpuset %s for driver #%u device #%u\n", cpuset_string, j, i);
free(cpuset_string);
if (hwloc_bitmap_isequal(hwloc_topology_get_complete_cpuset(topology), hwloc_topology_get_topology_cpuset(topology)))
/* only compare if the topology is complete, otherwise things can be significantly different */
assert(hwloc_bitmap_isequal(set, ancestor->cpuset));
}
hwloc_bitmap_free(set);
}
}
hwloc_topology_destroy(topology);
return err;
}