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

[v0.04] Fix bug in which NVML was using the wrong GPU id

Browse Source
This commit is contained in:
Dr-Noob
2021-08-16 15:18:55 +02:00
parent fd15008ab2
commit dea5211791
3 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/cuda/uarch.cpp
View File

@@ -252,9 +252,12 @@ void map_chip_to_uarch(struct uarch* arch) {
struct uarch* get_uarch_from_cuda(struct gpu_info* gpu) {
struct uarch* arch = (struct uarch*) emalloc(sizeof(struct uarch));
int dev = 0;
cudaError_t err = cudaSuccess;
cudaDeviceProp deviceProp;
cudaGetDeviceProperties(&deviceProp, dev);
if ((err = cudaGetDeviceProperties(&deviceProp, gpu->idx)) != cudaSuccess) {
printErr("%s: %s", cudaGetErrorName(err), cudaGetErrorString(err));
return NULL;
}
arch->chip_str = NULL;
arch->cc_major = deviceProp.major;