forked from openvinotoolkit/openvino
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'master' into mateuszm/slicescatter/cpu
- Loading branch information
Showing
114 changed files
with
27,252 additions
and
45,343 deletions.
There are no files selected for viewing
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1 +1 @@ | ||
| pr-26336 | ||
| pr-26588 |
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
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
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
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
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
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
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
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,116 @@ | ||
| name: JAX/Flax Models tests | ||
|
|
||
| on: | ||
| workflow_call: | ||
| inputs: | ||
| runner: | ||
| description: 'Machine on which the tests would run' | ||
| type: string | ||
| required: true | ||
| container: | ||
| description: 'JSON to be converted to the value of the "container" configuration for the job' | ||
| type: string | ||
| required: false | ||
| default: '{"image": null}' | ||
| model_scope: | ||
| description: 'Scope of models for testing.' | ||
| type: string | ||
| required: true | ||
|
|
||
| permissions: read-all | ||
|
|
||
| jobs: | ||
| JAX_Models_Tests: | ||
| name: JAX/Flax Models tests | ||
| timeout-minutes: ${{ inputs.model_scope == 'precommit' && 35 || 35 }} | ||
| runs-on: ${{ inputs.runner }} | ||
| container: ${{ fromJSON(inputs.container) }} | ||
| defaults: | ||
| run: | ||
| shell: bash | ||
| env: | ||
| DEBIAN_FRONTEND: noninteractive # to prevent apt-get from waiting user input | ||
| OPENVINO_REPO: ${{ github.workspace }}/openvino | ||
| INSTALL_DIR: ${{ github.workspace }}/install | ||
| INSTALL_TEST_DIR: ${{ github.workspace }}/install/tests | ||
| MODEL_HUB_TESTS_INSTALL_DIR: ${{ github.workspace }}/install/tests/model_hub_tests | ||
| steps: | ||
| - name: Download OpenVINO package | ||
| uses: actions/download-artifact@fa0a91b85d4f404e444e00e005971372dc801d16 # v4.1.8 | ||
| with: | ||
| name: openvino_package | ||
| path: ${{ env.INSTALL_DIR }} | ||
|
|
||
| - name: Download OpenVINO tokenizers extension | ||
| uses: actions/download-artifact@fa0a91b85d4f404e444e00e005971372dc801d16 # v4.1.8 | ||
| with: | ||
| name: openvino_tokenizers_wheel | ||
| path: ${{ env.INSTALL_DIR }} | ||
|
|
||
| - name: Download OpenVINO tests package | ||
| uses: actions/download-artifact@fa0a91b85d4f404e444e00e005971372dc801d16 # v4.1.8 | ||
| with: | ||
| name: openvino_tests | ||
| path: ${{ env.INSTALL_TEST_DIR }} | ||
|
|
||
| # Needed as ${{ github.workspace }} is not working correctly when using Docker | ||
| - name: Setup Variables | ||
| run: | | ||
| echo "OPENVINO_REPO=$GITHUB_WORKSPACE/openvino" >> "$GITHUB_ENV" | ||
| echo "INSTALL_DIR=$GITHUB_WORKSPACE/install" >> "$GITHUB_ENV" | ||
| echo "INSTALL_TEST_DIR=$GITHUB_WORKSPACE/install/tests" >> "$GITHUB_ENV" | ||
| echo "MODEL_HUB_TESTS_INSTALL_DIR=$GITHUB_WORKSPACE/install/tests/model_hub_tests" >> "$GITHUB_ENV" | ||
| - name: Extract OpenVINO packages | ||
| run: | | ||
| pushd ${INSTALL_DIR} | ||
| tar -xzf openvino_package.tar.gz -C ${INSTALL_DIR} | ||
| popd | ||
| pushd ${INSTALL_TEST_DIR} | ||
| tar -xzf openvino_tests.tar.gz -C ${INSTALL_DIR} | ||
| popd | ||
| - name: Fetch setup_python action | ||
| uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7 | ||
| with: | ||
| sparse-checkout: | | ||
| .github/actions/setup_python/action.yml | ||
| sparse-checkout-cone-mode: false | ||
| path: 'openvino' | ||
|
|
||
| - name: Setup Python 3.11 | ||
| uses: ./openvino/.github/actions/setup_python | ||
| with: | ||
| version: '3.11' | ||
| should-setup-pip-paths: 'false' | ||
| self-hosted-runner: ${{ contains(inputs.runner, 'aks') }} | ||
|
|
||
| - name: Install OpenVINO Python wheels | ||
| run: | | ||
| # To enable pytest parallel features | ||
| python3 -m pip install pytest-xdist[psutil] | ||
| python3 -m pip install ${INSTALL_DIR}/tools/openvino-* | ||
| python3 -m pip install ${INSTALL_DIR}/openvino_tokenizers-* | ||
| - name: Install JAX tests requirements for precommit | ||
| run: | | ||
| python3 -m pip install -r ${MODEL_HUB_TESTS_INSTALL_DIR}/jax/requirements.txt | ||
| - name: JAX/Flax Models Tests from Hugging Face | ||
| if: ${{ inputs.model_scope == 'precommit' || inputs.model_scope == 'nightly' }} | ||
| run: | | ||
| export PYTHONPATH=${MODEL_HUB_TESTS_INSTALL_DIR}:$PYTHONPATH | ||
| python3 -m pytest ${MODEL_HUB_TESTS_INSTALL_DIR}/jax/ -m ${TYPE} --html=${INSTALL_TEST_DIR}/TEST-jax_model_${{ inputs.model_scope }}_tests.html --self-contained-html -v | ||
| env: | ||
| TYPE: ${{ inputs.model_scope == 'precommit' && 'precommit' || 'nightly' }} | ||
| TEST_DEVICE: CPU | ||
|
|
||
| - name: Upload Test Results | ||
| uses: actions/upload-artifact@50769540e7f4bd5e21e526ee35c689e35e0d6874 # v4.4.0 | ||
| if: ${{ !cancelled() }} | ||
| with: | ||
| name: test-results-jax-models-${{ inputs.model_scope }} | ||
| path: | | ||
| ${{ env.INSTALL_TEST_DIR }}/TEST*.html | ||
| if-no-files-found: 'error' |
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
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
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
Oops, something went wrong.