Add SPIR-V compiler backend to burn-wgpu (#2386)

tracel-ai · Oct 21, 2024 · f3968cb · f3968cb
1 parent b7887b0
commit f3968cb
Show file tree

Hide file tree

Showing 15 changed files with 279 additions and 86 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -135,6 +135,8 @@ jobs:
   linux-std-tests:
     runs-on: ubuntu-22.04
     needs: prepare-checks
+    env:
+      DISABLE_WGPU_SPIRV: '1'
     strategy:
       matrix:
         rust: [stable, prev]

diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -152,10 +152,10 @@ tch = "0.15.0"
 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 = "609183cccb26e2ccaa834f2230b5b8f38f9ea507" }
-cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "609183cccb26e2ccaa834f2230b5b8f38f9ea507" }
+cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "e3fdc96ec2d68dcdde8135bd011907b4e662388c" }
+cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "e3fdc96ec2d68dcdde8135bd011907b4e662388c" }
 ### 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.2.0", default-features = false }