From 7eb3a7b27a63521ed1ed425b93b3e7adf59b3a87 Mon Sep 17 00:00:00 2001 From: nathaniel Date: Tue, 27 Aug 2024 12:36:19 -0400 Subject: [PATCH] Add cuda in CI publish --- .github/workflows/publish.yml | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml index ef8ea52d8b..ddd61d06f2 100644 --- 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: