ci: install cuda for selected packages

romnn · Jul 4, 2023 · a788523 · a788523
1 parent 2c7bce7
commit a788523
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 7 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -46,12 +46,13 @@ jobs:
     steps:
       - uses: actions/checkout@v3
       - uses: dtolnay/rust-toolchain@stable
-      # - name: Install CUDA
-      #   uses: Jimver/cuda-toolkit@v0.2.10
-      #   id: cuda-toolkit
-      #   with:
-      #     cuda: "11.8.0"
-      #     linux-local-args: '["--driver", "--toolkit"]'
+      - name: Install CUDA
+        uses: Jimver/cuda-toolkit@v0.2.10
+        id: cuda-toolkit
+        if: ${{ matrix.package.need_cuda }}
+        with:
+          cuda: "11.8.0"
+          linux-local-args: '["--driver", "--toolkit"]'
       - name: Build all
         run: >-
           cargo build  --package "${{ matrix.package.name }}" --features "${{ matrix.package.features }}" --all-targets
diff --git a/accelsim/sim/Cargo.toml b/accelsim/sim/Cargo.toml
@@ -9,6 +9,9 @@ default  = []
 [package.metadata.cargo-feature-combinations]
 denylist = ["default"]
 
+[package.metadata.cargo-feature-combinations.matrix]
+need_cuda = true
+
 [dependencies]
 color-eyre = "0"
 async-process = "1"
@@ -19,5 +22,5 @@ accelsim = { path = "../" }
 
 [build-dependencies]
 color-eyre = "0"
-duct = "0"
+# duct = "0"
 accelsim = { path = "../" }
diff --git a/accelsim/trace/Cargo.toml b/accelsim/trace/Cargo.toml
@@ -9,6 +9,9 @@ default  = []
 [package.metadata.cargo-feature-combinations]
 denylist = ["default"]
 
+[package.metadata.cargo-feature-combinations.matrix]
+need_cuda = true
+
 [dependencies]
 color-eyre = "0"
 dialoguer = "0"

diff --git a/playground/Cargo.toml b/playground/Cargo.toml
@@ -19,6 +19,9 @@ path = "src/main.rs"
 default = ["parallel"]
 parallel = ["cc/parallel"]
 
+[package.metadata.cargo-feature-combinations]
+denylist = ["default"]
+
 [dependencies]
 cxx = "1"
 accelsim = { path = "../accelsim" }