Test vulkan workflow

TabbyML · Mar 5, 2024 · 6c172aa · 6c172aa
1 parent ec9d25c
commit 6c172aa
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -27,7 +27,7 @@ jobs:
     container: ${{ matrix.container }}
     strategy:
       matrix:
-        binary: [aarch64-apple-darwin, x86_64-manylinux2014, x86_64-windows-msvc, x86_64-manylinux2014-cuda117, x86_64-manylinux2014-cuda122, x86_64-windows-msvc-cuda117, x86_64-windows-msvc-cuda122]
+        binary: [aarch64-apple-darwin, x86_64-manylinux2014, x86_64-windows-msvc, x86_64-manylinux2014-cuda117, x86_64-manylinux2014-cuda122, x86_64-windows-msvc-cuda117, x86_64-windows-msvc-cuda122, x86_64-manylinux2014-vulkan]
         include:
           - os: macos-latest
             target: aarch64-apple-darwin
@@ -64,6 +64,11 @@ jobs:
             ext: .exe
             build_args: --features cuda,prod-db
             windows_cuda: '12.2.0'
+          - os: dimerun-k3-ubuntu2204
+            target: x86_64-unknown-linux-gnu
+            container: quay.io/pypa/manylinux2014_x86_64
+            build_args: --features vulkan,prod-db
+            binary: x86_64-manylinux2014-vulkan
 
     env:
       SCCACHE_GHA_ENABLED: true

diff --git a/ci/prepare_build_environment.sh b/ci/prepare_build_environment.sh
@@ -16,7 +16,7 @@ if [[ "$OSTYPE" == "linux"* ]]; then
     sudo apt-get -y install protobuf-compiler libopenblas-dev
   else
     # Build from manylinux2014 container
-    yum -y install openblas-devel perl-IPC-Cmd unzip curl openssl-devel
+    yum -y install openblas-devel perl-IPC-Cmd unzip curl openssl-devel vulkan
 
     # Disable safe directory in docker
     git config --system --add safe.directory "*"
@@ -31,4 +31,4 @@ install_mailtutan() {
   cargo install mailtutan
 }
 
-install_mailtutan
+install_mailtutan