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
|
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/
#include "mpi.h"
#include <stdio.h>
int main(int argc, char **argv)
{
int cuda_support, ze_support, hip_support;
MPI_Init(NULL, NULL);
MPIX_GPU_query_support(MPIX_GPU_SUPPORT_CUDA, &cuda_support);
MPIX_GPU_query_support(MPIX_GPU_SUPPORT_ZE, &ze_support);
MPIX_GPU_query_support(MPIX_GPU_SUPPORT_HIP, &hip_support);
if (cuda_support && ze_support && hip_support) {
printf("CUDA and ZE and HIP are supported\n");
} else if (cuda_support && ze_support) {
printf("CUDA and ZE are supported\n");
} else if (cuda_support && hip_support) {
printf("CUDA and HIP are supported\n");
} else if (ze_support && hip_support) {
printf("ZE and HIP are supported\n");
} else if (cuda_support) {
printf("CUDA is supported\n");
} else if (ze_support) {
printf("ZE is supported\n");
} else if (hip_support) {
printf("HIP is supported\n");
} else {
printf("No GPUs are supported\n");
}
MPI_Finalize();
return 0;
}
|
