GOSTSec/ccminer
1
0
Fork 0
mirror of https://github.com/GOSTSec/ccminer synced 2025-01-10 23:08:02 +00:00
Code Issues Releases Wiki Activity

nvapi: match gpu index with pci/x bus id

Browse Source 
mapping by Name was a quick test...
This commit is contained in:
Tanguy Pruvot 2014-11-14 21:49:30 +01:00
parent cc05128ae3
commit 7a2fe83aaa
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
nvml.cpp
View File

@ -31,7 +31,6 @@
extern wrap_nvml_handle *hnvml; extern wrap_nvml_handle *hnvml;
extern int num_processors; // gpus extern int num_processors; // gpus
extern int device_map[8]; extern int device_map[8];
extern char* device_name[8];
static uint32_t device_bus_ids[8] = { 0 }; static uint32_t device_bus_ids[8] = { 0 };
@ -462,14 +461,21 @@ int wrap_nvapi_init()
return -1; return -1;
} }
for (int g = 0; g < num_processors; g++) {
cudaDeviceProp props;
if (cudaGetDeviceProperties(&props, g) == cudaSuccess)
device_bus_ids[g] = props.pciBusID;
}
for (NvU8 i = 0; i < nvapi_dev_cnt; i++) { for (NvU8 i = 0; i < nvapi_dev_cnt; i++) {
NvAPI_ShortString name; NvAPI_ShortString name;
nvapi_dev_map[i] = i; // default mapping nvapi_dev_map[i] = i; // default mapping
ret = NvAPI_GPU_GetFullName(phys[i], name); ret = NvAPI_GPU_GetFullName(phys[i], name);
if (ret == NVAPI_OK) { if (ret == NVAPI_OK) {
for (int g = 0; g < num_processors; g++) { for (int g = 0; g < num_processors; g++) {
//todo : device_bus_ids, could be wrong on rigs NvU32 busId;
if (strcmp(device_name[g], name) == 0 && nvapi_dev_map[i] == i) { ret = NvAPI_GPU_GetBusId(phys[i], &busId);
if (ret == NVAPI_OK && busId == device_bus_ids[g]) {
nvapi_dev_map[i] = g; nvapi_dev_map[i] = g;
break; break;
} }
@ -562,7 +568,6 @@ int gpu_pstate(struct cgpu_info *gpu)
if (support == -1) { if (support == -1) {
unsigned int pst = 0; unsigned int pst = 0;
nvapi_getpstate(nvapi_dev_map[gpu->thr_id], &pst); nvapi_getpstate(nvapi_dev_map[gpu->thr_id], &pst);
//todo : convert ?
pstate = (int) pst; pstate = (int) pst;
} }
#endif #endif