From 5c2fd295fa1057c2fa311938e3a9a9877e56d4fe Mon Sep 17 00:00:00 2001 From: FrAnCOisCokELaER Date: Sun, 12 Dec 2021 14:41:50 +0100 Subject: [PATCH 1/2] set mkl version to mkl==2021.4.0 --- .github/workflows/tpu-tests.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/tpu-tests.yml b/.github/workflows/tpu-tests.yml index b6b258627607..d3d6e61935c0 100644 --- a/.github/workflows/tpu-tests.yml +++ b/.github/workflows/tpu-tests.yml @@ -54,7 +54,7 @@ jobs: ## Install openblas, mkl, gsutil sudo apt-get install -y libopenblas-dev libomp5-10 libomp5 - pip install mkl requests gsutil + pip install mkl==2021.4.0 requests gsutil ## Install torch & xla curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py From 434ecfa4541d985fe539c62f5f96222eee6f7fd8 Mon Sep 17 00:00:00 2001 From: FrAnCOisCokELaER Date: Sun, 12 Dec 2021 17:08:04 +0100 Subject: [PATCH 2/2] add a comment ( reference to 2350 issue ) --- .github/workflows/tpu-tests.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/tpu-tests.yml b/.github/workflows/tpu-tests.yml index d3d6e61935c0..beb6ab17485c 100644 --- a/.github/workflows/tpu-tests.yml +++ b/.github/workflows/tpu-tests.yml @@ -49,6 +49,7 @@ jobs: restore-keys: | ${{ steps.get-date.outputs.date }}-pytorch-${{ runner.os }}-${{ matrix.xla-version }}- + # mkl version fixed due to https://github.com/pytorch/ignite/issues/2350 - name: Install Torch XLA and others run: |