Refactor/cubecl fusion (#2815)

tracel-ai · Feb 17, 2025 · f3dfd05 · f3dfd05
1 parent 136eeb6
commit f3dfd05
Show file tree

Hide file tree

Showing 51 changed files with 2,549 additions and 3,576 deletions.
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -14,6 +14,7 @@ jobs:
       - publish-burn-autodiff
       - publish-burn-candle
       - publish-burn-fusion
+      - publish-burn-cubecl-fusion
       - publish-burn-cubecl
       - publish-burn-ndarray
       - publish-burn-tch
@@ -113,12 +114,25 @@ jobs:
     secrets:
       CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }}
 
+  publish-burn-cubecl-fusion:
+    uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v1
+    needs:
+      - publish-burn-ir
+      - publish-burn-common
+      - publish-burn-fusion
+      - publish-burn-tensor
+    with:
+      crate: burn-cubecl-fusion
+    secrets:
+      CRATES_IO_API_TOKEN: ${{ secrets.CRATES_IO_API_TOKEN }}
+
   publish-burn-cubecl:
     uses: tracel-ai/github-actions/.github/workflows/publish-crate.yml@v1
     needs:
       - publish-burn-ir
       - publish-burn-common
       - publish-burn-fusion
+      - publish-burn-cubecl-fusion
       - publish-burn-tensor
       - publish-burn-ndarray
     with:

diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -155,8 +155,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/tracel-ai/cubecl", default-features = false, rev = "7f07d3969bc7d69c6ae2f87bd806dd4f18267741" }
-cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "7f07d3969bc7d69c6ae2f87bd806dd4f18267741" }
+cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "8b025f26e5badbf1b8f3e6787fc097427cd961ec" }
+cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "8b025f26e5badbf1b8f3e6787fc097427cd961ec" }
 ### For local development. ###
 # cubecl = { path = "../cubecl/crates/cubecl", default-features = false }
 # cubecl-common = { path = "../cubecl/crates/cubecl-common", default-features = false }