Skip to content

Commit b5e1d3a

Browse files
authored
[CI] [GHA] Switch PyTorch models tests to self-hosted runners (openvinotoolkit#28972)
### Tickets: - *162276*
1 parent 06a6b0c commit b5e1d3a

File tree

3 files changed

+17
-12
lines changed

3 files changed

+17
-12
lines changed

.github/dockerfiles/docker_tag

+1-1
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
pr-29066
1+
pr-28972

.github/dockerfiles/ov_test/ubuntu_22_04_x64/Dockerfile

+7
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,13 @@ RUN apt-get update && \
3232
libhdf5-dev \
3333
# For TF Models tests
3434
wget \
35+
# libGL for PyTorch Models tests
36+
ffmpeg \
37+
libsm6 \
38+
libxext6 \
39+
libgl1 \
40+
libgl1-mesa-glx \
41+
libglib2.0-0 \
3542
&& \
3643
rm -rf /var/lib/apt/lists/*
3744

.github/workflows/ubuntu_22.yml

+9-11
Original file line numberDiff line numberDiff line change
@@ -395,37 +395,35 @@ jobs:
395395
model_scope: 'nightly_hf'
396396
container: '{"image": "${{ fromJSON(needs.docker.outputs.images).ov_test.ubuntu_22_04_x64 }}", "volumes": ["/mount:/mount"]}'
397397

398-
# TODO: Switch back to self-hosted runners
399-
# container:
400-
# image: openvinogithubactions.azurecr.io/dockerhub/ubuntu:22.04
401-
# volumes:
402-
# - /mount:/mount
403398
PyTorch_Models_Tests:
404399
name: PyTorch Models tests
405400
if: ${{ github.event_name != 'schedule' && fromJSON(needs.smart_ci.outputs.affected_components).PyTorch_FE.test }}
406-
needs: [ Build, Smart_CI, Openvino_tokenizers ]
401+
needs: [ Docker, Build, Smart_CI, Openvino_tokenizers ]
407402
uses: ./.github/workflows/job_pytorch_models_tests.yml
408403
with:
409-
runner: 'ubuntu-22.04-8-cores'
404+
runner: 'aks-linux-8-cores-64gb'
410405
model_scope: 'precommit'
406+
container: '{"image": "${{ fromJSON(needs.docker.outputs.images).ov_test.ubuntu_22_04_x64 }}", "volumes": ["/mount:/mount"]}'
411407

412408
PyTorch_Models_Tests_Nightly_Scope1:
413409
name: PyTorch Models Nightly Scope1 tests
414410
if: ${{ github.event_name == 'schedule' }}
415-
needs: [ Build, Smart_CI, Openvino_tokenizers ]
411+
needs: [ Docker, Build, Smart_CI, Openvino_tokenizers ]
416412
uses: ./.github/workflows/job_pytorch_models_tests.yml
417413
with:
418-
runner: 'ubuntu-22.04-16-cores'
414+
runner: 'aks-linux-16-cores-64gb'
419415
model_scope: 'nightly_scope1'
416+
container: '{"image": "${{ fromJSON(needs.docker.outputs.images).ov_test.ubuntu_22_04_x64 }}", "volumes": ["/mount:/mount"]}'
420417

421418
PyTorch_Models_Tests_Nightly_Scope2:
422419
name: PyTorch Models Nightly Scope2 tests
423420
if: ${{ github.event_name == 'schedule' }}
424-
needs: [ Build, Smart_CI, Openvino_tokenizers ]
421+
needs: [ Docker, Build, Smart_CI, Openvino_tokenizers ]
425422
uses: ./.github/workflows/job_pytorch_models_tests.yml
426423
with:
427-
runner: 'ubuntu-22.04-16-cores'
424+
runner: 'aks-linux-16-cores-64gb'
428425
model_scope: 'nightly_scope2'
426+
container: '{"image": "${{ fromJSON(needs.docker.outputs.images).ov_test.ubuntu_22_04_x64 }}", "volumes": ["/mount:/mount"]}'
429427

430428
JAX_Models_Tests_Precommit:
431429
name: JAX/Flax Models tests

0 commit comments

Comments
 (0)