From faf424038e1e72aae278e232d22e64d71bed114b Mon Sep 17 00:00:00 2001 From: George Hotz Date: Fri, 24 Nov 2023 12:35:20 -0800 Subject: [PATCH] two cis --- .github/workflows/test.yml | 19 ++++++++++++++++--- 1 file changed, 16 insertions(+), 3 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index b1d2696..cb928a8 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -23,9 +23,22 @@ jobs: - name: Get code size - name: Train MNIST run: PYTHONPATH="." python mnist.py - - name: Install mypy + torch for testing - run: pip install mypy torch --extra-index-url https://download.pytorch.org/whl/cpu - - name: Test ops and dtype and optim + + tests: + name: Run tests + runs-on: ubuntu-latest + timeout-minutes: 20 + + steps: + - name: Checkout Code + uses: actions/checkout@v3 + - name: Set up Python 3.11 + uses: actions/setup-python@v4 + with: + python-version: 3.11 + - name: Install deps with mypy torch for testing + run: pip install numpy tqdm mypy torch --extra-index-url https://download.pytorch.org/whl/cpu + - name: Test ops dtype optim run: | PYTHONPATH="." python test/test_ops.py PYTHONPATH="." python test/test_dtype.py