add protection

zhangmeng

2019-07-11 fb217f9ffc1b40d49feae8d4683efa5d8c5da1c9

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#ifdef __cplusplus
extern "C"{
#endif
 
#include "cgpu.h"
 
#ifdef __cplusplus
}
#endif
 
#include "gpu/info.h"
#include "gpu/info.cpp"
 
int get_idle_gpu(const int mem_size){
    return gpu::getIdleGPU(mem_size);
}
 
char* get_gpu_info(){
    return gpu::getGpuInfo();
}