Bugfix: if two gpus have the same name/model, the actual gpu used was the first in the list

This commit is contained in:
Laurent Clouet
2015-01-24 17:49:27 +00:00
parent 93d34bc571
commit dce52320c9
2 changed files with 35 additions and 12 deletions

View File

@@ -96,7 +96,7 @@ public class GPU {
return true;
}
public static List<String> listDevices() {
public static List<String> listModels() {
if (devices == null) {
generate();
}
@@ -111,6 +111,17 @@ public class GPU {
return devs;
}
public static List<GPUDevice> listDevices() {
if (devices == null) {
generate();
}
if (devices == null) {
return null;
}
return devices;
}
public static GPUDevice getGPUDevice(String device_model) {
if (device_model == null) {
return null;