Skip to content

Commit

Permalink
Add a CPU check and CI
Browse files Browse the repository at this point in the history
Similar to #13, we
also add a CPU github action. This action will run `pytest` on the repo.

Currently there is only one test, which is the Llama test in
torch_xla_models.

In order to run the test today, we need a HF_TOKEN. I created a personal
read only token and
#14 tracks avoiding
the need for HF_TOKEN, after which I'll need to remember to invalidate
the token.
  • Loading branch information
tengyifei committed Jan 8, 2025
1 parent 5b12c36 commit 47c3dd7
Showing 1 changed file with 31 additions and 0 deletions.
31 changes: 31 additions & 0 deletions .github/workflows/cpu_test.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
name: CPU tests

on:
push:
branches:
- main
pull_request:
schedule:
- cron: "0 8 * * *"

jobs:
lint:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ["3.10", "3.11", "3.12"]
steps:
- uses: actions/checkout@v4
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v3
with:
python-version: ${{ matrix.python-version }}
- 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

0 comments on commit 47c3dd7

Please sign in to comment.