diff --git a/.github/workflows/server-test.yml b/.github/workflows/server-test.yml index 6aec537f83837a..c5f819220a5af9 100644 --- a/.github/workflows/server-test.yml +++ b/.github/workflows/server-test.yml @@ -38,6 +38,7 @@ jobs: - build: 'cublas' defines: '-DLLAMA_NATIVE=OFF -DLLAMA_BUILD_SERVER=ON -DLLAMA_CUBLAS=ON' image: nvidia/cuda:12.3.1-devel-ubuntu22.04 + arch_not_available: true # require nvidia docker engine - build: 'clblast' defines: '-DLLAMA_NATIVE=OFF -DLLAMA_BUILD_SERVER=ON -DLLAMA_CLBLAST=ON' image: ubuntu:latest @@ -58,7 +59,7 @@ jobs: image: ${{ matrix.image }} ports: - 8888 - options: --cpus 4 + options: --cpus 4 ${{ matrix.build == 'cublas' && '--gpu all'}} steps: - name: Clone