Switching IPU testing to Poplar SDK 3.2 and 3.3.

graphcore-research · Sep 19, 2023 · ab1b9ca · ab1b9ca
1 parent fcdff26
commit ab1b9ca
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/.github/workflows/jax-ipu-tests-internal.yaml b/.github/workflows/jax-ipu-tests-internal.yaml
@@ -37,7 +37,7 @@ jobs:
     runs-on: [self-hosted, Linux, X64, 20.04, Ubuntu]
     strategy:
       matrix:
-        container: ["graphcore/pytorch:3.1.0-ubuntu-20.04", "graphcore/pytorch:3.2.0-ubuntu-20.04"]
+        container: ["graphcore/pytorch:3.2.0-ubuntu-20.04", "graphcore/pytorch:3.3.0-ubuntu-20.04"]
     container:
       image: ${{ matrix.container }}
     timeout-minutes: 10

diff --git a/.github/workflows/jax-ipu-tests-public.yaml b/.github/workflows/jax-ipu-tests-public.yaml
@@ -37,7 +37,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        container: ["graphcore/pytorch:3.1.0-ubuntu-20.04", "graphcore/pytorch:3.2.0-ubuntu-20.04"]
+        container: ["graphcore/pytorch:3.2.0-ubuntu-20.04", "graphcore/pytorch:3.3.0-ubuntu-20.04"]
     container:
       image: ${{ matrix.container }}
     timeout-minutes: 10
@@ -59,6 +59,8 @@ jobs:
       - name: Pip install pre-compiled JAXLIB & build JAX
         run: |
           ls ./wheels/
+          export PATH=$PATH:/opt/poplar/bin
+          export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/poplar/lib
           export SDK_VERSION_SHORT="$(python ./build/ipu/get_poplar_sdk_version.py)"
           echo $SDK_VERSION_SHORT
           pip3 install -U numpy==1.23.5 scipy etils pytest
@@ -68,6 +70,9 @@ jobs:
       # Run IPU specific unit tests
       - name: Run JAX IPU unit tests
         run: |
+          export PATH=$PATH:/opt/poplar/bin
+          export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/poplar/lib
+          export CPATH=$CPATH:/opt/poplar/include
           JAX_IPU_DEVICE_COUNT=2 JAX_IPU_USE_MODEL=true JAX_IPU_MODEL_NUM_TILES=16 pytest --tb=short -vv --log-cli-level=INFO ./tests/ipu/
       # Dockerized workflow known to create issues with self-hosted servers.
       # Solution is to fully cleanup the workspace for the next action.