CI: Update publish step

tracel-ai · Oct 24, 2023 · ae0de59 · ae0de59
1 parent 9f2bc59
commit ae0de59
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -24,6 +24,15 @@ jobs:
       crate: burn-common
     secrets: inherit
 
+  publish-burn-compute:
+    uses: burn-rs/burn/.github/workflows/publish-template.yml@main
+    needs:
+      - publish-burn-common
+    with:
+    with:
+      crate: burn-compute
+    secrets: inherit
+
   publish-burn-tensor-testgen:
     uses: burn-rs/burn/.github/workflows/publish-template.yml@main
     with:
@@ -34,6 +43,7 @@ jobs:
     uses: burn-rs/burn/.github/workflows/publish-template.yml@main
     needs:
       - publish-burn-tensor-testgen
+      - publish-burn-common
     with:
       crate: burn-tensor
     secrets: inherit
@@ -71,13 +81,24 @@ jobs:
     uses: burn-rs/burn/.github/workflows/publish-template.yml@main
     needs:
       - publish-burn-tensor
+      - publish-burn-compute
       - publish-burn-autodiff
       - publish-burn-ndarray
       - publish-burn-common
     with:
       crate: burn-wgpu
     secrets: inherit
 
+  publish-burn-candle:
+    uses: burn-rs/burn/.github/workflows/publish-template.yml@main
+    needs:
+      - publish-burn-tensor
+      - publish-burn-autodiff
+      - publish-burn-tch
+    with:
+      crate: burn-candle
+    secrets: inherit
+
   publish-burn-core:
     uses: burn-rs/burn/.github/workflows/publish-template.yml@main
     needs: