Add vision tests to xtask

tracel-ai · Feb 3, 2025 · e6126c8 · e6126c8
1 parent ee5ad73
commit e6126c8
Showing 1 changed file with 23 additions and 0 deletions.
diff --git 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",
+                    )?;
                 }
             }