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

Chore: Update CubeCL (#2292)

This commit is contained in:
Nathaniel Simard 2024-09-21 13:28:07 -04:00 committed by GitHub
parent aa79e36a8d
commit 20ab5e31d7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 15 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
Cargo.lock generated
View File

@ -1397,7 +1397,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl" name = "cubecl"
version = "0.2.0" version = "0.2.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b#0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b" source = "git+https://github.com/tracel-ai/cubecl?rev=447968efed59e8659c255f1d84d82039b533b464#447968efed59e8659c255f1d84d82039b533b464"
dependencies = [ dependencies = [
"cubecl-core", "cubecl-core",
"cubecl-cuda", "cubecl-cuda",
@ -1408,7 +1408,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-common" name = "cubecl-common"
version = "0.2.0" version = "0.2.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b#0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b" source = "git+https://github.com/tracel-ai/cubecl?rev=447968efed59e8659c255f1d84d82039b533b464#447968efed59e8659c255f1d84d82039b533b464"
dependencies = [ dependencies = [
"derive-new", "derive-new",
"getrandom", "getrandom",
@ -1423,7 +1423,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-core" name = "cubecl-core"
version = "0.2.0" version = "0.2.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b#0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b" source = "git+https://github.com/tracel-ai/cubecl?rev=447968efed59e8659c255f1d84d82039b533b464#447968efed59e8659c255f1d84d82039b533b464"
dependencies = [ dependencies = [
"bytemuck", "bytemuck",
"cubecl-common", "cubecl-common",
@ -1440,7 +1440,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-cuda" name = "cubecl-cuda"
version = "0.2.0" version = "0.2.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b#0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b" source = "git+https://github.com/tracel-ai/cubecl?rev=447968efed59e8659c255f1d84d82039b533b464#447968efed59e8659c255f1d84d82039b533b464"
dependencies = [ dependencies = [
"bytemuck", "bytemuck",
"cubecl-common", "cubecl-common",
@ -1455,7 +1455,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-linalg" name = "cubecl-linalg"
version = "0.2.0" version = "0.2.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b#0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b" source = "git+https://github.com/tracel-ai/cubecl?rev=447968efed59e8659c255f1d84d82039b533b464#447968efed59e8659c255f1d84d82039b533b464"
dependencies = [ dependencies = [
"bytemuck", "bytemuck",
"cubecl-core", "cubecl-core",
@ -1466,7 +1466,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-macros" name = "cubecl-macros"
version = "0.2.0" version = "0.2.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b#0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b" source = "git+https://github.com/tracel-ai/cubecl?rev=447968efed59e8659c255f1d84d82039b533b464#447968efed59e8659c255f1d84d82039b533b464"
dependencies = [ dependencies = [
"cubecl-common", "cubecl-common",
"darling", "darling",
@ -1481,7 +1481,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-runtime" name = "cubecl-runtime"
version = "0.2.0" version = "0.2.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b#0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b" source = "git+https://github.com/tracel-ai/cubecl?rev=447968efed59e8659c255f1d84d82039b533b464#447968efed59e8659c255f1d84d82039b533b464"
dependencies = [ dependencies = [
"async-channel", "async-channel",
"cfg_aliases 0.2.1", "cfg_aliases 0.2.1",
@ -1501,7 +1501,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-wgpu" name = "cubecl-wgpu"
version = "0.2.0" version = "0.2.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b#0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b" source = "git+https://github.com/tracel-ai/cubecl?rev=447968efed59e8659c255f1d84d82039b533b464#447968efed59e8659c255f1d84d82039b533b464"
dependencies = [ dependencies = [
"async-channel", "async-channel",
"bytemuck", "bytemuck",

4
Cargo.toml
View File

@ -151,8 +151,8 @@ tch = "0.15.0"
portable-atomic-util = { version = "0.2.2", features = ["alloc"] } portable-atomic-util = { version = "0.2.2", features = ["alloc"] }
### For the main burn branch. ### ### For the main burn branch. ###
cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b" } cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "447968efed59e8659c255f1d84d82039b533b464" }
cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "0c494b6d6ec3cc28635e0c6d7a4935008c3c8b8b" } cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "447968efed59e8659c255f1d84d82039b533b464" }
### For local development. ### ### For local development. ###
# cubecl = { path = "../cubecl/crates/cubecl" } # cubecl = { path = "../cubecl/crates/cubecl" }
# cubecl-common = { path = "../cubecl/crates/cubecl-common" } # cubecl-common = { path = "../cubecl/crates/cubecl-common" }

5
crates/burn-jit/src/fusion/tracing/builder.rs
View File

@ -514,6 +514,11 @@ impl TraceBuilder {
Operator::AtomicCompareAndSwap(_op) => { Operator::AtomicCompareAndSwap(_op) => {
// Nothing to do. // Nothing to do.
} }
Operator::Magnitude(op) => mark_unary(
op,
&mut local_tensor_ids_input,
&mut local_tensor_ids_output,
),
}, },
Operation::Procedure(proc) => { Operation::Procedure(proc) => {
match proc { match proc {