diff --git a/.github/workflows/fast_tests.yml b/.github/workflows/fast_tests.yml index 4d17885ed8..6e2e5e0751 100644 --- a/.github/workflows/fast_tests.yml +++ b/.github/workflows/fast_tests.yml @@ -40,11 +40,11 @@ jobs: diffusers: name: Run tests for optimum.habana.diffusers runs-on: - group: itac-bm-emr-gaudi3-dell-1gaudi + group: denvr-intel-gaudi2-1gaudi container: image: docker://vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest - options: --runtime=habana --shm-size=64G --env HABANA_VISIBLE_DEVICES --env HABANA_VISIBLE_MODULES + #options: --runtime=habana --shm-size=64G --env HABANA_VISIBLE_DEVICES --env HABANA_VISIBLE_MODULES env: OMPI_MCA_btl_vader_single_copy_mechanism: none diff --git a/.github/workflows/slow_tests_gaudi2.yml b/.github/workflows/slow_tests_gaudi2.yml index 54508a5461..634648ead0 100644 --- a/.github/workflows/slow_tests_gaudi2.yml +++ b/.github/workflows/slow_tests_gaudi2.yml @@ -11,27 +11,22 @@ concurrency: jobs: example-diff: name: Test examples differences - runs-on: [self-hosted, linux, x64, gaudi2, nightly] + runs-on: + group: denvr-intel-gaudi2-1gaudi + container: + image: docker://vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest + env: + OMPI_MCA_btl_vader_single_copy_mechanism: none steps: - - name: Checkout - uses: actions/checkout@v2 - - name: Pull image + - name: HL-SMI run: | - docker pull vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest + hl-smi + echo "HABANA_VISIBLE_DEVICES=${HABANA_VISIBLE_DEVICES}" + echo "HABANA_VISIBLE_MODULES=${HABANA_VISIBLE_MODULES}" + - name: Checkout + uses: actions/checkout@v4 - name: Run tests - run: | - docker run \ - --rm \ - -v $PWD:/root/workspace \ - --workdir=/root/workspace \ - --runtime=habana \ - -e HABANA_VISIBLE_DEVICES=all \ - -e OMPI_MCA_btl_vader_single_copy_mechanism=none \ - --cap-add=sys_nice \ - --net=host \ - --ipc=host \ - vault.habana.ai/gaudi-docker/1.22.0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1:latest \ - /bin/bash tests/ci/example_diff_tests.sh + run: /bin/bash tests/ci/example_diff_tests.sh stable-diffusion: name: Test Stable Diffusion if: ${{ !cancelled() && (success() || failure()) }}