openvino: GPU to AUTO fallback on error

koush · Sep 11, 2024 · 674034f · 674034f
1 parent 44dcfe5
commit 674034f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/plugins/openvino/package-lock.json b/plugins/openvino/package-lock.json
diff --git a/plugins/openvino/package.json b/plugins/openvino/package.json
@@ -42,5 +42,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.113"
+   "version": "0.1.114"
 }
diff --git a/plugins/openvino/src/ov/__init__.py b/plugins/openvino/src/ov/__init__.py
@@ -136,6 +136,7 @@ def __init__(self, nativeId: str | None = None):
                     mode = f"AUTO:NPU,CPU"
             elif len(dgpus):
                 mode = f"AUTO:{','.join(dgpus)},CPU"
+            # forcing GPU can cause crashes on older GPU.
             elif gpu:
                 mode = f"GPU"
 
@@ -214,6 +215,9 @@ def __init__(self, nativeId: str | None = None):
             self.storage.removeItem("mode")
             self.storage.removeItem("model")
             self.storage.removeItem("precision")
+            if mode == "GPU":
+                print("GPU mode failed, reverting to AUTO.")
+                self.storage.setItem("mode", "AUTO")
             self.requestRestart()
 
         # mobilenet 1,300,300,3