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 (#2219)

This commit is contained in:
Nathaniel Simard 2024-08-30 15:28:00 -04:00 committed by GitHub
parent a9abd8f746
commit 0dbb7f7e91
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 23 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
Cargo.lock generated
View File

@ -838,7 +838,7 @@ dependencies = [
"byteorder",
"candle-kernels",
"candle-metal-kernels",
"cudarc",
"cudarc 0.11.5",
"gemm",
"half",
"libc",
@ -1395,8 +1395,7 @@ dependencies = [
[[package]]
name = "cubecl"
version = "0.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "222d7052864d80ae9009a51ad6abfd5fb141a4ccf7e1d061888cc11c3c674de3"
source = "git+https://github.com/tracel-ai/cubecl?rev=446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e#446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e"
dependencies = [
"cubecl-core",
"cubecl-cuda",
@ -1407,8 +1406,7 @@ dependencies = [
[[package]]
name = "cubecl-common"
version = "0.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a6e358cb0b4de9f602e1b6f6656d88f351cc29664536cdb54ea17f7e44f7a393"
source = "git+https://github.com/tracel-ai/cubecl?rev=446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e#446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e"
dependencies = [
"derive-new",
"getrandom",
@ -1423,8 +1421,7 @@ dependencies = [
[[package]]
name = "cubecl-core"
version = "0.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "719df0318ac22c09f16801225c205780615e97ecf79fd07fdb0e08aa32d332f6"
source = "git+https://github.com/tracel-ai/cubecl?rev=446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e#446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e"
dependencies = [
"bytemuck",
"cubecl-macros",
@ -1439,14 +1436,13 @@ dependencies = [
[[package]]
name = "cubecl-cuda"
version = "0.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b9491684e95c27fd989dcefdcaff5047afd50b0dcf3d9f540c0c0c7d85815f9a"
source = "git+https://github.com/tracel-ai/cubecl?rev=446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e#446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e"
dependencies = [
"bytemuck",
"cubecl-common",
"cubecl-core",
"cubecl-runtime",
"cudarc",
"cudarc 0.12.0",
"derive-new",
"half",
"log",
@ -1455,8 +1451,7 @@ dependencies = [
[[package]]
name = "cubecl-linalg"
version = "0.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ad47dab1a1347cb0df322eaddfff82441890760eb367dd2afd7f5cd463c88193"
source = "git+https://github.com/tracel-ai/cubecl?rev=446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e#446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e"
dependencies = [
"bytemuck",
"cubecl-core",
@ -1467,8 +1462,7 @@ dependencies = [
[[package]]
name = "cubecl-macros"
version = "0.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b02de789371373c9e09924b93ce1f7248c42e2dac7f4bc7be4872e3a18025daf"
source = "git+https://github.com/tracel-ai/cubecl?rev=446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e#446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e"
dependencies = [
"derive-new",
"proc-macro2",
@ -1479,8 +1473,7 @@ dependencies = [
[[package]]
name = "cubecl-runtime"
version = "0.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "2fab9586f79f8a4db35a16556575217786f743b2d1c030cd4515174b3dfd8817"
source = "git+https://github.com/tracel-ai/cubecl?rev=446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e#446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e"
dependencies = [
"async-channel",
"cfg_aliases 0.2.1",
@ -1500,8 +1493,7 @@ dependencies = [
[[package]]
name = "cubecl-wgpu"
version = "0.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "83293243697e7270e3c0e3354cd75cfe0a2745f6305bda5c3637248b0d368afc"
source = "git+https://github.com/tracel-ai/cubecl?rev=446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e#446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e"
dependencies = [
"async-channel",
"bytemuck",
@ -1526,6 +1518,15 @@ dependencies = [
"libloading",
]
[[package]]
name = "cudarc"
version = "0.12.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e736ec4e22548a5677d65fe191301c243cd63365db2e2def0e263b176f51531c"
dependencies = [
"libloading",
]
[[package]]
name = "custom-csv-dataset"
version = "0.15.0"

8
Cargo.toml
View File

@ -151,14 +151,14 @@ systemstat = "0.2.3"
portable-atomic-util = { version = "0.2.2", features = ["alloc"] }
### For the main burn branch. ###
# cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "d41dd0c00381fb935e4389927ee4bef0b8aba785" }
# cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "d41dd0c00381fb935e4389927ee4bef0b8aba785" }
cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e" }
cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "446dc854ef3f7f93fe4ab2df0f1ab779a7d6a98e" }
### For local development. ###
# cubecl = { path = "../cubecl/crates/cubecl" }
# cubecl-common = { path = "../cubecl/crates/cubecl-common" }
### For the release. ###
cubecl = { version="0.2.0", default-features = false }
cubecl-common = { version="0.2.0", default-features = false }
# cubecl = { version="0.2.0", default-features = false }
# cubecl-common = { version="0.2.0", default-features = false }
### For xtask crate ###
tracel-xtask = { git = "https://github.com/tracel-ai/xtask", rev = "921408bc16e74d3ef8ae59356d928fb6706fb8f4" }

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

@ -544,21 +544,6 @@ impl TraceBuilder {
&mut local_tensor_ids_input,
&mut local_tensor_ids_output,
),
Subcube::And(op) => mark_unary(
op,
&mut local_tensor_ids_input,
&mut local_tensor_ids_output,
),
Subcube::Or(op) => mark_unary(
op,
&mut local_tensor_ids_input,
&mut local_tensor_ids_output,
),
Subcube::Xor(op) => mark_unary(
op,
&mut local_tensor_ids_input,
&mut local_tensor_ids_output,
),
Subcube::Min(op) => mark_unary(
op,
&mut local_tensor_ids_input,