Merge pull request #62 from CCob/master

Fix issue where platform can return no devices
mandiant · Nov 4, 2020 · bd43069 · bd43069
2 parents e3b20e4 + 1d6b33b
commit bd43069
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/worker/opencl_info.go b/worker/opencl_info.go
@@ -20,6 +20,9 @@ func GetAvailableDevices() (shared.DeviceMap, error) {
 	for _, platform := range platforms {
 		devices, err := platform.GetDevices(opencl.DeviceTypeAll)
 		if err != nil {
+			if err == opencl.ErrDeviceNotFound {
+				continue
+			}
 			return nil, err
 		}
 
@@ -39,5 +42,9 @@ func GetAvailableDevices() (shared.DeviceMap, error) {
 		}
 	}
 
+	if len(devs) == 0 {
+		return nil, opencl.ErrDeviceNotFound
+	}
+
 	return devs, nil
 }