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

WIP

This commit is contained in:
nathaniel 2024-08-02 14:50:51 -04:00
parent bb13729b20
commit f04fda4335
5 changed files with 8 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Cargo.lock generated
View File

@ -1378,7 +1378,6 @@ dependencies = [
[[package]]
name = "cubecl"
version = "0.1.1"
source = "git+https://github.com/tracel-ai/cubecl?rev=dd274b6515f853ab1c82db410d6fc7697060f69e#dd274b6515f853ab1c82db410d6fc7697060f69e"
dependencies = [
"cubecl-core",
"cubecl-cuda",
@ -1389,7 +1388,6 @@ dependencies = [
[[package]]
name = "cubecl-common"
version = "0.1.1"
source = "git+https://github.com/tracel-ai/cubecl?rev=dd274b6515f853ab1c82db410d6fc7697060f69e#dd274b6515f853ab1c82db410d6fc7697060f69e"
dependencies = [
"derive-new",
"getrandom",
@ -1403,7 +1401,6 @@ dependencies = [
[[package]]
name = "cubecl-core"
version = "0.1.1"
source = "git+https://github.com/tracel-ai/cubecl?rev=dd274b6515f853ab1c82db410d6fc7697060f69e#dd274b6515f853ab1c82db410d6fc7697060f69e"
dependencies = [
"bytemuck",
"cubecl-macros",
@ -1418,7 +1415,6 @@ dependencies = [
[[package]]
name = "cubecl-cuda"
version = "0.1.1"
source = "git+https://github.com/tracel-ai/cubecl?rev=dd274b6515f853ab1c82db410d6fc7697060f69e#dd274b6515f853ab1c82db410d6fc7697060f69e"
dependencies = [
"bytemuck",
"cubecl-common",
@ -1433,7 +1429,6 @@ dependencies = [
[[package]]
name = "cubecl-linalg"
version = "0.1.1"
source = "git+https://github.com/tracel-ai/cubecl?rev=dd274b6515f853ab1c82db410d6fc7697060f69e#dd274b6515f853ab1c82db410d6fc7697060f69e"
dependencies = [
"bytemuck",
"cubecl-core",
@ -1444,7 +1439,6 @@ dependencies = [
[[package]]
name = "cubecl-macros"
version = "0.1.1"
source = "git+https://github.com/tracel-ai/cubecl?rev=dd274b6515f853ab1c82db410d6fc7697060f69e#dd274b6515f853ab1c82db410d6fc7697060f69e"
dependencies = [
"derive-new",
"proc-macro2",
@ -1455,7 +1449,6 @@ dependencies = [
[[package]]
name = "cubecl-runtime"
version = "0.1.1"
source = "git+https://github.com/tracel-ai/cubecl?rev=dd274b6515f853ab1c82db410d6fc7697060f69e#dd274b6515f853ab1c82db410d6fc7697060f69e"
dependencies = [
"async-channel",
"cubecl-common",
@ -1474,7 +1467,6 @@ dependencies = [
[[package]]
name = "cubecl-wgpu"
version = "0.1.1"
source = "git+https://github.com/tracel-ai/cubecl?rev=dd274b6515f853ab1c82db410d6fc7697060f69e#dd274b6515f853ab1c82db410d6fc7697060f69e"
dependencies = [
"async-channel",
"bytemuck",

8
Cargo.toml
View File

@ -144,11 +144,11 @@ sysinfo = "0.30.13"
systemstat = "0.2.3"
### For the main burn branch. ###
cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "dd274b6515f853ab1c82db410d6fc7697060f69e" }
cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "dd274b6515f853ab1c82db410d6fc7697060f69e" }
# cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "dd274b6515f853ab1c82db410d6fc7697060f69e" }
# cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "dd274b6515f853ab1c82db410d6fc7697060f69e" }
### For local development. ###
# cubecl = { path = "../cubecl/crates/cubecl" }
# cubecl-common = { path = "../cubecl/crates/cubecl-common" }
cubecl = { path = "../cubecl/crates/cubecl" }
cubecl-common = { path = "../cubecl/crates/cubecl-common" }
[profile.dev]
debug = 0 # Speed up compilation time and not necessary.

2
crates/burn-train/Cargo.toml
View File

@ -17,7 +17,7 @@ metrics = ["nvml-wrapper", "sysinfo", "systemstat"]
tui = ["ratatui", "crossterm"]
[dependencies]
burn-core = { path = "../burn-core", version = "0.14.0", features = ["dataset"] }
burn-core = { path = "../burn-core", version = "0.14.0", features = ["dataset"], default-features = false }
log = { workspace = true }
tracing-subscriber = { workspace = true }

4
crates/burn-wgpu/Cargo.toml
View File

@ -11,12 +11,12 @@ repository = "https://github.com/tracel-ai/burn/tree/main/crates/burn-wgpu"
version.workspace = true
[features]
default = ["fusion", "burn-jit/default", "cubecl/default"]
default = ["std", "autotune", "fusion", "burn-jit/default", "cubecl/default"]
fusion = ["burn-fusion", "burn-jit/fusion"]
autotune = ["burn-jit/autotune"]
template = ["burn-jit/template", "cubecl/template"]
doc = ["burn-jit/doc"]
std = ["burn-jit/std"]
std = ["burn-jit/std", "cubecl/std"]
[dependencies]
cubecl = { workspace = true, features = ["wgpu"] }

2
examples/text-classification/Cargo.toml
View File

@ -19,7 +19,7 @@ wgpu = ["burn/wgpu"]
[dependencies]
# Burn
burn = {path = "../../crates/burn", features=["train", "ndarray", "fusion"]}
burn = {path = "../../crates/burn", features=["train", "ndarray", "std", "tui", "metrics", "autotune", "fusion", "default"], default-features = false}
# Tokenizer
tokenizers = { version = "0.19.1", default-features = false, features = [