feat: add cuda

MaaXYZ · Nov 28, 2024 · ebf85fa · ebf85fa
1 parent 55ca480
commit ebf85fa
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/vcpkg-overlay/ports/maa-onnxruntime/vcpkg.json b/vcpkg-overlay/ports/maa-onnxruntime/vcpkg.json
@@ -85,6 +85,18 @@
     },
     "coreml": {
       "description": "Build with CoreML support"
+    },
+    "tensorrt": {
+      "description": "Build with TensorRT support",
+      "dependencies": [
+        {
+          "name": "onnxruntime",
+          "default-features": false,
+          "features": [
+            "cuda"
+          ]
+        }
+      ]
     }
   }
 }
diff --git a/vcpkg.json b/vcpkg.json
@@ -20,7 +20,7 @@
     {
       "name": "maa-onnxruntime",
       "platform": "windows",
-      "features": ["directml"]
+      "features": ["directml", "cuda"]
     },
     {
       "name": "maa-onnxruntime",
@@ -29,7 +29,8 @@
     },
     {
       "name": "maa-onnxruntime",
-      "platform": "linux"
+      "platform": "linux",
+      "features": ["cuda"]
     },
     "maa-fastdeploy",
     "zlib",