From e6126c8fc57f66d5f26ba6dea695519b4c8b5e8d Mon Sep 17 00:00:00 2001
From: Genna Wingert <1058083+wingertge@users.noreply.github.com>
Date: Mon, 3 Feb 2025 17:35:26 +0100
Subject: [PATCH] Add vision tests to xtask

---
 xtask/src/commands/test.rs | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/xtask/src/commands/test.rs b/xtask/src/commands/test.rs
index 5b94b2909e..ff0581089c 100644
--- a/xtask/src/commands/test.rs
+++ b/xtask/src/commands/test.rs
@@ -67,6 +67,15 @@ pub(crate) fn handle_command(
                 "std with features: test-tch,record-item-custom-serde",
             )?;
 
+            // burn-vision
+            helpers::custom_crates_tests(
+                vec!["burn-vision"],
+                vec!["--features", "cpu"],
+                None,
+                None,
+                "std cpu",
+            )?;
+
             if std::env::var("DISABLE_WGPU").is_err() {
                 helpers::custom_crates_tests(
                     vec!["burn-core"],
@@ -75,6 +84,13 @@ pub(crate) fn handle_command(
                     None,
                     "std wgpu",
                 )?;
+                helpers::custom_crates_tests(
+                    vec!["burn-vision"],
+                    vec!["--features", "wgpu"],
+                    None,
+                    None,
+                    "std wgpu",
+                )?;
                 // Vulkan isn't available on MacOS
                 #[cfg(not(target_os = "macos"))]
                 if std::env::var("DISABLE_WGPU_SPIRV").is_err() {
@@ -85,6 +101,13 @@ pub(crate) fn handle_command(
                         None,
                         "std vulkan",
                     )?;
+                    helpers::custom_crates_tests(
+                        vec!["burn-vision"],
+                        vec!["--features", "vulkan"],
+                        None,
+                        None,
+                        "std vulkan",
+                    )?;
                 }
             }