Add cuda in CI publish

tracel-ai · Aug 27, 2024 · 7eb3a7b · 7eb3a7b
1 parent 9881ca6
commit 7eb3a7b
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -111,6 +111,19 @@ jobs:
       crate: burn-wgpu
     secrets: inherit
 
+  publish-burn-cuda:
+    uses: tracel-ai/burn/.github/workflows/publish-template.yml@main
+    needs:
+      - publish-burn-tensor
+      - publish-burn-compute
+      - publish-burn-autodiff
+      - publish-burn-ndarray
+      - publish-burn-common
+      - publish-burn-jit
+    with:
+      crate: burn-cuda
+    secrets: inherit
+
   publish-burn-candle:
     uses: tracel-ai/burn/.github/workflows/publish-template.yml@main
     needs: