dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[backend-comparison] Fix automatic fusion activation with wgpu (#1542)

This commit is contained in:
Sylvain Benner 2024-03-27 17:40:12 -04:00 committed by GitHub
parent 5f7fd5952a
commit 3a1d52067f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend-comparison/Cargo.toml
View File

@ -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"]
@ -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 }