blob: a08fe333e0b83fa6856e9d59fe1e10d48aaaa53e (
plain)
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
|
/*
* Test for host vGPU initialization
*/
#include <gvm/apis.h>
#include <gvm/nvidia/manager.h>
#include <gvm/nvidia/creator.h>
#include <gvm/nvidia/open/signature.h>
#include <utils/configs.h>
#include <stdio.h>
#include <stdlib.h>
int main(int argc, char *argv[])
{
int ret = 0;
enum SupportedGpus selected_gpu = Nvidia;
const struct ManagerAPI *api = MANAGER_APIS[selected_gpu];
struct NvManager *mgr = NULL;
printf("Testing host vGPU initialization...\n");
mgr = calloc(1, sizeof(struct NvManager));
if (!mgr) {
printf("Failed to allocate manager\n");
return 1;
}
mgr->ignore_version_check = 1;
mgr->create_signature = nvidia_open_create_signature;
mgr->use_vmiop = 1;
mgr->vmiop_plugin = "/usr/lib/x86_64-linux-gnu/libnvidia-vgpu.so";
if (api->init(mgr)) {
printf("FAIL: Manager initialization failed\n");
ret = 1;
goto cleanup;
}
printf("PASS: Manager initialized successfully\n");
if (!api->running(mgr)) {
printf("FAIL: Manager not running\n");
ret = 1;
goto cleanup;
}
printf("PASS: Manager is running\n");
if (mgr->vmiop) {
printf("PASS: VMIOP initialized\n");
} else {
printf("FAIL: VMIOP not initialized\n");
ret = 1;
}
cleanup:
if (mgr) {
api->stop(mgr);
free(mgr);
}
if (ret == 0)
printf("\nAll host initialization tests PASSED\n");
else
printf("\nSome host initialization tests FAILED\n");
return ret;
}
|
