Skip to content

Add a CPU check and CI (#15) #13

Add a CPU check and CI (#15)

Add a CPU check and CI (#15) #13

Workflow file for this run

name: CPU tests
on:
push:
branches:
- main
pull_request:
schedule:
- cron: "0 8 * * *"
jobs:
pytest:
runs-on: ubuntu-22.04
strategy:
matrix:
python-version: ["3.10", "3.11"]
container:
image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:nightly_${{ matrix.python-version }}_tpuvm
steps:
- uses: actions/checkout@v4
- name: Install dev dependencies
run: |
python -m pip install --upgrade pip
pip install -e '.[dev]'
- name: Run PyTest
run: |
# TODO(https://github.com/AI-Hypercomputer/torchprime/issues/14): Remove and burn the token.
export HF_TOKEN=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
pytest