wasi-nn: add feature to use custom ONNX Runtime (#11060)

* wasi-nn: add feature to use custom ONNX Runtime

* Change to onnx-download feature

* Update onnx feature in CI, prtest:full

Author: yanwucai

.github/workflows/main.yml

@@ -778,7 +778,7 @@ jobs:
   test_wasi_nn:
     strategy:
       matrix:
-        feature: ["openvino", "onnx"]
+        feature: ["openvino", "onnx-download"]
         os: ["ubuntu-24.04", "windows-2025"]
         include:
           - os: windows-2025

crates/wasi-nn/Cargo.toml

@@ -33,7 +33,6 @@ thiserror = { workspace = true }
 
 ort = { version = "2.0.0-rc.2", default-features = false, features = [
     "copy-dylibs",
-    "download-binaries",
 ], optional = true }
 tch = { version = "0.17.0", default-features = false, optional = true}
 
@@ -70,6 +69,8 @@ default = ["openvino", "winml"]
 openvino = ["dep:openvino"]
 # ONNX is available on all platforms.
 onnx = ["dep:ort"]
+# Use prebuilt ONNX Runtime binaries from ort.
+onnx-download = ["onnx", "ort/download-binaries"]
 # WinML is only available on Windows 10 1809 and later.
 winml = ["dep:windows"]
 # PyTorch is available on all platforms; requires Libtorch to be installed

crates/wasi-nn/examples/classification-component-onnx/README.md

@@ -12,7 +12,7 @@ cargo component build
 In the Wasmtime root directory, run the following command to build the Wasmtime CLI and run the WebAssembly component:
 ```sh
 # build wasmtime with component-model and WASI-NN with ONNX runtime support
-cargo build --features component-model,wasi-nn,wasmtime-wasi-nn/onnx
+cargo build --features component-model,wasi-nn,wasmtime-wasi-nn/onnx-download
 
 # run the component with wasmtime
 ./target/debug/wasmtime run \