Franscobec/gpufetch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[v0.23] Sort PCI devices; this makes the devices list to match CUDA driver ordering, which fixes a bug when there was more than one NVIDIA GPU

Browse Source
This commit is contained in:
Dr-Noob
2022-01-22 13:25:22 +01:00
parent 23586a18e9
commit bd1158c139
11 changed files with 103 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/common/master.cpp
View File

@@ -1,6 +1,7 @@
#include <cstdlib>
#include <cstdio>
#include "pci.hpp"
#include "global.hpp"
#include "colors.hpp"
#include "master.hpp"
@@ -16,6 +17,7 @@ struct gpu_list {
struct gpu_list* get_gpu_list() {
int idx = 0;
struct pci_dev *devices = get_pci_devices_from_pciutils();
struct gpu_list* list = (struct gpu_list*) malloc(sizeof(struct gpu_list));
list->num_gpus = 0;
list->gpus = (struct gpu_info**) malloc(sizeof(struct info*) * MAX_GPUS);
@@ -24,7 +26,7 @@ struct gpu_list* get_gpu_list() {
bool valid = true;
while(valid) {
list->gpus[idx] = get_gpu_info_cuda(idx);
list->gpus[idx] = get_gpu_info_cuda(devices, idx);
if(list->gpus[idx] != NULL) idx++;
else valid = false;
}
@@ -33,7 +35,7 @@ struct gpu_list* get_gpu_list() {
#endif
#ifdef BACKEND_INTEL
list->gpus[idx] = get_gpu_info_intel();
list->gpus[idx] = get_gpu_info_intel(devices);
if(list->gpus[idx] != NULL) list->num_gpus++;
#endif