ci: support linux sve

Signed-off-by: thxCode <thxcode0824@gmail.com>
gpustack · Oct 9, 2024 · 2c04140 · 2c04140
1 parent 28e5d9e
commit 2c04140
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -168,13 +168,13 @@ jobs:
       fail-fast: false
       matrix:
         # AVX2 ==> CentOS 7.
-        # NEON ==> Ubuntu 18.04.
+        # NEON+SVE ==> Ubuntu 18.04.
         include:
           - arch: 'amd64'
             instruction: 'avx2'
             distro_container_image: 'centos:7'
           - arch: 'arm64'
-            instruction: 'neon'
+            instruction: 'neon+sve'
             distro_container_image: 'ubuntu:18.04'
     runs-on: ubuntu-22.04
     steps:
@@ -282,6 +282,8 @@ jobs:
           cmake -S ${{ github.workspace }} -B ${{ github.workspace }}/build -DCMAKE_BUILD_TYPE=Release \
             -DGGML_NATIVE=off \
             ${{ matrix.instruction == 'avx2' && '-DGGML_AVX2=on' || '' }} \
+            ${{ matrix.instruction == 'neon+sve' && '-DCMAKE_C_FLAGS_INIT="-march=armv8.2-a+sve"' || '' }} \
+            ${{ matrix.instruction == 'neon+sve' && '-DCMAKE_CXX_FLAGS_INIT="-march=armv8.2-a+sve"' || '' }} \
             -DGGML_BLAS_VENDOR=OpenBLAS \
             -DGGML_STATIC=on \
             -DGGML_BLAS=on \
@@ -1050,8 +1052,9 @@ jobs:
           
           Write-Host "===== BUILD ====="
           New-Item -Force -ItemType Directory -Path "${{ github.workspace }}\.cache" -ErrorAction Ignore | Out-Null
-          cmake ${{ matrix.arch == 'arm64' && format('-D CMAKE_TOOLCHAIN_FILE={0}\llama-box\scripts\build-windows-neon.cmake', github.workspace) || '' }} -S ${{ github.workspace }} -B ${{ github.workspace }}\build -DCMAKE_BUILD_TYPE=Release `
+          cmake -S ${{ github.workspace }} -B ${{ github.workspace }}\build -DCMAKE_BUILD_TYPE=Release `
             -DGGML_NATIVE=off `
+            ${{ matrix.arch == 'arm64' && format('-DCMAKE_TOOLCHAIN_FILE={0}\llama-box\scripts\build-windows-arm64-msvc.cmake', github.workspace) || '' }} `
             ${{ matrix.instruction == 'avx2' && '-DGGML_AVX2=on' || '' }} `
             -DGGML_STATIC=on `
             -DGGML_OPENMP=off `

diff --git a/llama-box/scripts/build-windows-neon.cmake → ...ox/scripts/build-windows-arm64-msvc.cmake b/llama-box/scripts/build-windows-neon.cmake → ...ox/scripts/build-windows-arm64-msvc.cmake