openvino: improve device selection

koush · Aug 30, 2024 · 7104ad6 · 7104ad6
1 parent 51f893e
commit 7104ad6
Show file tree

Hide file tree

Showing 5 changed files with 24 additions and 11 deletions.
diff --git a/plugins/openvino/.vscode/launch.json b/plugins/openvino/.vscode/launch.json
@@ -21,7 +21,7 @@
                 },
                 {
                     "localRoot": "${workspaceFolder}/src",
-                    "remoteRoot": "${config:scrypted.pythonRemoteRoot}"
+                    "remoteRoot": "."
                 },
 
             ]

diff --git a/plugins/openvino/.vscode/settings.json b/plugins/openvino/.vscode/settings.json
@@ -1,12 +1,12 @@
 
 {
     // docker installation
-    "scrypted.debugHost": "scrypted-demo",
-    "scrypted.serverRoot": "/server",
+    // "scrypted.debugHost": "scrypted-demo",
+    // "scrypted.serverRoot": "/server",
 
     // proxmox installation
-    // "scrypted.debugHost": "scrypted-server",
-    // "scrypted.serverRoot": "/root/.scrypted",
+    "scrypted.debugHost": "scrypted-server",
+    "scrypted.serverRoot": "/root/.scrypted",
 
     // pi local installation
     // "scrypted.debugHost": "192.168.2.119",
@@ -18,7 +18,7 @@
     // "scrypted.debugHost": "koushik-winvm",
     // "scrypted.serverRoot": "C:\\Users\\koush\\.scrypted",
 
-    "scrypted.pythonRemoteRoot": "${config:scrypted.serverRoot}/volume/plugin.zip",
+    "scrypted.pythonRemoteRoot": "${config:scrypted.serverRoot}/volume",
     "python.analysis.extraPaths": [
         "./node_modules/@scrypted/sdk/types/scrypted_python"
     ]

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.109"
+   "version": "0.1.110"
 }
diff --git a/plugins/openvino/src/ov/__init__.py b/plugins/openvino/src/ov/__init__.py
@@ -101,6 +101,8 @@ def __init__(self, nativeId: str | None = None):
         nvidia = False
         iris_xe = False
         arc = False
+        npu = False
+        gpu = False
 
         dgpus = []
         # search for NVIDIA dGPU, as that is not preferred by AUTO for some reason?
@@ -116,15 +118,26 @@ def __init__(self, nativeId: str | None = None):
                 if "NVIDIA" in full_device_name and "dGPU" in full_device_name:
                     dgpus.append(device)
                     nvidia = True
+                if "NPU" in device:
+                    npu = True
+                if "GPU" in device:
+                    gpu = True
             except:
                 pass
 
         mode = self.storage.getItem("mode")
         if mode == "Default":
             mode = "AUTO"
 
-            if len(dgpus):
+            if npu:
+                if gpu:
+                    mode = f"AUTO:NPU,GPU,CPU"
+                else:
+                    mode = f"AUTO:NPU,CPU"
+            elif len(dgpus):
                 mode = f"AUTO:{','.join(dgpus)},CPU"
+            elif gpu:
+                mode = f"GPU"
 
         mode = mode or "AUTO"
         self.mode = mode
@@ -137,7 +150,7 @@ def __init__(self, nativeId: str | None = None):
         if model == "Default" or model not in availableModels:
             if model != "Default":
                 self.storage.setItem("model", "Default")
-            if arc or nvidia:
+            if arc or nvidia or npu:
                 model = "scrypted_yolov9c_320"
             elif iris_xe:
                 model = "scrypted_yolov9s_320"