Fix burn-vision backend features

Author: laggui

Cargo.lock

@@ -33,13 +33,14 @@ tracing = [
 cubecl-backend = ["cubecl", "burn-cubecl"]
 fusion = ["burn-fusion", "burn-cuda/fusion", "burn-wgpu/fusion"]
 
+# TODO: should remove backend crate deps and use re-exported types from burn-core
 cuda = ["cubecl-backend", "burn-core/cuda", "burn-cuda"]
-# rocm = ["cubecl-backend", "burn-core/rocm", "burn-rocm"]
+rocm = ["cubecl-backend", "burn-core/rocm", "burn-rocm"]
 wgpu = ["cubecl-backend", "burn-core/wgpu", "burn-wgpu"]
-# vulkan = ["wgpu", "burn-core/vulkan", "burn-wgpu/vulkan"]
-# webgpu = ["wgpu", "burn-core/webgpu", "burn-wgpu/webgpu"]
-# metal = ["wgpu", "burn-core/metal", "burn-wgpu/metal"]
-# cpu = ["cubecl-backend", "burn-core/cpu", "burn-cpu"]
+vulkan = ["wgpu", "burn-core/vulkan", "burn-wgpu/vulkan"]
+webgpu = ["wgpu", "burn-core/webgpu", "burn-wgpu/webgpu"]
+metal = ["wgpu", "burn-core/metal", "burn-wgpu/metal"]
+cpu = ["cubecl-backend", "burn-core/cpu", "burn-cpu"]
 flex = ["burn-core/flex", "burn-flex"]
 tch = ["burn-core/tch", "burn-tch"]
 
@@ -73,7 +74,9 @@ paste = { workspace = true }
 serde = { workspace = true }
 
 # Backends
+burn-cpu = { workspace = true, optional = true }
 burn-cuda = { workspace = true, optional = true }
 burn-wgpu = { workspace = true, optional = true }
+burn-rocm = { workspace = true, optional = true }
 burn-flex = { workspace = true, optional = true, features = ["default"]}
 burn-tch = { workspace = true, optional = true, features = ["default"]}

crates/burn-vision/Cargo.toml

@@ -341,7 +341,7 @@ pub(crate) fn handle_command(
                     )?;
                     helpers::custom_crates_tests(
                         vec!["burn-vision"],
-                        handle_test_args(&[], args.release),
+                        handle_test_args(&["--features", "metal"], args.release),
                         None,
                         None,
                         "std metal",