Add check for WinML availability.

crates/wasi-nn/src/testing.rs

@@ -3,10 +3,13 @@
 //!
 //! This module checks:
 //! - that OpenVINO can be found in the environment
+//! - that WinML is available
 //! - that some ML model artifacts can be downloaded and cached.
 
 use anyhow::{anyhow, Context, Result};
 use std::{env, fs, path::Path, path::PathBuf, process::Command, sync::Mutex};
+#[cfg(feature = "winml")]
+use windows::AI::MachineLearning::{LearningModelDevice, LearningModelDeviceKind};
 
 /// Return the directory in which the test artifacts are stored.
 pub fn artifacts_dir() -> PathBuf {
@@ -41,6 +44,7 @@ pub fn check() -> Result<()> {
     }
     #[cfg(feature = "winml")]
     {
+        check_winml_is_available()?;
         check_winml_artifacts_are_available()?;
     }
     Ok(())
@@ -56,6 +60,19 @@ fn check_openvino_is_installed() -> Result<()> {
     }
 }
 
+#[cfg(feature = "winml")]
+fn check_winml_is_available() -> Result<()> {
+    match std::panic::catch_unwind(|| {
+        println!(
+            "> WinML learning device is available: {:?}",
+            LearningModelDevice::Create(LearningModelDeviceKind::Default)
+        )
+    }) {
+        Ok(_) => Ok(()),
+        Err(e) => Err(anyhow!("WinML learning device is not available: {:?}", e)),
+    }
+}
+
 /// Protect `check_openvino_artifacts_are_available` from concurrent access;
 /// when running tests in parallel, we want to avoid two threads attempting to
 /// create the same directory or download the same file.

crates/wasi-nn/tests/all.rs

@@ -31,8 +31,11 @@ fn run(path: &str, preload_model: bool) -> Result<()> {
     {
         backends.push(Backend::from(backend::winml::WinMLBackend::default()));
     }
-    for backend in backends{
-        let mut store = Store::new(&engine, Ctx::new(&testing::artifacts_dir(), preload_model,backend)?);
+    for backend in backends {
+        let mut store = Store::new(
+            &engine,
+            Ctx::new(&testing::artifacts_dir(), preload_model, backend)?,
+        );
         let instance = linker.instantiate(&mut store, &module)?;
         let start = instance.get_typed_func::<(), ()>(&mut store, "_start")?;
         start.call(&mut store, ())?;