Skip to content

Commit

Permalink
[backend-comparison] Fix automatic fusion activation with wgpu (#1542)
Browse files Browse the repository at this point in the history
  • Loading branch information
syl20bnr committed Mar 27, 2024
1 parent 5f7fd59 commit 3a1d520
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions backend-comparison/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,8 @@ repository = "https://github.com/tracel-ai/burn/tree/main/backend-comparison"
version.workspace = true

[features]
default = ["burn/std", "burn/autodiff"]
# we depend on wgpu and autotune by default because we use the burn-wgpu crate to get system information
default = ["burn/std", "burn/autodiff", "burn/wgpu", "burn/autotune"]
candle-cpu = ["burn/candle"]
candle-cuda = ["burn/candle", "burn/cuda"]
candle-metal = ["burn/candle", "burn/metal"]
Expand All @@ -29,7 +30,7 @@ wgpu-fusion = ["wgpu", "burn/fusion"]
arboard = { workspace = true }
burn = { path = "../crates/burn", default-features = false }
burn-common = { path = "../crates/burn-common", version = "0.13.0" }
burn-wgpu = { path = "../crates/burn-wgpu", version = "0.13.0" }
burn-wgpu = { path = "../crates/burn-wgpu", default-features = false, version = "0.13.0" }
clap = { workspace = true }
crossterm = { workspace = true, optional = true }
derive-new = { workspace = true }
Expand Down

0 comments on commit 3a1d520

Please sign in to comment.