Support unfixed kv heads number (#1416) #4933
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: macOS (12, Python 3.9) | |
| on: | |
| workflow_dispatch: | |
| pull_request: | |
| merge_group: | |
| push: | |
| branches: | |
| - master | |
| - 'releases/**' | |
| permissions: read-all # Required by https://github.com/ossf/scorecard/blob/e23b8ad91fd6a64a0a971ca4fc0a4d1650725615/docs/checks.md#token-permissions | |
| concurrency: | |
| # github.ref is not unique in post-commit | |
| group: ${{ github.event_name == 'push' && github.run_id || github.ref }}-mac | |
| cancel-in-progress: true | |
| env: | |
| PYTHON_VERSION: '3.9' | |
| OV_BRANCH: master | |
| OV_TARBALL: '' | |
| jobs: | |
| openvino_download: | |
| name: Download OpenVINO package | |
| outputs: | |
| status: ${{ steps.openvino_download.outcome }} | |
| timeout-minutes: 10 | |
| defaults: | |
| run: | |
| shell: bash | |
| runs-on: ubuntu-20.04 | |
| steps: | |
| - name: Download OpenVINO build | |
| id: openvino_download | |
| run: | | |
| wget ${{ env.OV_TARBALL}} --progress=bar:force:noscroll -O openvino_package.tar.gz | |
| tar -tvf openvino_package.tar.gz | |
| continue-on-error: true | |
| # | |
| # Upload to artifacts | |
| # | |
| - name: Upload openvino package | |
| if: steps.openvino_download.outcome == 'success' | |
| uses: actions/upload-artifact@v4 | |
| with: | |
| name: openvino_package | |
| path: openvino_package.tar.gz | |
| if-no-files-found: 'error' | |
| openvino_build: | |
| name: Build OpenVINO package | |
| needs: [openvino_download] | |
| if: needs.openvino_download.outputs.status != 'success' | |
| timeout-minutes: 150 | |
| defaults: | |
| run: | |
| shell: bash | |
| runs-on: 'macos-12-large' | |
| env: | |
| MACOSX_DEPLOYMENT_TARGET: '10.15' | |
| CMAKE_BUILD_TYPE: 'Release' | |
| CMAKE_GENERATOR: 'Ninja Multi-Config' | |
| CMAKE_CXX_COMPILER_LAUNCHER: ccache | |
| CMAKE_C_COMPILER_LAUNCHER: ccache | |
| OPENVINO_REPO: ${{ github.workspace }}/openvino | |
| INSTALL_DIR: ${{ github.workspace }}/openvino/install | |
| BUILD_DIR: ${{ github.workspace }}/openvino/build | |
| steps: | |
| - name: Clone OpenVINO | |
| uses: actions/checkout@v4 | |
| with: | |
| repository: 'openvinotoolkit/openvino' | |
| path: ${{ env.OPENVINO_REPO }} | |
| submodules: 'true' | |
| ref: ${{ env.OV_BRANCH }} | |
| # | |
| # Dependencies | |
| # | |
| - name: Install build dependencies | |
| run: brew install coreutils ninja | |
| - name: Setup Python ${{ env.PYTHON_VERSION }} | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: ${{ env.PYTHON_VERSION }} | |
| cache: 'pip' | |
| - name: Install python dependencies | |
| run: | | |
| # For Python API: build and wheel packaging | |
| python3 -m pip install -r ${OPENVINO_REPO}/src/bindings/python/wheel/requirements-dev.txt | |
| # | |
| # Build | |
| # | |
| - name: Setup ccache | |
| uses: hendrikmuhs/[email protected] | |
| with: | |
| max-size: "2000M" | |
| # Should save cache only if run in the master branch of the base repo | |
| # github.ref_name is 'ref/PR_#' in case of the PR, and 'branch_name' when executed on push | |
| save: ${{ github.ref_name == 'master' && 'true' || 'false' }} | |
| verbose: 2 | |
| key: ccache-mac | |
| restore-keys: | | |
| ccache-mac | |
| - name: CMake configure - OpenVINO | |
| run: | | |
| cmake \ | |
| -G "${{ env.CMAKE_GENERATOR }}" \ | |
| -DENABLE_CPPLINT=OFF \ | |
| -DENABLE_NCC_STYLE=OFF \ | |
| -DENABLE_TESTS=OFF \ | |
| -DENABLE_STRICT_DEPENDENCIES=OFF \ | |
| -DENABLE_SYSTEM_OPENCL=ON \ | |
| -DCMAKE_VERBOSE_MAKEFILE=ON \ | |
| -DCPACK_GENERATOR=TGZ \ | |
| -DENABLE_JS=OFF \ | |
| -DENABLE_SAMPLES=ON \ | |
| -DENABLE_OV_ONNX_FRONTEND=OFF \ | |
| -DENABLE_OV_PADDLE_FRONTEND=OFF \ | |
| -DENABLE_OV_PYTORCH_FRONTEND=ON \ | |
| -DENABLE_OV_TF_FRONTEND=ON \ | |
| -DENABLE_OV_TF_LITE_FRONTEND=OFF \ | |
| -DENABLE_INTEL_GPU=OFF \ | |
| -DENABLE_INTEL_NPU=OFF \ | |
| -DCMAKE_COMPILE_WARNING_AS_ERROR=ON \ | |
| -DCMAKE_CXX_COMPILER_LAUNCHER=ccache \ | |
| -DCMAKE_C_COMPILER_LAUNCHER=ccache \ | |
| -DENABLE_PYTHON=ON \ | |
| -DENABLE_WHEEL=ON \ | |
| -S ${OPENVINO_REPO} \ | |
| -B ${BUILD_DIR} | |
| - name: Clean ccache stats | |
| run: ccache --zero-stats --show-config | |
| - name: Cmake build - OpenVINO | |
| run: cmake --build ${BUILD_DIR} --parallel --config ${{ env.CMAKE_BUILD_TYPE }} | |
| - name: Show ccache stats | |
| run: ccache --show-stats | |
| - name: Cmake install - OpenVINO | |
| run: | | |
| cmake -DCMAKE_INSTALL_PREFIX=${INSTALL_DIR}/openvino_package -P ${BUILD_DIR}/cmake_install.cmake | |
| cmake -DCMAKE_INSTALL_PREFIX=${INSTALL_DIR}/openvino_package -DCOMPONENT=python_wheels -P ${BUILD_DIR}/cmake_install.cmake | |
| - name: Pack Artifacts | |
| run: | | |
| pushd ${INSTALL_DIR} | |
| tar -czvf ${BUILD_DIR}/openvino_package.tar.gz * | |
| popd | |
| # | |
| # Upload build artifacts and logs | |
| # | |
| - name: Upload openvino package | |
| if: ${{ always() }} | |
| uses: actions/upload-artifact@v4 | |
| with: | |
| name: openvino_package | |
| path: ${{ env.BUILD_DIR }}/openvino_package.tar.gz | |
| if-no-files-found: 'error' | |
| genai_python_lib: | |
| name: OpenVINO genai extension (cmake + wheel) | |
| needs: [ openvino_download, openvino_build ] | |
| if: | | |
| always() && | |
| (needs.openvino_download.outputs.status == 'success' || needs.openvino_build.result == 'success') | |
| timeout-minutes: 120 | |
| defaults: | |
| run: | |
| shell: bash | |
| runs-on: macos-13 | |
| env: | |
| OV_INSTALL_DIR: ${{ github.workspace }}/ov | |
| MACOSX_DEPLOYMENT_TARGET: '11.0' | |
| steps: | |
| - name: Clone openvino.genai | |
| uses: actions/checkout@v4 | |
| with: | |
| submodules: recursive | |
| - name: Setup Python ${{ env.PYTHON_VERSION }} | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: ${{ env.PYTHON_VERSION }} | |
| cache: 'pip' | |
| - name: Download OpenVINO package | |
| uses: actions/download-artifact@v4 | |
| with: | |
| name: openvino_package | |
| path: ${{ env.OV_INSTALL_DIR }} | |
| - name: Extract OpenVINO packages | |
| run: | | |
| pushd ${OV_INSTALL_DIR} | |
| tar -xzf openvino_package.tar.gz -C ${OV_INSTALL_DIR} --strip-components=1 | |
| popd | |
| - name: Install build dependencies | |
| run: brew install coreutils ninja scons | |
| - name: Build genai | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| cmake -DCMAKE_BUILD_TYPE=Release -S ./ -B ./build/ | |
| cmake --build ./build/ --config Release -j | |
| - name: Test bindings | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| python -m pip install ./thirdparty/openvino_tokenizers/[transformers] -r ./tests/python_tests/requirements.txt --find-links ${OV_INSTALL_DIR}/wheels | |
| python -m pytest -v ./tests/python_tests/test_tokenizer.py::test_set_chat_template | |
| env: | |
| PYTHONPATH: "./build/:$PYTHONPATH" | |
| - name: Test bindings (wheel) | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| python -m pip install . --verbose --find-links ${OV_INSTALL_DIR}/wheels | |
| python -c "from openvino_genai import LLMPipeline" | |
| python -m pip install ./tools/who_what_benchmark --find-links ${OV_INSTALL_DIR}/wheels | |
| python -m pytest -v ./tests/python_tests/ --ignore ./tests/python_tests/test_whisper_pipeline.py --ignore ./tests/python_tests/test_vlm_pipeline.py -k "not test_set_chat_template" | |
| genai_python_lib_whisper: | |
| name: OpenVINO genai extension whisper tests (cmake + wheel) | |
| needs: [ openvino_download, openvino_build ] | |
| if: | | |
| always() && | |
| (needs.openvino_download.outputs.status == 'success' || needs.openvino_build.result == 'success') | |
| timeout-minutes: 90 | |
| defaults: | |
| run: | |
| shell: bash | |
| runs-on: macos-13 | |
| env: | |
| OV_INSTALL_DIR: ${{ github.workspace }}/ov | |
| MACOSX_DEPLOYMENT_TARGET: '11.0' | |
| steps: | |
| - name: Clone openvino.genai | |
| uses: actions/checkout@v4 | |
| with: | |
| submodules: recursive | |
| - name: Setup Python ${{ env.PYTHON_VERSION }} | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: ${{ env.PYTHON_VERSION }} | |
| cache: 'pip' | |
| - name: Download OpenVINO package | |
| uses: actions/download-artifact@v4 | |
| with: | |
| name: openvino_package | |
| path: ${{ env.OV_INSTALL_DIR }} | |
| - name: Extract OpenVINO packages | |
| run: | | |
| pushd ${OV_INSTALL_DIR} | |
| tar -xzf openvino_package.tar.gz -C ${OV_INSTALL_DIR} --strip-components=1 | |
| popd | |
| - name: Install build dependencies | |
| run: brew install coreutils ninja scons | |
| - name: Build genai | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| cmake -DCMAKE_BUILD_TYPE=Release -S ./ -B ./build/ | |
| cmake --build ./build/ --config Release --target py_openvino_genai -j | |
| - name: Test bindings | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| python -m pip install ./thirdparty/openvino_tokenizers/[transformers] -r ./tests/python_tests/requirements.txt --find-links ${OV_INSTALL_DIR}/wheels | |
| python -m pytest -v ./tests/python_tests/test_whisper_pipeline.py -k test_smoke | |
| env: | |
| PYTHONPATH: "./build/:$PYTHONPATH" | |
| - name: Test bindings (wheel) | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| python -m pip install . --verbose --find-links ${OV_INSTALL_DIR}/wheels | |
| python -c "from openvino_genai import LLMPipeline" | |
| python -m pip install ./tools/who_what_benchmark --find-links ${OV_INSTALL_DIR}/wheels | |
| python -m pytest -v ./tests/python_tests/test_whisper_pipeline.py -k "not test_smoke" | |
| genai_package: | |
| name: OpenVINO genai extension (install to OpenVINO package) | |
| strategy: | |
| matrix: | |
| build-type: [Release, Debug] | |
| needs: [ openvino_download, openvino_build ] | |
| if: | | |
| always() && | |
| (needs.openvino_download.outputs.status == 'success' || needs.openvino_build.result == 'success') | |
| timeout-minutes: 30 | |
| defaults: | |
| run: | |
| shell: bash | |
| runs-on: macos-13 | |
| env: | |
| OV_INSTALL_DIR: ${{ github.workspace }}/ov | |
| steps: | |
| - name: Clone openvino.genai | |
| uses: actions/checkout@v4 | |
| with: | |
| submodules: recursive | |
| - name: Setup Python ${{ env.PYTHON_VERSION }} | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: ${{ env.PYTHON_VERSION }} | |
| cache: 'pip' | |
| - name: Download OpenVINO package | |
| uses: actions/download-artifact@v4 | |
| with: | |
| name: openvino_package | |
| path: ${{ env.OV_INSTALL_DIR }} | |
| - name: Extract OpenVINO packages | |
| run: | | |
| pushd ${OV_INSTALL_DIR} | |
| tar -xzf openvino_package.tar.gz -C ${OV_INSTALL_DIR} --strip-components=1 | |
| popd | |
| - name: Install build dependencies | |
| run: brew install coreutils scons | |
| - name: Build genai | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| cmake -DCMAKE_BUILD_TYPE=${{ matrix.build-type }} -S ./ -B ./build/ | |
| cmake --build ./build/ --config ${{ matrix.build-type }} --target package -j | |
| - name: Build and Install dependencies | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| python -m pip install ./thirdparty/openvino_tokenizers/[transformers] --find-links ${OV_INSTALL_DIR}/wheels | |
| python -m pip install -r ./samples/requirements.txt --find-links ${OV_INSTALL_DIR}/wheels | |
| optimum-cli export openvino --trust-remote-code --model TinyLlama/TinyLlama-1.1B-Chat-v1.0 TinyLlama-1.1B-Chat-v1.0 | |
| optimum-cli export openvino --trust-remote-code --model openai/whisper-tiny whisper-tiny | |
| - name: Install samples | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| cmake --install ./build/ --config ${{ matrix.build-type }} --prefix ${OV_INSTALL_DIR} | |
| - name: Build samples (Release) | |
| if: ${{ 'Release' == matrix.build-type }} # build_samples enforces Release build | |
| run: | | |
| ${OV_INSTALL_DIR}/samples/cpp/build_samples.sh -i ${{ github.workspace }}/s\ pace | |
| - name: Build samples (Debug) | |
| if: ${{ 'Release' != matrix.build-type }} | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| cmake -DCMAKE_BUILD_TYPE=${{ matrix.build-type }} -S ${OV_INSTALL_DIR}/samples/cpp/ -B ./samples\ build/ | |
| cmake --build ./samples\ build/ --config ${{ matrix.build-type }} -j | |
| cmake --install ./samples\ build/ --config ${{ matrix.build-type }} --component samples_bin --prefix s\ pace | |
| - name: Test C++ samples (greedy_causal_lm) | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| ${{ github.workspace }}/s\ pace/samples_bin/greedy_causal_lm ./TinyLlama-1.1B-Chat-v1.0/ "" | |
| timeout-minutes: 1 | |
| - name: Test C++ samples (whisper_speech_recognition) | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| wget https://storage.openvinotoolkit.org/models_contrib/speech/2021.2/librispeech_s5/how_are_you_doing_today.wav | |
| ${{ github.workspace }}/s\ pace/samples_bin/whisper_speech_recognition ./whisper-tiny/ how_are_you_doing_today.wav | |
| timeout-minutes: 1 | |
| - name: Test python samples (multinomial_causal_lm) | |
| if: ${{ 'Release' == matrix.build-type }} # Python bindings can be built in Release only | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| ${OV_INSTALL_DIR}/samples/python/multinomial_causal_lm/multinomial_causal_lm.py ./TinyLlama-1.1B-Chat-v1.0/ 0 | |
| timeout-minutes: 1 | |
| - name: Test python samples (whisper_speech_recognition) | |
| if: ${{ 'Release' == matrix.build-type }} # Python bindings can be built in Release only | |
| run: | | |
| source ${OV_INSTALL_DIR}/setupvars.sh | |
| ${OV_INSTALL_DIR}/samples/python/whisper_speech_recognition/whisper_speech_recognition.py ./whisper-tiny/ how_are_you_doing_today.wav | |
| timeout-minutes: 1 | |
| Overall_Status: | |
| name: ci/gha_overall_status_macos | |
| needs: [openvino_download, openvino_build, genai_python_lib, genai_package, genai_python_lib_whisper] | |
| if: ${{ always() }} | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Check status of all jobs | |
| if: >- | |
| ${{ | |
| contains(needs.*.result, 'failure') || | |
| contains(needs.*.result, 'cancelled') | |
| }} | |
| run: exit 1 |