From afcbb62aea05a5e83e486d079f909148b756ff88 Mon Sep 17 00:00:00 2001 From: Alvaro Moran Date: Wed, 28 Feb 2024 12:18:00 +0000 Subject: [PATCH] WIP --- .../{test_tpu_tgi.yml => test-pytorch-xla-tpu-tgi.yml} | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) rename .github/workflows/{test_tpu_tgi.yml => test-pytorch-xla-tpu-tgi.yml} (92%) diff --git a/.github/workflows/test_tpu_tgi.yml b/.github/workflows/test-pytorch-xla-tpu-tgi.yml similarity index 92% rename from .github/workflows/test_tpu_tgi.yml rename to .github/workflows/test-pytorch-xla-tpu-tgi.yml index ab06082b..f089bf35 100644 --- a/.github/workflows/test_tpu_tgi.yml +++ b/.github/workflows/test-pytorch-xla-tpu-tgi.yml @@ -21,12 +21,12 @@ jobs: runs-on: optimum-tpu container: image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.2.0_3.10_tpuvm - options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/ --privileged + options: --shm-size "16gb" --ipc host --privileged env: PJRT_DEVICE: TPU steps: - name: Checkout - uses: actions/checkout@v2 + uses: actions/checkout@v4 - name: Create venv and install Pytorch/XLA run: | echo "PJRT_DEVICE set to $PJRT_DEVICE"