diff --git a/.github/workflows/server-test.yml b/.github/workflows/server-test.yml index 61ca12ffb714b..d9a4ac595d0ff 100644 --- a/.github/workflows/server-test.yml +++ b/.github/workflows/server-test.yml @@ -44,8 +44,8 @@ jobs: defines: '-DLLAMA_NATIVE=OFF -DLLAMA_BUILD_SERVER=ON -DLLAMA_KOMPUTE=ON -DKOMPUTE_OPT_DISABLE_VULKAN_VERSION_CHECK=ON' - build: 'vulkan' defines: '-DLLAMA_NATIVE=OFF -DLLAMA_BUILD_SERVER=ON -DLLAMA_VULKAN=ON' - - build: 'cuda' - defines: '-DLLAMA_NATIVE=OFF -DLLAMA_BUILD_SERVER=ON -DLLAMA_CUBLAS=ON' + - build: 'cublas' + defines: '-DLLAMA_NATIVE=OFF -DLLAMA_BUILD_SERVER=ON -DLLAMA_CUBLAS=ON -DCUDAToolkit_ROOT=/usr/local/cuda' steps: - name: Clone @@ -87,7 +87,7 @@ jobs: - name: Install CUDA id: get_cuda - if: ${{ matrix.build == 'cuda' }} + if: ${{ matrix.build == 'cublas' }} run: | wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb dpkg -i cuda-keyring_1.1-1_all.deb