mirror of https://github.com/tracel-ai/burn.git
Chore: Update CubeCL (#2292)
This commit is contained in:
parent
aa79e36a8d
commit
20ab5e31d7
|
@ -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",
|
||||||
|
|
|
@ -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" }
|
||||||
|
|
|
@ -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 {
|
||||||
|
|
Loading…
Reference in New Issue