Raincloud/sheepit-shadow-nabber
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Bugfix: GPU could be on the same bus (and having the same pci bus id)

Browse Source
This commit is contained in:
Laurent Clouet
2019-11-06 14:04:59 +01:00
parent 7b693a8f36
commit 2ceecb5830
2 changed files with 12 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/com/sheepit/client/hardware/gpu/GPUDevice.java
View File

@@ -35,6 +35,11 @@ public class GPUDevice {
this.id = id;
}
public GPUDevice(String type, String model, long ram, String id, String oldId) {
this(type, model, ram, id);
this.oldId = oldId;
}
public String getType() {
return type;
}

25
src/com/sheepit/client/hardware/gpu/nvidia/Nvidia.java
View File

@@ -1,11 +1,8 @@
package com.sheepit.client.hardware.gpu.nvidia;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import com.sheepit.client.hardware.gpu.nvidia.CUDeviceAttribute;
import com.sheepit.client.hardware.gpu.GPUDevice;
import com.sheepit.client.hardware.gpu.GPULister;
import com.sheepit.client.os.OS;
@@ -64,9 +61,8 @@ public class Nvidia implements GPULister {
return null;
}
List<GPUDevice> devices = new LinkedList<GPUDevice>();
HashMap<String, GPUDevice> devicesWithPciId = new HashMap<String, GPUDevice>(count.getValue());
List<GPUDevice> devices = new ArrayList<>(count.getValue());
for (int num = 0; num < count.getValue(); num++) {
IntByReference aDevice = new IntByReference();
@@ -122,17 +118,10 @@ public class Nvidia implements GPULister {
pciDomainId.getValue(),
pciBusId.getValue(),
pciDeviceId.getValue());
devicesWithPciId.put(Integer.toString(pciBusId.getValue()), new GPUDevice(TYPE, new String(name).trim(), ram.getValue(), blenderId));
}
// for backward compatibility generate a CUDA_N id
// in theory a set to environment "CUDA_DEVICE_ORDER=PCI_BUS_ID" should be enough but it didn't work
int i = 0;
for (Map.Entry<String, GPUDevice> entry : devicesWithPciId.entrySet()){
GPUDevice aDevice = entry.getValue();
aDevice.setOldId(TYPE + "_" + Integer.toString(i));
devices.add(aDevice);
i++;
GPUDevice gpu = new GPUDevice(TYPE, new String(name).trim(), ram.getValue(), blenderId);
// for backward compatibility generate a CUDA_N id
gpu.setOldId(TYPE + "_" + num);
devices.add(gpu);
}
return devices;