tpu-test-workflow.yml

name: TPU Test
run-name: ${{ github.actor }} is testing self-hosted runners on TPUs
on: [push]
jobs:
  tpu-test:
    runs-on: runner-set
    steps:
      - uses: actions/checkout@v4
        with:
          ref: r2.0
      - uses: actions/setup-python@v4
        with:
          python-version: '3.8'
      - run: pip install numpy
      - run: pip install pyyaml
      - run: pip install torchvision
      - run: pip install torch~=2.0.0 https://storage.googleapis.com/tpu-pytorch/wheels/tpuvm/torch_xla-2.0-cp38-cp38-linux_x86_64.whl
      - run: python3 test/test_operations.py -v
        #whatever