From 9779ef44b81cef1eb7bff2088ca9be23c1de784b Mon Sep 17 00:00:00 2001
From: AsherJingkongChen <37398747+AsherJingkongChen@users.noreply.github.com>
Date: Tue, 29 Oct 2024 10:48:15 +0800
Subject: [PATCH 1/7] Refactor for wgpu v23 compatibility

---
 Cargo.lock                    | 280 +++++++++++++++++-----------------
 Cargo.toml                    |   6 +-
 crates/burn-tensor/src/lib.rs |   6 +-
 3 files changed, 142 insertions(+), 150 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index 803c70aa355..d260764506b 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -394,18 +394,18 @@ dependencies = [
 
 [[package]]
 name = "bit-set"
-version = "0.6.0"
+version = "0.8.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f0481a0e032742109b1133a095184ee93d88f3dc9e0d28a5d033dc77a073f44f"
+checksum = "08807e080ed7f9d5433fa9b275196cfc35414f66a0c79d864dc51a0d825231a3"
 dependencies = [
  "bit-vec",
 ]
 
 [[package]]
 name = "bit-vec"
-version = "0.7.0"
+version = "0.8.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d2c54ff287cfc0a34f38a6b832ea1bd8e448a330b3e40a50859e6488bee07f22"
+checksum = "5e764a1d40d510daf35e07be9eb06e75770908c27d411ee6c92109c9840eaaf7"
 
 [[package]]
 name = "bit_field"
@@ -524,7 +524,7 @@ dependencies = [
 name = "burn-common"
 version = "0.16.0"
 dependencies = [
- "cubecl-common",
+ "cubecl-common 0.3.0",
  "dashmap",
  "getrandom",
  "indicatif",
@@ -1165,37 +1165,6 @@ dependencies = [
  "windows-sys 0.48.0",
 ]
 
-[[package]]
-name = "com"
-version = "0.6.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "7e17887fd17353b65b1b2ef1c526c83e26cd72e74f598a8dc1bee13a48f3d9f6"
-dependencies = [
- "com_macros",
-]
-
-[[package]]
-name = "com_macros"
-version = "0.6.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d375883580a668c7481ea6631fc1a8863e33cc335bf56bfad8d7e6d4b04b13a5"
-dependencies = [
- "com_macros_support",
- "proc-macro2",
- "syn 1.0.109",
-]
-
-[[package]]
-name = "com_macros_support"
-version = "0.6.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "ad899a1087a9296d5644792d7cb72b8e34c1bec8e7d4fbc002230169a6e8710c"
-dependencies = [
- "proc-macro2",
- "quote",
- "syn 1.0.109",
-]
-
 [[package]]
 name = "comfy-table"
 version = "7.1.1"
@@ -1459,15 +1428,13 @@ dependencies = [
 
 [[package]]
 name = "cubecl"
-version = "0.3.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "75e75c7e982b943380665c5901fe0b69d5df2627644e0e50199c52b64d8d5a1c"
+version = "0.4.0"
 dependencies = [
  "cubecl-core",
  "cubecl-cuda",
  "cubecl-hip",
  "cubecl-linalg",
- "cubecl-runtime",
+ "cubecl-runtime 0.4.0",
  "cubecl-wgpu",
 ]
 
@@ -1489,16 +1456,30 @@ dependencies = [
  "web-time",
 ]
 
+[[package]]
+name = "cubecl-common"
+version = "0.4.0"
+dependencies = [
+ "derive-new",
+ "embassy-futures",
+ "futures-lite",
+ "getrandom",
+ "log",
+ "portable-atomic",
+ "rand",
+ "serde",
+ "spin",
+ "web-time",
+]
+
 [[package]]
 name = "cubecl-core"
-version = "0.3.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "ec33b64139d1dfc747df8aed5834d10c3c55c716f5219041c6eb17241c96c929"
+version = "0.4.0"
 dependencies = [
  "bytemuck",
- "cubecl-common",
+ "cubecl-common 0.4.0",
  "cubecl-macros",
- "cubecl-runtime",
+ "cubecl-runtime 0.4.0",
  "derive-new",
  "half",
  "log",
@@ -1509,14 +1490,12 @@ dependencies = [
 
 [[package]]
 name = "cubecl-cpp"
-version = "0.3.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "4ded461feb0ff342a4f675131dc0ae8ad94e58f66bad11e57f852cb7f190a731"
+version = "0.4.0"
 dependencies = [
  "bytemuck",
- "cubecl-common",
+ "cubecl-common 0.4.0",
  "cubecl-core",
- "cubecl-runtime",
+ "cubecl-runtime 0.4.0",
  "derive-new",
  "half",
  "log",
@@ -1524,15 +1503,13 @@ dependencies = [
 
 [[package]]
 name = "cubecl-cuda"
-version = "0.3.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "88dfdfe616124d2abe5e82052ff56f86843c369440e181d6936f7409e161dd82"
+version = "0.4.0"
 dependencies = [
  "bytemuck",
- "cubecl-common",
+ "cubecl-common 0.4.0",
  "cubecl-core",
  "cubecl-cpp",
- "cubecl-runtime",
+ "cubecl-runtime 0.4.0",
  "cudarc 0.12.1",
  "derive-new",
  "half",
@@ -1541,16 +1518,14 @@ dependencies = [
 
 [[package]]
 name = "cubecl-hip"
-version = "0.3.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "409e0e176152ab51a60bbebb940b7a72aba210cd42b5f8cd2e87e7d7e674a13a"
+version = "0.4.0"
 dependencies = [
  "bytemuck",
- "cubecl-common",
+ "cubecl-common 0.4.0",
  "cubecl-core",
  "cubecl-cpp",
  "cubecl-hip-sys",
- "cubecl-runtime",
+ "cubecl-runtime 0.4.0",
  "derive-new",
  "half",
  "log",
@@ -1567,23 +1542,19 @@ dependencies = [
 
 [[package]]
 name = "cubecl-linalg"
-version = "0.3.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "3c5634782d790e9b6562fc267ffd15e9a510b4d6ec32c144cd2b166af2ba0cfb"
+version = "0.4.0"
 dependencies = [
  "bytemuck",
  "cubecl-core",
- "cubecl-runtime",
+ "cubecl-runtime 0.4.0",
  "half",
 ]
 
 [[package]]
 name = "cubecl-macros"
-version = "0.3.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d2d22663257d9cdbcd67f5048d6f4e6eb965dd87104c3a173a7b0ea0d720e99b"
+version = "0.4.0"
 dependencies = [
- "cubecl-common",
+ "cubecl-common 0.4.0",
  "darling",
  "derive-new",
  "ident_case",
@@ -1595,11 +1566,9 @@ dependencies = [
 
 [[package]]
 name = "cubecl-opt"
-version = "0.3.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "59fba2561a6ceb99e9c5fe7313db0aeead02b848dc9cdacf8373e0fe98d3c247"
+version = "0.4.0"
 dependencies = [
- "cubecl-common",
+ "cubecl-common 0.4.0",
  "cubecl-core",
  "float-ord",
  "log",
@@ -1618,7 +1587,27 @@ dependencies = [
  "async-channel",
  "async-lock",
  "cfg_aliases 0.2.1",
- "cubecl-common",
+ "cubecl-common 0.3.0",
+ "derive-new",
+ "dirs 5.0.1",
+ "hashbrown 0.14.5",
+ "log",
+ "md5",
+ "sanitize-filename",
+ "serde",
+ "serde_json",
+ "spin",
+ "wasm-bindgen-futures",
+]
+
+[[package]]
+name = "cubecl-runtime"
+version = "0.4.0"
+dependencies = [
+ "async-channel",
+ "async-lock",
+ "cfg_aliases 0.2.1",
+ "cubecl-common 0.4.0",
  "derive-new",
  "dirs 5.0.1",
  "hashbrown 0.14.5",
@@ -1633,31 +1622,27 @@ dependencies = [
 
 [[package]]
 name = "cubecl-spirv"
-version = "0.3.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "835bc234cdd40fbb5e3e5e41bfb4a6e2ee2d7fd899b66d44dcbcb3a825ca1a59"
+version = "0.4.0"
 dependencies = [
- "cubecl-common",
+ "cubecl-common 0.4.0",
  "cubecl-core",
  "cubecl-opt",
- "cubecl-runtime",
+ "cubecl-runtime 0.4.0",
  "hashbrown 0.14.5",
  "rspirv",
 ]
 
 [[package]]
 name = "cubecl-wgpu"
-version = "0.3.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "6779f1072d70923758421c6214fd0cd19a6f25b91035a522f9cd9407d03b5cae"
+version = "0.4.0"
 dependencies = [
  "ash",
  "async-channel",
  "bytemuck",
  "cfg_aliases 0.2.1",
- "cubecl-common",
+ "cubecl-common 0.4.0",
  "cubecl-core",
- "cubecl-runtime",
+ "cubecl-runtime 0.4.0",
  "cubecl-spirv",
  "derive-new",
  "hashbrown 0.14.5",
@@ -1753,17 +1738,6 @@ dependencies = [
  "serde",
 ]
 
-[[package]]
-name = "d3d12"
-version = "22.0.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "bdbd1f579714e3c809ebd822c81ef148b1ceaeb3d535352afc73fd0c4c6a0017"
-dependencies = [
- "bitflags 2.6.0",
- "libloading",
- "winapi",
-]
-
 [[package]]
 name = "darling"
 version = "0.20.10"
@@ -2720,9 +2694,9 @@ dependencies = [
 
 [[package]]
 name = "glow"
-version = "0.13.1"
+version = "0.14.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "bd348e04c43b32574f2de31c8bb397d96c9fcfa1371bd4ca6d8bdc464ab121b1"
+checksum = "d51fa363f025f5c111e03f13eda21162faeacb6911fe8caa0c0349f9cf0c4483"
 dependencies = [
  "js-sys",
  "slotmap",
@@ -2760,15 +2734,14 @@ dependencies = [
 
 [[package]]
 name = "gpu-allocator"
-version = "0.26.0"
+version = "0.27.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "fdd4240fc91d3433d5e5b0fc5b67672d771850dc19bbee03c1381e19322803d7"
+checksum = "c151a2a5ef800297b4e79efa4f4bec035c5f51d5ae587287c9b952bdf734cacd"
 dependencies = [
  "log",
  "presser",
  "thiserror",
- "winapi",
- "windows",
+ "windows 0.58.0",
 ]
 
 [[package]]
@@ -2901,21 +2874,6 @@ dependencies = [
  "hashbrown 0.14.5",
 ]
 
-[[package]]
-name = "hassle-rs"
-version = "0.11.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "af2a7e73e1f34c48da31fb668a907f250794837e08faa144fd24f0b8b741e890"
-dependencies = [
- "bitflags 2.6.0",
- "com",
- "libc",
- "libloading",
- "thiserror",
- "widestring",
- "winapi",
-]
-
 [[package]]
 name = "heck"
 version = "0.4.1"
@@ -3190,7 +3148,7 @@ dependencies = [
  "iana-time-zone-haiku",
  "js-sys",
  "wasm-bindgen",
- "windows-core",
+ "windows-core 0.52.0",
 ]
 
 [[package]]
@@ -3265,7 +3223,7 @@ dependencies = [
  "burn-candle",
  "burn-import",
  "console_error_panic_hook",
- "cubecl-runtime",
+ "cubecl-runtime 0.3.0",
  "js-sys",
  "log",
  "serde",
@@ -3793,7 +3751,7 @@ version = "0.16.0"
 dependencies = [
  "burn",
  "console_error_panic_hook",
- "cubecl-runtime",
+ "cubecl-runtime 0.3.0",
  "js-sys",
  "serde",
  "wasm-bindgen",
@@ -3853,9 +3811,8 @@ dependencies = [
 
 [[package]]
 name = "naga"
-version = "22.1.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "8bd5a652b6faf21496f2cfd88fc49989c8db0825d1f6746b1a71a6ede24a63ad"
+version = "23.0.0"
+source = "git+https://github.com/erichdongubler-mozilla/wgpu?branch=v23#194b6ccfa554a73b9852cd88f69d04cb642e975a"
 dependencies = [
  "arrayvec",
  "bit-set",
@@ -6481,7 +6438,7 @@ dependencies = [
  "once_cell",
  "rayon",
  "serde",
- "windows",
+ "windows 0.52.0",
 ]
 
 [[package]]
@@ -7347,9 +7304,8 @@ checksum = "53a85b86a771b1c87058196170769dd264f66c0782acf1ae6cc51bfd64b39082"
 
 [[package]]
 name = "wgpu"
-version = "22.1.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "e1d1c4ba43f80542cf63a0a6ed3134629ae73e8ab51e4b765a67f3aa062eb433"
+version = "23.0.0"
+source = "git+https://github.com/erichdongubler-mozilla/wgpu?branch=v23#194b6ccfa554a73b9852cd88f69d04cb642e975a"
 dependencies = [
  "arrayvec",
  "cfg_aliases 0.1.1",
@@ -7372,9 +7328,8 @@ dependencies = [
 
 [[package]]
 name = "wgpu-core"
-version = "22.1.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "0348c840d1051b8e86c3bcd31206080c5e71e5933dabd79be1ce732b0b2f089a"
+version = "23.0.0"
+source = "git+https://github.com/erichdongubler-mozilla/wgpu?branch=v23#194b6ccfa554a73b9852cd88f69d04cb642e975a"
 dependencies = [
  "arrayvec",
  "bit-vec",
@@ -7397,9 +7352,8 @@ dependencies = [
 
 [[package]]
 name = "wgpu-hal"
-version = "22.0.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f6bbf4b4de8b2a83c0401d9e5ae0080a2792055f25859a02bf9be97952bbed4f"
+version = "23.0.0"
+source = "git+https://github.com/erichdongubler-mozilla/wgpu?branch=v23#194b6ccfa554a73b9852cd88f69d04cb642e975a"
 dependencies = [
  "android_system_properties",
  "arrayvec",
@@ -7407,15 +7361,14 @@ dependencies = [
  "bit-set",
  "bitflags 2.6.0",
  "block",
+ "bytemuck",
  "cfg_aliases 0.1.1",
  "core-graphics-types",
- "d3d12",
  "glow",
  "glutin_wgl_sys",
  "gpu-alloc",
  "gpu-allocator",
  "gpu-descriptor",
- "hassle-rs",
  "js-sys",
  "khronos-egl",
  "libc",
@@ -7437,14 +7390,14 @@ dependencies = [
  "wasm-bindgen",
  "web-sys",
  "wgpu-types",
- "winapi",
+ "windows 0.58.0",
+ "windows-core 0.58.0",
 ]
 
 [[package]]
 name = "wgpu-types"
-version = "22.0.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "bc9d91f0e2c4b51434dfa6db77846f2793149d8e73f800fa2e41f52b8eac3c5d"
+version = "23.0.0"
+source = "git+https://github.com/erichdongubler-mozilla/wgpu?branch=v23#194b6ccfa554a73b9852cd88f69d04cb642e975a"
 dependencies = [
  "bitflags 2.6.0",
  "js-sys",
@@ -7463,12 +7416,6 @@ dependencies = [
  "rustix",
 ]
 
-[[package]]
-name = "widestring"
-version = "1.1.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "7219d36b6eac893fa81e84ebe06485e7dcbb616177469b142df14f1f4deb1311"
-
 [[package]]
 name = "winapi"
 version = "0.3.9"
@@ -7506,7 +7453,17 @@ version = "0.52.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "e48a53791691ab099e5e2ad123536d0fff50652600abaf43bbf952894110d0be"
 dependencies = [
- "windows-core",
+ "windows-core 0.52.0",
+ "windows-targets 0.52.6",
+]
+
+[[package]]
+name = "windows"
+version = "0.58.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "dd04d41d93c4992d421894c18c8b43496aa748dd4c081bac0dc93eb0489272b6"
+dependencies = [
+ "windows-core 0.58.0",
  "windows-targets 0.52.6",
 ]
 
@@ -7519,6 +7476,41 @@ dependencies = [
  "windows-targets 0.52.6",
 ]
 
+[[package]]
+name = "windows-core"
+version = "0.58.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6ba6d44ec8c2591c134257ce647b7ea6b20335bf6379a27dac5f1641fcf59f99"
+dependencies = [
+ "windows-implement",
+ "windows-interface",
+ "windows-result",
+ "windows-strings",
+ "windows-targets 0.52.6",
+]
+
+[[package]]
+name = "windows-implement"
+version = "0.58.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2bbd5b46c938e506ecbce286b6628a02171d56153ba733b6c741fc627ec9579b"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.85",
+]
+
+[[package]]
+name = "windows-interface"
+version = "0.58.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "053c4c462dc91d3b1504c6fe5a726dd15e216ba718e84a0e46a88fbe5ded3515"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.85",
+]
+
 [[package]]
 name = "windows-registry"
 version = "0.2.0"
diff --git a/Cargo.toml b/Cargo.toml
index af68c1b44d2..4de7ad71f4e 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -101,7 +101,7 @@ ratatui = "0.29.0"
 
 # WGPU stuff
 text_placeholder = "0.5.1"
-wgpu = "22.1.0"
+wgpu = { git = "https://github.com/erichdongubler-mozilla/wgpu", branch = "v23" }
 
 # Benchmarks and Burnbench
 arboard = "3.4.1"
@@ -155,10 +155,10 @@ portable-atomic-util = { version = "0.2.2", features = ["alloc"] }
 # cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "63da837b5ae78ff1a3b7363fe3af2b02c2bc864f" }
 # cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "63da837b5ae78ff1a3b7363fe3af2b02c2bc864f" }
 ### For local development. ###
-# cubecl = { path = "../cubecl/crates/cubecl", default-features = false }
+cubecl = { path = "../cubecl/crates/cubecl", default-features = false }
 # cubecl-common = { path = "../cubecl/crates/cubecl-common", default-features = false }
 ### For the release. ###
-cubecl = { version="0.3.0", default-features = false }
+# cubecl = { version="0.3.0", default-features = false }
 cubecl-common = { version="0.3.0", default-features = false }
 
 ### For xtask crate ###
diff --git a/crates/burn-tensor/src/lib.rs b/crates/burn-tensor/src/lib.rs
index acb3d75ecf5..4abc274e647 100644
--- a/crates/burn-tensor/src/lib.rs
+++ b/crates/burn-tensor/src/lib.rs
@@ -66,12 +66,12 @@ mod cube_wgpu {
                 WgpuDevice::VirtualGpu(index) => DeviceId::new(2, *index as u32),
                 WgpuDevice::Cpu => DeviceId::new(3, 0),
                 WgpuDevice::BestAvailable => DeviceId::new(4, 0),
-                // For an existing device, use the 64 bit wgpu device ID as the burn DeviceID.
-                // We're only storing 32 bits, so wrap the the 64 bit value to 32 bits. This
+                // For an existing device, use the 32 or 64 bit wgpu device ID as the burn DeviceID.
+                // We're only storing 32 bits, so wrap the the 32 or 64 bit value to 32 bits. This
                 // might collide - but a 1 in 4 billion chance seems ok given there's only a few
                 // devices in flight at any time.
                 WgpuDevice::Existing(id) => {
-                    DeviceId::new(5, (id.inner() % (u32::MAX as u64)) as u32)
+                    DeviceId::new(5, (*id % (u32::MAX as usize)) as u32)
                 }
             }
         }

From b62508292773d8e7b06a1a9cc14e679b823832ba Mon Sep 17 00:00:00 2001
From: AsherJingkongChen <37398747+AsherJingkongChen@users.noreply.github.com>
Date: Tue, 29 Oct 2024 11:34:41 +0800
Subject: [PATCH 2/7] Format

---
 crates/burn-tensor/src/lib.rs | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/crates/burn-tensor/src/lib.rs b/crates/burn-tensor/src/lib.rs
index 4abc274e647..362e519907d 100644
--- a/crates/burn-tensor/src/lib.rs
+++ b/crates/burn-tensor/src/lib.rs
@@ -70,9 +70,7 @@ mod cube_wgpu {
                 // We're only storing 32 bits, so wrap the the 32 or 64 bit value to 32 bits. This
                 // might collide - but a 1 in 4 billion chance seems ok given there's only a few
                 // devices in flight at any time.
-                WgpuDevice::Existing(id) => {
-                    DeviceId::new(5, (*id % (u32::MAX as usize)) as u32)
-                }
+                WgpuDevice::Existing(id) => DeviceId::new(5, (*id % (u32::MAX as usize)) as u32),
             }
         }
     }

From 001cbbc4fe7a3bebb0a9c2210a776d1130474861 Mon Sep 17 00:00:00 2001
From: AsherJingkongChen <37398747+AsherJingkongChen@users.noreply.github.com>
Date: Tue, 29 Oct 2024 12:08:42 +0800
Subject: [PATCH 3/7] Update CubeCL

---
 Cargo.toml | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index 4de7ad71f4e..c0d49f330d9 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -152,14 +152,14 @@ ahash = { version = "0.8.11", default-features = false }
 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 = "63da837b5ae78ff1a3b7363fe3af2b02c2bc864f" }
-# cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "63da837b5ae78ff1a3b7363fe3af2b02c2bc864f" }
+cubecl = { git = "https://github.com/AsherJingkongChen/cubecl", default-features = false, rev = "a9ae356413b7706f2fe382f65aec16b9ba701274" }
+cubecl-common = { git = "https://github.com/AsherJingkongChen/cubecl", default-features = false, rev = "a9ae356413b7706f2fe382f65aec16b9ba701274" }
 ### For local development. ###
-cubecl = { path = "../cubecl/crates/cubecl", default-features = false }
+# cubecl = { path = "../cubecl/crates/cubecl", default-features = false }
 # cubecl-common = { path = "../cubecl/crates/cubecl-common", default-features = false }
 ### For the release. ###
 # cubecl = { version="0.3.0", default-features = false }
-cubecl-common = { version="0.3.0", default-features = false }
+# cubecl-common = { version = "0.3.0", default-features = false }
 
 ### For xtask crate ###
 tracel-xtask = { version = "~1.1" }

From 615c17444a1fe18b292fc92286e0f8443ad9de6e Mon Sep 17 00:00:00 2001
From: AsherJingkongChen <37398747+AsherJingkongChen@users.noreply.github.com>
Date: Tue, 29 Oct 2024 22:13:05 +0800
Subject: [PATCH 4/7] Update CubeCL

* The comments in `WgpuDevice::id()` are removed since `id` in `WgpuDevice::Existing` is always `u32`
---
 Cargo.lock                    | 14 +++++++++++++-
 Cargo.toml                    |  4 ++--
 crates/burn-tensor/src/lib.rs |  6 +-----
 3 files changed, 16 insertions(+), 8 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index d260764506b..3c6f6722d8f 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -524,7 +524,7 @@ dependencies = [
 name = "burn-common"
 version = "0.16.0"
 dependencies = [
- "cubecl-common 0.3.0",
+ "cubecl-common 0.4.0",
  "dashmap",
  "getrandom",
  "indicatif",
@@ -1429,6 +1429,7 @@ dependencies = [
 [[package]]
 name = "cubecl"
 version = "0.4.0"
+source = "git+https://github.com/AsherJingkongChen/cubecl?rev=fcb608503a59f0ef82562e9aec16df5e61904372#fcb608503a59f0ef82562e9aec16df5e61904372"
 dependencies = [
  "cubecl-core",
  "cubecl-cuda",
@@ -1459,6 +1460,7 @@ dependencies = [
 [[package]]
 name = "cubecl-common"
 version = "0.4.0"
+source = "git+https://github.com/AsherJingkongChen/cubecl?rev=fcb608503a59f0ef82562e9aec16df5e61904372#fcb608503a59f0ef82562e9aec16df5e61904372"
 dependencies = [
  "derive-new",
  "embassy-futures",
@@ -1475,6 +1477,7 @@ dependencies = [
 [[package]]
 name = "cubecl-core"
 version = "0.4.0"
+source = "git+https://github.com/AsherJingkongChen/cubecl?rev=fcb608503a59f0ef82562e9aec16df5e61904372#fcb608503a59f0ef82562e9aec16df5e61904372"
 dependencies = [
  "bytemuck",
  "cubecl-common 0.4.0",
@@ -1491,6 +1494,7 @@ dependencies = [
 [[package]]
 name = "cubecl-cpp"
 version = "0.4.0"
+source = "git+https://github.com/AsherJingkongChen/cubecl?rev=fcb608503a59f0ef82562e9aec16df5e61904372#fcb608503a59f0ef82562e9aec16df5e61904372"
 dependencies = [
  "bytemuck",
  "cubecl-common 0.4.0",
@@ -1504,6 +1508,7 @@ dependencies = [
 [[package]]
 name = "cubecl-cuda"
 version = "0.4.0"
+source = "git+https://github.com/AsherJingkongChen/cubecl?rev=fcb608503a59f0ef82562e9aec16df5e61904372#fcb608503a59f0ef82562e9aec16df5e61904372"
 dependencies = [
  "bytemuck",
  "cubecl-common 0.4.0",
@@ -1519,6 +1524,7 @@ dependencies = [
 [[package]]
 name = "cubecl-hip"
 version = "0.4.0"
+source = "git+https://github.com/AsherJingkongChen/cubecl?rev=fcb608503a59f0ef82562e9aec16df5e61904372#fcb608503a59f0ef82562e9aec16df5e61904372"
 dependencies = [
  "bytemuck",
  "cubecl-common 0.4.0",
@@ -1543,6 +1549,7 @@ dependencies = [
 [[package]]
 name = "cubecl-linalg"
 version = "0.4.0"
+source = "git+https://github.com/AsherJingkongChen/cubecl?rev=fcb608503a59f0ef82562e9aec16df5e61904372#fcb608503a59f0ef82562e9aec16df5e61904372"
 dependencies = [
  "bytemuck",
  "cubecl-core",
@@ -1553,6 +1560,7 @@ dependencies = [
 [[package]]
 name = "cubecl-macros"
 version = "0.4.0"
+source = "git+https://github.com/AsherJingkongChen/cubecl?rev=fcb608503a59f0ef82562e9aec16df5e61904372#fcb608503a59f0ef82562e9aec16df5e61904372"
 dependencies = [
  "cubecl-common 0.4.0",
  "darling",
@@ -1567,6 +1575,7 @@ dependencies = [
 [[package]]
 name = "cubecl-opt"
 version = "0.4.0"
+source = "git+https://github.com/AsherJingkongChen/cubecl?rev=fcb608503a59f0ef82562e9aec16df5e61904372#fcb608503a59f0ef82562e9aec16df5e61904372"
 dependencies = [
  "cubecl-common 0.4.0",
  "cubecl-core",
@@ -1603,6 +1612,7 @@ dependencies = [
 [[package]]
 name = "cubecl-runtime"
 version = "0.4.0"
+source = "git+https://github.com/AsherJingkongChen/cubecl?rev=fcb608503a59f0ef82562e9aec16df5e61904372#fcb608503a59f0ef82562e9aec16df5e61904372"
 dependencies = [
  "async-channel",
  "async-lock",
@@ -1623,6 +1633,7 @@ dependencies = [
 [[package]]
 name = "cubecl-spirv"
 version = "0.4.0"
+source = "git+https://github.com/AsherJingkongChen/cubecl?rev=fcb608503a59f0ef82562e9aec16df5e61904372#fcb608503a59f0ef82562e9aec16df5e61904372"
 dependencies = [
  "cubecl-common 0.4.0",
  "cubecl-core",
@@ -1635,6 +1646,7 @@ dependencies = [
 [[package]]
 name = "cubecl-wgpu"
 version = "0.4.0"
+source = "git+https://github.com/AsherJingkongChen/cubecl?rev=fcb608503a59f0ef82562e9aec16df5e61904372#fcb608503a59f0ef82562e9aec16df5e61904372"
 dependencies = [
  "ash",
  "async-channel",
diff --git a/Cargo.toml b/Cargo.toml
index c0d49f330d9..0c1c8a4c040 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -152,8 +152,8 @@ ahash = { version = "0.8.11", default-features = false }
 portable-atomic-util = { version = "0.2.2", features = ["alloc"] }
 
 ### For the main burn branch. ###
-cubecl = { git = "https://github.com/AsherJingkongChen/cubecl", default-features = false, rev = "a9ae356413b7706f2fe382f65aec16b9ba701274" }
-cubecl-common = { git = "https://github.com/AsherJingkongChen/cubecl", default-features = false, rev = "a9ae356413b7706f2fe382f65aec16b9ba701274" }
+cubecl = { git = "https://github.com/AsherJingkongChen/cubecl", default-features = false, rev = "fcb608503a59f0ef82562e9aec16df5e61904372" }
+cubecl-common = { git = "https://github.com/AsherJingkongChen/cubecl", default-features = false, rev = "fcb608503a59f0ef82562e9aec16df5e61904372" }
 ### For local development. ###
 # cubecl = { path = "../cubecl/crates/cubecl", default-features = false }
 # cubecl-common = { path = "../cubecl/crates/cubecl-common", default-features = false }
diff --git a/crates/burn-tensor/src/lib.rs b/crates/burn-tensor/src/lib.rs
index 362e519907d..b78c66e0307 100644
--- a/crates/burn-tensor/src/lib.rs
+++ b/crates/burn-tensor/src/lib.rs
@@ -66,11 +66,7 @@ mod cube_wgpu {
                 WgpuDevice::VirtualGpu(index) => DeviceId::new(2, *index as u32),
                 WgpuDevice::Cpu => DeviceId::new(3, 0),
                 WgpuDevice::BestAvailable => DeviceId::new(4, 0),
-                // For an existing device, use the 32 or 64 bit wgpu device ID as the burn DeviceID.
-                // We're only storing 32 bits, so wrap the the 32 or 64 bit value to 32 bits. This
-                // might collide - but a 1 in 4 billion chance seems ok given there's only a few
-                // devices in flight at any time.
-                WgpuDevice::Existing(id) => DeviceId::new(5, (*id % (u32::MAX as usize)) as u32),
+                WgpuDevice::Existing(id) => DeviceId::new(5, *id),
             }
         }
     }

From 230212e3eb6e3c1ea0bcf6660dea22b94f04a9d0 Mon Sep 17 00:00:00 2001
From: AsherJingkongChen <37398747+AsherJingkongChen@users.noreply.github.com>
Date: Thu, 31 Oct 2024 03:35:15 +0800
Subject: [PATCH 5/7] Update CubeCL and Wgpu

---
 Cargo.lock                                   | 39 +++++++++++---------
 Cargo.toml                                   |  6 +--
 examples/image-classification-web/src/web.rs |  4 +-
 3 files changed, 27 insertions(+), 22 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index 8079625bec9..48af83fd25f 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -1429,7 +1429,7 @@ dependencies = [
 [[package]]
 name = "cubecl"
 version = "0.4.0"
-source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#c8ff1e188687d1db04d0d12eb604241745952f38"
 dependencies = [
  "cubecl-core",
  "cubecl-cuda",
@@ -1460,7 +1460,7 @@ dependencies = [
 [[package]]
 name = "cubecl-common"
 version = "0.4.0"
-source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#c8ff1e188687d1db04d0d12eb604241745952f38"
 dependencies = [
  "derive-new",
  "embassy-futures",
@@ -1477,7 +1477,7 @@ dependencies = [
 [[package]]
 name = "cubecl-core"
 version = "0.4.0"
-source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#c8ff1e188687d1db04d0d12eb604241745952f38"
 dependencies = [
  "bytemuck",
  "cubecl-common 0.4.0",
@@ -1494,7 +1494,7 @@ dependencies = [
 [[package]]
 name = "cubecl-cpp"
 version = "0.4.0"
-source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#c8ff1e188687d1db04d0d12eb604241745952f38"
 dependencies = [
  "bytemuck",
  "cubecl-common 0.4.0",
@@ -1508,7 +1508,7 @@ dependencies = [
 [[package]]
 name = "cubecl-cuda"
 version = "0.4.0"
-source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#c8ff1e188687d1db04d0d12eb604241745952f38"
 dependencies = [
  "bytemuck",
  "cubecl-common 0.4.0",
@@ -1524,7 +1524,7 @@ dependencies = [
 [[package]]
 name = "cubecl-hip"
 version = "0.4.0"
-source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#c8ff1e188687d1db04d0d12eb604241745952f38"
 dependencies = [
  "bytemuck",
  "cubecl-common 0.4.0",
@@ -1549,7 +1549,7 @@ dependencies = [
 [[package]]
 name = "cubecl-linalg"
 version = "0.4.0"
-source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#c8ff1e188687d1db04d0d12eb604241745952f38"
 dependencies = [
  "bytemuck",
  "cubecl-core",
@@ -1560,7 +1560,7 @@ dependencies = [
 [[package]]
 name = "cubecl-macros"
 version = "0.4.0"
-source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#c8ff1e188687d1db04d0d12eb604241745952f38"
 dependencies = [
  "cubecl-common 0.4.0",
  "darling",
@@ -1575,7 +1575,7 @@ dependencies = [
 [[package]]
 name = "cubecl-opt"
 version = "0.4.0"
-source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#c8ff1e188687d1db04d0d12eb604241745952f38"
 dependencies = [
  "cubecl-common 0.4.0",
  "cubecl-core",
@@ -1612,7 +1612,7 @@ dependencies = [
 [[package]]
 name = "cubecl-runtime"
 version = "0.4.0"
-source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#c8ff1e188687d1db04d0d12eb604241745952f38"
 dependencies = [
  "async-channel",
  "async-lock",
@@ -1633,7 +1633,7 @@ dependencies = [
 [[package]]
 name = "cubecl-spirv"
 version = "0.4.0"
-source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#c8ff1e188687d1db04d0d12eb604241745952f38"
 dependencies = [
  "cubecl-common 0.4.0",
  "cubecl-core",
@@ -1646,7 +1646,7 @@ dependencies = [
 [[package]]
 name = "cubecl-wgpu"
 version = "0.4.0"
-source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#c8ff1e188687d1db04d0d12eb604241745952f38"
 dependencies = [
  "ash",
  "async-channel",
@@ -3824,7 +3824,8 @@ dependencies = [
 [[package]]
 name = "naga"
 version = "23.0.0"
-source = "git+https://github.com/erichdongubler-mozilla/wgpu?branch=v23#194b6ccfa554a73b9852cd88f69d04cb642e975a"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3d5941e45a15b53aad4375eedf02033adb7a28931eedc31117faffa52e6a857e"
 dependencies = [
  "arrayvec",
  "bit-set",
@@ -7317,7 +7318,8 @@ checksum = "53a85b86a771b1c87058196170769dd264f66c0782acf1ae6cc51bfd64b39082"
 [[package]]
 name = "wgpu"
 version = "23.0.0"
-source = "git+https://github.com/erichdongubler-mozilla/wgpu?branch=v23#194b6ccfa554a73b9852cd88f69d04cb642e975a"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "76ab52f2d3d18b70d5ab8dd270a1cff3ebe6dbe4a7d13c1cc2557138a9777fdc"
 dependencies = [
  "arrayvec",
  "cfg_aliases 0.1.1",
@@ -7341,7 +7343,8 @@ dependencies = [
 [[package]]
 name = "wgpu-core"
 version = "23.0.0"
-source = "git+https://github.com/erichdongubler-mozilla/wgpu?branch=v23#194b6ccfa554a73b9852cd88f69d04cb642e975a"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0e0c68e7b6322a03ee5b83fcd92caeac5c2a932f6457818179f4652ad2a9c065"
 dependencies = [
  "arrayvec",
  "bit-vec",
@@ -7365,7 +7368,8 @@ dependencies = [
 [[package]]
 name = "wgpu-hal"
 version = "23.0.0"
-source = "git+https://github.com/erichdongubler-mozilla/wgpu?branch=v23#194b6ccfa554a73b9852cd88f69d04cb642e975a"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "de6e7266b869de56c7e3ed72a954899f71d14fec6cc81c102b7530b92947601b"
 dependencies = [
  "android_system_properties",
  "arrayvec",
@@ -7409,7 +7413,8 @@ dependencies = [
 [[package]]
 name = "wgpu-types"
 version = "23.0.0"
-source = "git+https://github.com/erichdongubler-mozilla/wgpu?branch=v23#194b6ccfa554a73b9852cd88f69d04cb642e975a"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "610f6ff27778148c31093f3b03abc4840f9636d58d597ca2f5977433acfe0068"
 dependencies = [
  "bitflags 2.6.0",
  "js-sys",
diff --git a/Cargo.toml b/Cargo.toml
index 1f1ec5eae9f..489fbbf4697 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -101,7 +101,7 @@ ratatui = "0.29.0"
 
 # WGPU stuff
 text_placeholder = "0.5.1"
-wgpu = { git = "https://github.com/erichdongubler-mozilla/wgpu", branch = "v23" }
+wgpu = "23.0.0"
 
 # Benchmarks and Burnbench
 arboard = "3.4.1"
@@ -152,8 +152,8 @@ ahash = { version = "0.8.11", default-features = false }
 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 = "99404b1e29946832a42b72a5c26d4cf42c67692e" }
-cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "99404b1e29946832a42b72a5c26d4cf42c67692e" }
+cubecl = { git = "https://github.com/AsherJingkongChen/cubecl", default-features = false, branch = "refactor/wgpu-v23" }
+cubecl-common = { git = "https://github.com/AsherJingkongChen/cubecl", default-features = false, branch = "refactor/wgpu-v23" }
 ### For local development. ###
 # cubecl = { path = "../cubecl/crates/cubecl", default-features = false }
 # cubecl-common = { path = "../cubecl/crates/cubecl-common", default-features = false }
diff --git a/examples/image-classification-web/src/web.rs b/examples/image-classification-web/src/web.rs
index bcff11eb69c..4b20507abcc 100644
--- a/examples/image-classification-web/src/web.rs
+++ b/examples/image-classification-web/src/web.rs
@@ -9,7 +9,7 @@ use core::convert::Into;
 use crate::model::{label::LABELS, normalizer::Normalizer, squeezenet::Model as SqueezenetModel};
 
 use burn::{
-    backend::{wgpu::init_device, NdArray},
+    backend::{wgpu::init_setup_async, NdArray},
     prelude::*,
     tensor::activation::softmax,
 };
@@ -110,7 +110,7 @@ impl ImageClassifier {
         log::info!("Loading the model to the Wgpu backend");
         let start = Instant::now();
         let device = WgpuDevice::default();
-        init_device::<AutoGraphicsApi>(&device, Default::default()).await;
+        init_setup_async::<AutoGraphicsApi>(&device, Default::default()).await;
         self.model = ModelType::WithWgpuBackend(Model::new(&device));
         let duration = start.elapsed();
         log::debug!("Model is loaded to the Wgpu backend in {:?}", duration);

From 591047f18c41c2d67cec08822516b6537e846791 Mon Sep 17 00:00:00 2001
From: AsherJingkongChen <37398747+AsherJingkongChen@users.noreply.github.com>
Date: Tue, 26 Nov 2024 04:02:27 +0800
Subject: [PATCH 6/7] Patch Cargo.lock

---
 Cargo.lock | 225 ++++++++++++++++++++++++++++-------------------------
 1 file changed, 118 insertions(+), 107 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index f098787e247..7c44c6891d4 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -321,7 +321,7 @@ dependencies = [
  "http 1.1.0",
  "http-body 1.0.1",
  "http-body-util",
- "hyper 1.5.0",
+ "hyper 1.5.1",
  "hyper-util",
  "itoa",
  "matchit",
@@ -335,7 +335,7 @@ dependencies = [
  "serde_path_to_error",
  "serde_urlencoded",
  "sha1",
- "sync_wrapper 1.0.1",
+ "sync_wrapper 1.0.2",
  "tokio",
  "tokio-tungstenite",
  "tower",
@@ -359,7 +359,7 @@ dependencies = [
  "mime",
  "pin-project-lite",
  "rustversion",
- "sync_wrapper 1.0.1",
+ "sync_wrapper 1.0.2",
  "tower-layer",
  "tower-service",
  "tracing",
@@ -632,7 +632,7 @@ dependencies = [
  "serde_json",
  "spin",
  "tempfile",
- "thiserror",
+ "thiserror 1.0.69",
  "uuid",
 ]
 
@@ -680,7 +680,7 @@ dependencies = [
  "strum",
  "strum_macros",
  "tempfile",
- "thiserror",
+ "thiserror 1.0.69",
 ]
 
 [[package]]
@@ -741,10 +741,10 @@ dependencies = [
  "serde",
  "serde_json",
  "syn 2.0.89",
- "thiserror",
+ "thiserror 1.0.69",
  "tracing-core",
  "tracing-subscriber",
- "zip 2.2.0",
+ "zip 2.2.1",
 ]
 
 [[package]]
@@ -998,7 +998,7 @@ dependencies = [
  "rand_distr",
  "rayon",
  "safetensors 0.4.5",
- "thiserror",
+ "thiserror 1.0.69",
  "yoke",
  "zip 1.1.4",
 ]
@@ -1020,7 +1020,7 @@ checksum = "166a92826d615d98b205e346e52128fa0439f2ab3302587403fdc558b4219e19"
 dependencies = [
  "metal 0.27.0",
  "once_cell",
- "thiserror",
+ "thiserror 1.0.69",
  "tracing",
 ]
 
@@ -1374,9 +1374,9 @@ dependencies = [
 
 [[package]]
 name = "cpufeatures"
-version = "0.2.15"
+version = "0.2.16"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "0ca741a962e1b0bff6d724a1a0958b686406e853bb14061f218562e1896f95e6"
+checksum = "16b80225097f2e5ae4e7179dd2266824648f3e2f49d9134d584b76389d31c4c3"
 dependencies = [
  "libc",
 ]
@@ -1513,7 +1513,7 @@ dependencies = [
 [[package]]
 name = "cubecl"
 version = "0.4.0"
-source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#ba78b17f15b97fd8a316b60ed9631f3dd1e3be9e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#e04d19ec3dd060a9b4d0809143606230ee55ba2c"
 dependencies = [
  "cubecl-core",
  "cubecl-cuda",
@@ -1544,7 +1544,7 @@ dependencies = [
 [[package]]
 name = "cubecl-common"
 version = "0.4.0"
-source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#ba78b17f15b97fd8a316b60ed9631f3dd1e3be9e"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#e04d19ec3dd060a9b4d0809143606230ee55ba2c"
 dependencies = [
  "derive-new 0.6.0",
  "embassy-futures",
@@ -1561,8 +1561,7 @@ dependencies = [
 [[package]]
 name = "cubecl-core"
 version = "0.4.0"
-source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#ba78b17f15b97fd8a316b60ed9631f3dd1e3be9e"
-source = "git+https://github.com/tracel-ai/cubecl?rev=2c09d4dd1ecb9f474e524dc47b05599edb7049e7#2c09d4dd1ecb9f474e524dc47b05599edb7049e7"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#e04d19ec3dd060a9b4d0809143606230ee55ba2c"
 dependencies = [
  "bytemuck",
  "cubecl-common 0.4.0",
@@ -1580,8 +1579,7 @@ dependencies = [
 [[package]]
 name = "cubecl-cpp"
 version = "0.4.0"
-source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#ba78b17f15b97fd8a316b60ed9631f3dd1e3be9e"
-source = "git+https://github.com/tracel-ai/cubecl?rev=2c09d4dd1ecb9f474e524dc47b05599edb7049e7#2c09d4dd1ecb9f474e524dc47b05599edb7049e7"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#e04d19ec3dd060a9b4d0809143606230ee55ba2c"
 dependencies = [
  "bytemuck",
  "cubecl-common 0.4.0",
@@ -1595,8 +1593,7 @@ dependencies = [
 [[package]]
 name = "cubecl-cuda"
 version = "0.4.0"
-source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#ba78b17f15b97fd8a316b60ed9631f3dd1e3be9e"
-source = "git+https://github.com/tracel-ai/cubecl?rev=2c09d4dd1ecb9f474e524dc47b05599edb7049e7#2c09d4dd1ecb9f474e524dc47b05599edb7049e7"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#e04d19ec3dd060a9b4d0809143606230ee55ba2c"
 dependencies = [
  "bytemuck",
  "cubecl-common 0.4.0",
@@ -1612,8 +1609,7 @@ dependencies = [
 [[package]]
 name = "cubecl-hip"
 version = "0.4.0"
-source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#ba78b17f15b97fd8a316b60ed9631f3dd1e3be9e"
-source = "git+https://github.com/tracel-ai/cubecl?rev=2c09d4dd1ecb9f474e524dc47b05599edb7049e7#2c09d4dd1ecb9f474e524dc47b05599edb7049e7"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#e04d19ec3dd060a9b4d0809143606230ee55ba2c"
 dependencies = [
  "bytemuck",
  "cubecl-common 0.4.0",
@@ -1624,6 +1620,7 @@ dependencies = [
  "derive-new 0.6.0",
  "half",
  "log",
+ "paste",
 ]
 
 [[package]]
@@ -1638,8 +1635,7 @@ dependencies = [
 [[package]]
 name = "cubecl-linalg"
 version = "0.4.0"
-source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#ba78b17f15b97fd8a316b60ed9631f3dd1e3be9e"
-source = "git+https://github.com/tracel-ai/cubecl?rev=2c09d4dd1ecb9f474e524dc47b05599edb7049e7#2c09d4dd1ecb9f474e524dc47b05599edb7049e7"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#e04d19ec3dd060a9b4d0809143606230ee55ba2c"
 dependencies = [
  "bytemuck",
  "cubecl-core",
@@ -1650,8 +1646,7 @@ dependencies = [
 [[package]]
 name = "cubecl-macros"
 version = "0.4.0"
-source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#ba78b17f15b97fd8a316b60ed9631f3dd1e3be9e"
-source = "git+https://github.com/tracel-ai/cubecl?rev=2c09d4dd1ecb9f474e524dc47b05599edb7049e7#2c09d4dd1ecb9f474e524dc47b05599edb7049e7"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#e04d19ec3dd060a9b4d0809143606230ee55ba2c"
 dependencies = [
  "cubecl-common 0.4.0",
  "darling",
@@ -1666,8 +1661,7 @@ dependencies = [
 [[package]]
 name = "cubecl-opt"
 version = "0.4.0"
-source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#ba78b17f15b97fd8a316b60ed9631f3dd1e3be9e"
-source = "git+https://github.com/tracel-ai/cubecl?rev=2c09d4dd1ecb9f474e524dc47b05599edb7049e7#2c09d4dd1ecb9f474e524dc47b05599edb7049e7"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#e04d19ec3dd060a9b4d0809143606230ee55ba2c"
 dependencies = [
  "cubecl-common 0.4.0",
  "cubecl-core",
@@ -1704,8 +1698,7 @@ dependencies = [
 [[package]]
 name = "cubecl-runtime"
 version = "0.4.0"
-source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#ba78b17f15b97fd8a316b60ed9631f3dd1e3be9e"
-source = "git+https://github.com/tracel-ai/cubecl?rev=2c09d4dd1ecb9f474e524dc47b05599edb7049e7#2c09d4dd1ecb9f474e524dc47b05599edb7049e7"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#e04d19ec3dd060a9b4d0809143606230ee55ba2c"
 dependencies = [
  "async-channel",
  "async-lock",
@@ -1726,8 +1719,7 @@ dependencies = [
 [[package]]
 name = "cubecl-spirv"
 version = "0.4.0"
-source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#ba78b17f15b97fd8a316b60ed9631f3dd1e3be9e"
-source = "git+https://github.com/tracel-ai/cubecl?rev=2c09d4dd1ecb9f474e524dc47b05599edb7049e7#2c09d4dd1ecb9f474e524dc47b05599edb7049e7"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#e04d19ec3dd060a9b4d0809143606230ee55ba2c"
 dependencies = [
  "cubecl-common 0.4.0",
  "cubecl-core",
@@ -1741,8 +1733,7 @@ dependencies = [
 [[package]]
 name = "cubecl-wgpu"
 version = "0.4.0"
-source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#ba78b17f15b97fd8a316b60ed9631f3dd1e3be9e"
-source = "git+https://github.com/tracel-ai/cubecl?rev=2c09d4dd1ecb9f474e524dc47b05599edb7049e7#2c09d4dd1ecb9f474e524dc47b05599edb7049e7"
+source = "git+https://github.com/AsherJingkongChen/cubecl?branch=refactor/wgpu-v23#e04d19ec3dd060a9b4d0809143606230ee55ba2c"
 dependencies = [
  "ash",
  "async-channel",
@@ -2740,7 +2731,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "f93d7df7366121b5018f947a04d37f034717e113dcf9ccd85c34b58e57a74d5e"
 dependencies = [
  "faster-hex",
- "thiserror",
+ "thiserror 1.0.69",
 ]
 
 [[package]]
@@ -2864,7 +2855,7 @@ checksum = "c151a2a5ef800297b4e79efa4f4bec035c5f51d5ae587287c9b952bdf734cacd"
 dependencies = [
  "log",
  "presser",
- "thiserror",
+ "thiserror 1.0.69",
  "windows 0.58.0",
 ]
 
@@ -2918,9 +2909,9 @@ dependencies = [
 
 [[package]]
 name = "h2"
-version = "0.4.6"
+version = "0.4.7"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "524e8ac6999421f49a846c2d4411f337e53497d8ec55d67753beffa43c5d9205"
+checksum = "ccae279728d634d083c00f6099cb58f01cc99c145b84b8be2f6c74618d79922e"
 dependencies = [
  "atomic-waker",
  "bytes",
@@ -3050,7 +3041,7 @@ dependencies = [
  "rand",
  "serde",
  "serde_json",
- "thiserror",
+ "thiserror 1.0.69",
  "ureq",
 ]
 
@@ -3178,14 +3169,14 @@ dependencies = [
 
 [[package]]
 name = "hyper"
-version = "1.5.0"
+version = "1.5.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "bbbff0a806a4728c99295b254c8838933b5b082d75e3cb70c8dab21fdfbcfa9a"
+checksum = "97818827ef4f364230e16705d4706e2897df2bb60617d6ca15d598025a3c481f"
 dependencies = [
  "bytes",
  "futures-channel",
  "futures-util",
- "h2 0.4.6",
+ "h2 0.4.7",
  "http 1.1.0",
  "http-body 1.0.1",
  "httparse",
@@ -3205,7 +3196,7 @@ checksum = "08afdbb5c31130e3034af566421053ab03787c640246a446327f550d11bcb333"
 dependencies = [
  "futures-util",
  "http 1.1.0",
- "hyper 1.5.0",
+ "hyper 1.5.1",
  "hyper-util",
  "rustls",
  "rustls-pki-types",
@@ -3235,7 +3226,7 @@ checksum = "70206fc6890eaca9fde8a0bf71caa2ddfc9fe045ac9e5c70df101a7dbde866e0"
 dependencies = [
  "bytes",
  "http-body-util",
- "hyper 1.5.0",
+ "hyper 1.5.1",
  "hyper-util",
  "native-tls",
  "tokio",
@@ -3254,7 +3245,7 @@ dependencies = [
  "futures-util",
  "http 1.1.0",
  "http-body 1.0.1",
- "hyper 1.5.0",
+ "hyper 1.5.1",
  "pin-project-lite",
  "socket2",
  "tokio",
@@ -3628,9 +3619,9 @@ dependencies = [
 
 [[package]]
 name = "itoa"
-version = "1.0.11"
+version = "1.0.14"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "49f1f14873335454500d59611f1cf4a4b0f786f9ac11f4312a78e4cf2566695b"
+checksum = "d75a2a4b1b190afb6f5425f10f6a8f959d2ea0b9c2b1d79553551850539e4674"
 
 [[package]]
 name = "itoap"
@@ -3720,7 +3711,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "4979f22fdb869068da03c9f7528f8297c6fd2606bc3a4affe42e6a823fdb8da4"
 dependencies = [
  "cfg-if",
- "windows-targets 0.48.5",
+ "windows-targets 0.52.6",
 ]
 
 [[package]]
@@ -3759,9 +3750,9 @@ checksum = "78b3ae25bc7c8c38cec158d1f2757ee79e9b3740fbc7ccf0e59e4b08d793fa89"
 
 [[package]]
 name = "litemap"
-version = "0.7.3"
+version = "0.7.4"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "643cb0b8d4fcc284004d5fd0d67ccf61dfffadb7f75e1e71bc420f4688a3a704"
+checksum = "4ee93343901ab17bd981295f2cf0026d4ad018c7c31ba84549a4ddbb47a45104"
 
 [[package]]
 name = "litrs"
@@ -4088,7 +4079,7 @@ dependencies = [
  "rustc-hash",
  "spirv",
  "termcolor",
- "thiserror",
+ "thiserror 1.0.69",
  "unicode-xid",
 ]
 
@@ -4370,7 +4361,7 @@ dependencies = [
  "libloading",
  "nvml-wrapper-sys",
  "static_assertions",
- "thiserror",
+ "thiserror 1.0.69",
  "wrapcenum-derive",
 ]
 
@@ -4585,7 +4576,7 @@ dependencies = [
  "flate2",
  "native-tls",
  "tar",
- "thiserror",
+ "thiserror 1.0.69",
  "ureq",
  "walkdir",
 ]
@@ -4984,7 +4975,7 @@ dependencies = [
  "rayon",
  "regex",
  "smartstring",
- "thiserror",
+ "thiserror 1.0.69",
  "version_check",
  "xxhash-rust",
 ]
@@ -4998,7 +4989,7 @@ dependencies = [
  "polars-arrow-format",
  "regex",
  "simdutf8",
- "thiserror",
+ "thiserror 1.0.69",
 ]
 
 [[package]]
@@ -5274,9 +5265,9 @@ dependencies = [
 
 [[package]]
 name = "portable-atomic"
-version = "1.9.0"
+version = "1.10.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "cc9c68a3f6da06753e9335d63e27f6b9754dd1920d941135b7ea8224f141adb2"
+checksum = "280dc24453071f1b63954171985a0b0d30058d287960968b9b2aca264c8d4ee6"
 
 [[package]]
 name = "portable-atomic-util"
@@ -5398,7 +5389,7 @@ dependencies = [
  "bytes",
  "once_cell",
  "protobuf-support",
- "thiserror",
+ "thiserror 1.0.69",
 ]
 
 [[package]]
@@ -5413,7 +5404,7 @@ dependencies = [
  "protobuf-parse",
  "regex",
  "tempfile",
- "thiserror",
+ "thiserror 1.0.69",
 ]
 
 [[package]]
@@ -5428,7 +5419,7 @@ dependencies = [
  "protobuf",
  "protobuf-support",
  "tempfile",
- "thiserror",
+ "thiserror 1.0.69",
  "which",
 ]
 
@@ -5438,7 +5429,7 @@ version = "3.7.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "b088fd20b938a875ea00843b6faf48579462630015c3788d397ad6a786663252"
 dependencies = [
- "thiserror",
+ "thiserror 1.0.69",
 ]
 
 [[package]]
@@ -5627,7 +5618,7 @@ dependencies = [
  "rand_chacha",
  "simd_helpers",
  "system-deps",
- "thiserror",
+ "thiserror 1.0.69",
  "v_frame",
  "wasm-bindgen",
 ]
@@ -5760,7 +5751,7 @@ checksum = "ba009ff324d1fc1b900bd1fdb31564febe58a8ccc8a6fdbb93b543d33b13ca43"
 dependencies = [
  "getrandom",
  "libredox",
- "thiserror",
+ "thiserror 1.0.69",
 ]
 
 [[package]]
@@ -5871,11 +5862,11 @@ dependencies = [
  "futures-channel",
  "futures-core",
  "futures-util",
- "h2 0.4.6",
+ "h2 0.4.7",
  "http 1.1.0",
  "http-body 1.0.1",
  "http-body-util",
- "hyper 1.5.0",
+ "hyper 1.5.1",
  "hyper-rustls",
  "hyper-tls 0.6.0",
  "hyper-util",
@@ -5891,7 +5882,7 @@ dependencies = [
  "serde",
  "serde_json",
  "serde_urlencoded",
- "sync_wrapper 1.0.1",
+ "sync_wrapper 1.0.2",
  "system-configuration 0.6.1",
  "tokio",
  "tokio-native-tls",
@@ -6035,9 +6026,9 @@ dependencies = [
 
 [[package]]
 name = "rustix"
-version = "0.38.40"
+version = "0.38.41"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "99e4ea3e1cdc4b559b8e5650f9c8e5998e3e5c1343b4eaf034565f32318d63c0"
+checksum = "d7f649912bc1495e167a6edee79151c84b1bad49748cb4f1f1167f459f6224f6"
 dependencies = [
  "bitflags 2.6.0",
  "errno",
@@ -6171,9 +6162,9 @@ dependencies = [
 
 [[package]]
 name = "schannel"
-version = "0.1.26"
+version = "0.1.27"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "01227be5826fa0690321a2ba6c5cd57a19cf3f6a09e76973b58e61de6ab9d1c1"
+checksum = "1f29ebaa345f945cec9fbbc532eb307f0fdad8161f281b6369539c8d84876b3d"
 dependencies = [
  "windows-sys 0.59.0",
 ]
@@ -6276,9 +6267,9 @@ dependencies = [
 
 [[package]]
 name = "serde_json"
-version = "1.0.132"
+version = "1.0.133"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d726bfaff4b320266d395898905d0eba0345aae23b54aee3a737e260fd46db03"
+checksum = "c7fceb2473b9166b2294ef05efcb65a3db80803f0b03ef86a5fc88a2b85ee377"
 dependencies = [
  "itoa",
  "memchr",
@@ -6675,9 +6666,9 @@ checksum = "2047c6ded9c721764247e62cd3b03c09ffc529b2ba5b10ec482ae507a4a70160"
 
 [[package]]
 name = "sync_wrapper"
-version = "1.0.1"
+version = "1.0.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "a7065abeca94b6a8a577f9bd45aa0867a2238b74e8eb67cf10d492bc39351394"
+checksum = "0bf256ce5efdfa370213c1dabab5935a12e49f2c58d15e9eac2870d3b4f27263"
 dependencies = [
  "futures-core",
 ]
@@ -6703,7 +6694,7 @@ dependencies = [
  "byteorder",
  "enum-as-inner",
  "libc",
- "thiserror",
+ "thiserror 1.0.69",
  "walkdir",
 ]
 
@@ -6827,7 +6818,7 @@ dependencies = [
  "ndarray 0.15.6",
  "rand",
  "safetensors 0.3.3",
- "thiserror",
+ "thiserror 1.0.69",
  "torch-sys",
  "zip 0.6.6",
 ]
@@ -6908,7 +6899,16 @@ version = "1.0.69"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "b6aaf5339b578ea85b50e080feb250a3e8ae8cfcdff9a461c9ec2904bc923f52"
 dependencies = [
- "thiserror-impl",
+ "thiserror-impl 1.0.69",
+]
+
+[[package]]
+name = "thiserror"
+version = "2.0.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c006c85c7651b3cf2ada4584faa36773bd07bac24acfb39f3c431b36d7e667aa"
+dependencies = [
+ "thiserror-impl 2.0.3",
 ]
 
 [[package]]
@@ -6922,6 +6922,17 @@ dependencies = [
  "syn 2.0.89",
 ]
 
+[[package]]
+name = "thiserror-impl"
+version = "2.0.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f077553d607adc1caf65430528a576c757a71ed73944b66ebb58ef2bbd243568"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.89",
+]
+
 [[package]]
 name = "thread-tree"
 version = "0.3.3"
@@ -7036,7 +7047,7 @@ dependencies = [
  "serde",
  "serde_json",
  "spm_precompiled",
- "thiserror",
+ "thiserror 1.0.69",
  "unicode-normalization-alignments",
  "unicode-segmentation",
  "unicode_categories",
@@ -7241,7 +7252,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "3566e8ce28cc0a3fe42519fc80e6b4c943cc4c8cef275620eb8dac2d3d4e06cf"
 dependencies = [
  "crossbeam-channel",
- "thiserror",
+ "thiserror 1.0.69",
  "time",
  "tracing-subscriber",
 ]
@@ -7316,7 +7327,7 @@ dependencies = [
  "log",
  "rand",
  "sha1",
- "thiserror",
+ "thiserror 1.0.69",
  "utf-8",
 ]
 
@@ -7328,9 +7339,9 @@ checksum = "42ff0bf0c66b8238c6f3b578df37d0b7848e55df8577b3f74f92a69acceeb825"
 
 [[package]]
 name = "unicode-ident"
-version = "1.0.13"
+version = "1.0.14"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "e91b56cd4cadaeb79bbf1a5645f6b4f8dc5bde8834ad5894a8db35fda9efa1fe"
+checksum = "adb9e6ca4f869e1180728b7950e35922a7fc6397f7b641499e8f3ef06e50dc83"
 
 [[package]]
 name = "unicode-normalization"
@@ -7428,9 +7439,9 @@ dependencies = [
 
 [[package]]
 name = "url"
-version = "2.5.3"
+version = "2.5.4"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "8d157f1b96d14500ffdc1f10ba712e780825526c03d9a49b4d0324b0d9113ada"
+checksum = "32f8b686cadd1473f4bd0117a5d28d36b1ade384ea9b5069a1c40aefed7fda60"
 dependencies = [
  "form_urlencoded",
  "idna",
@@ -7646,9 +7657,9 @@ dependencies = [
 
 [[package]]
 name = "webpki-roots"
-version = "0.26.6"
+version = "0.26.7"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "841c67bff177718f1d4dfefde8d8f0e78f9b6589319ba88312f567fc5841a958"
+checksum = "5d642ff16b7e79272ae451b7322067cdc17cadf68c23264be9d94a32319efe7e"
 dependencies = [
  "rustls-pki-types",
 ]
@@ -7661,9 +7672,9 @@ checksum = "53a85b86a771b1c87058196170769dd264f66c0782acf1ae6cc51bfd64b39082"
 
 [[package]]
 name = "wgpu"
-version = "23.0.0"
+version = "23.0.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "76ab52f2d3d18b70d5ab8dd270a1cff3ebe6dbe4a7d13c1cc2557138a9777fdc"
+checksum = "80f70000db37c469ea9d67defdc13024ddf9a5f1b89cb2941b812ad7cde1735a"
 dependencies = [
  "arrayvec",
  "cfg_aliases 0.1.1",
@@ -7686,9 +7697,9 @@ dependencies = [
 
 [[package]]
 name = "wgpu-core"
-version = "23.0.0"
+version = "23.0.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "0e0c68e7b6322a03ee5b83fcd92caeac5c2a932f6457818179f4652ad2a9c065"
+checksum = "d63c3c478de8e7e01786479919c8769f62a22eec16788d8c2ac77ce2c132778a"
 dependencies = [
  "arrayvec",
  "bit-vec",
@@ -7705,16 +7716,16 @@ dependencies = [
  "raw-window-handle",
  "rustc-hash",
  "smallvec",
- "thiserror",
+ "thiserror 1.0.69",
  "wgpu-hal",
  "wgpu-types",
 ]
 
 [[package]]
 name = "wgpu-hal"
-version = "23.0.0"
+version = "23.0.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "de6e7266b869de56c7e3ed72a954899f71d14fec6cc81c102b7530b92947601b"
+checksum = "89364b8a0b211adc7b16aeaf1bd5ad4a919c1154b44c9ce27838213ba05fd821"
 dependencies = [
  "android_system_properties",
  "arrayvec",
@@ -7747,7 +7758,7 @@ dependencies = [
  "renderdoc-sys",
  "rustc-hash",
  "smallvec",
- "thiserror",
+ "thiserror 1.0.69",
  "wasm-bindgen",
  "web-sys",
  "wgpu-types",
@@ -7859,7 +7870,7 @@ checksum = "2bbd5b46c938e506ecbce286b6628a02171d56153ba733b6c741fc627ec9579b"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn 2.0.87",
+ "syn 2.0.89",
 ]
 
 [[package]]
@@ -7870,7 +7881,7 @@ checksum = "053c4c462dc91d3b1504c6fe5a726dd15e216ba718e84a0e46a88fbe5ded3515"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn 2.0.87",
+ "syn 2.0.89",
 ]
 
 [[package]]
@@ -8158,9 +8169,9 @@ checksum = "cfe53a6657fd280eaa890a3bc59152892ffa3e30101319d168b781ed6529b049"
 
 [[package]]
 name = "yoke"
-version = "0.7.4"
+version = "0.7.5"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "6c5b1314b079b0930c31e3af543d8ee1757b1951ae1e1565ec704403a7240ca5"
+checksum = "120e6aef9aa629e3d4f52dc8cc43a015c7724194c97dfaf45180d2daf2b77f40"
 dependencies = [
  "serde",
  "stable_deref_trait",
@@ -8170,9 +8181,9 @@ dependencies = [
 
 [[package]]
 name = "yoke-derive"
-version = "0.7.4"
+version = "0.7.5"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "28cc31741b18cb6f1d5ff12f5b7523e3d6eb0852bbbad19d73905511d9849b95"
+checksum = "2380878cad4ac9aac1e2435f3eb4020e8374b5f13c296cb75b4620ff8e229154"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -8203,18 +8214,18 @@ dependencies = [
 
 [[package]]
 name = "zerofrom"
-version = "0.1.4"
+version = "0.1.5"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "91ec111ce797d0e0784a1116d0ddcdbea84322cd79e5d5ad173daeba4f93ab55"
+checksum = "cff3ee08c995dee1859d998dea82f7374f2826091dd9cd47def953cae446cd2e"
 dependencies = [
  "zerofrom-derive",
 ]
 
 [[package]]
 name = "zerofrom-derive"
-version = "0.1.4"
+version = "0.1.5"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "0ea7b4a3637ea8669cedf0f1fd5c286a17f3de97b8dd5a70a6c167a1730e63a5"
+checksum = "595eed982f7d355beb85837f651fa22e90b3c044842dc7f2c2842c086f295808"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -8296,14 +8307,14 @@ dependencies = [
  "displaydoc",
  "indexmap 2.6.0",
  "num_enum",
- "thiserror",
+ "thiserror 1.0.69",
 ]
 
 [[package]]
 name = "zip"
-version = "2.2.0"
+version = "2.2.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "dc5e4288ea4057ae23afc69a4472434a87a2495cafce6632fd1c4ec9f5cf3494"
+checksum = "99d52293fc86ea7cf13971b3bb81eb21683636e7ae24c729cdaf1b7c4157a352"
 dependencies = [
  "aes",
  "arbitrary",
@@ -8321,7 +8332,7 @@ dependencies = [
  "pbkdf2 0.12.2",
  "rand",
  "sha1",
- "thiserror",
+ "thiserror 2.0.3",
  "time",
  "zeroize",
  "zopfli",

From 1d8b1316fd06682ab21b7f24da47e62d06828652 Mon Sep 17 00:00:00 2001
From: AsherJingkongChen <37398747+AsherJingkongChen@users.noreply.github.com>
Date: Fri, 6 Dec 2024 02:03:34 +0800
Subject: [PATCH 7/7] Update CubeCL

---
 Cargo.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index 09e1db30228..8b10b2bb3a0 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -153,8 +153,8 @@ ahash = { version = "0.8.11", default-features = false }
 portable-atomic-util = { version = "0.2.4", features = ["alloc"] }
 
 ### For the main burn branch. ###
-cubecl = { git = "https://github.com/AsherJingkongChen/cubecl", default-features = false, branch = "refactor/wgpu-v23" }
-cubecl-common = { git = "https://github.com/AsherJingkongChen/cubecl", default-features = false, branch = "refactor/wgpu-v23" }
+cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "13fc62c22f311d2d153abbc4ffb4ef5271bc82a3" }
+cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "13fc62c22f311d2d153abbc4ffb4ef5271bc82a3" }
 ### For local development. ###
 # cubecl = { path = "../cubecl/crates/cubecl", default-features = false }
 # cubecl-common = { path = "../cubecl/crates/cubecl-common", default-features = false }