From e23c9c22dcd4cf407ca6d05fa6d1665d56d34239 Mon Sep 17 00:00:00 2001
From: Koushik Dutta <koushd@gmail.com>
Date: Thu, 5 Sep 2024 10:34:22 -0700
Subject: [PATCH] onnx: fix platform check

---
 plugins/onnx/package-lock.json   | 4 ++--
 plugins/onnx/package.json        | 2 +-
 plugins/onnx/src/ort/__init__.py | 2 +-
 server/package-lock.json         | 4 ++--
 4 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/plugins/onnx/package-lock.json b/plugins/onnx/package-lock.json
index 0b98cb625..87f787903 100644
--- a/plugins/onnx/package-lock.json
+++ b/plugins/onnx/package-lock.json
@@ -1,12 +1,12 @@
 {
    "name": "@scrypted/onnx",
-   "version": "0.1.109",
+   "version": "0.1.110",
    "lockfileVersion": 2,
    "requires": true,
    "packages": {
       "": {
          "name": "@scrypted/onnx",
-         "version": "0.1.109",
+         "version": "0.1.110",
          "devDependencies": {
             "@scrypted/sdk": "file:../../sdk"
          }
diff --git a/plugins/onnx/package.json b/plugins/onnx/package.json
index aca01bd29..748692256 100644
--- a/plugins/onnx/package.json
+++ b/plugins/onnx/package.json
@@ -42,5 +42,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.109"
+   "version": "0.1.110"
 }
diff --git a/plugins/onnx/src/ort/__init__.py b/plugins/onnx/src/ort/__init__.py
index b1b9d341f..fd6f2044c 100644
--- a/plugins/onnx/src/ort/__init__.py
+++ b/plugins/onnx/src/ort/__init__.py
@@ -94,7 +94,7 @@ class ONNXPlugin(
                 if sys.platform == 'darwin':
                     providers.append("CoreMLExecutionProvider")
 
-                if ('linux' in sys.platform or 'win' in sys.platform) and platform.machine() == 'x86_64':
+                if ('linux' in sys.platform or 'win' in sys.platform) and (platform.machine() == 'x86_64' or platform.machine() == 'AMD64'):
                     deviceId = int(deviceId)
                     providers.append(("CUDAExecutionProvider", { "device_id": deviceId }))
 
diff --git a/server/package-lock.json b/server/package-lock.json
index e04c92588..1eb4cb951 100644
--- a/server/package-lock.json
+++ b/server/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "@scrypted/server",
-  "version": "0.117.2",
+  "version": "0.117.3",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "@scrypted/server",
-      "version": "0.117.2",
+      "version": "0.117.3",
       "hasInstallScript": true,
       "license": "ISC",
       "dependencies": {