From 40619a4bbc7bf847ac9d78129d29dec405ea0301 Mon Sep 17 00:00:00 2001 From: chenyu Date: Mon, 3 Mar 2025 13:05:24 -0500 Subject: [PATCH] separate workflow for TINY_BACKEND=1 mnist (#9339) * separate workflow for TINY_BACKEND=1 mnist * rebalance --- .github/workflows/test.yml | 24 +++++++++++++++++++++--- 1 file changed, 21 insertions(+), 3 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index e67aa8a02f..1885499ab0 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -146,7 +146,7 @@ jobs: torchbackend: name: Torch Backend Tests runs-on: ubuntu-latest - timeout-minutes: 20 + timeout-minutes: 10 steps: - name: Checkout Code uses: actions/checkout@v4 @@ -169,10 +169,28 @@ jobs: run: PYTHONPATH=. python3 extra/torch_backend/test.py - name: Test one op in torch tests run: PYTHONPATH=. DEBUG=2 python3 extra/torch_backend/torch_tests.py TestTinyBackendPRIVATEUSE1.test_unary_log_tiny_float32 + - name: Test Ops with TINY_BACKEND (expect failure) + run: PYTHONPATH=. LLVM=1 LLVMOPT=0 TINY_BACKEND=1 python3 -m pytest -n auto test/test_ops.py --durations=20 || true + + torchbackendmore: + name: Torch Backend Tests More + runs-on: ubuntu-latest + timeout-minutes: 10 + steps: + - name: Checkout Code + uses: actions/checkout@v4 + - name: Setup Environment + uses: ./.github/actions/setup-tinygrad + with: + key: torch-backend-pillow-torchvision-et-pt + deps: testing_minimal + llvm: 'true' + - name: Install ninja + run: | + sudo apt update || true + sudo apt install -y --no-install-recommends ninja-build - name: Test beautiful_mnist in torch with TINY_BACKEND run: PYTHONPATH=. LLVM=1 TARGET_EVAL_ACC_PCT=96.0 TINY_BACKEND=1 python3 examples/other_mnist/beautiful_mnist_torch.py - - name: Test Ops with TINY_BACKEND (expect failure) - run: PYTHONPATH=. LLVM=1 LLVMOPT=0 TINY_BACKEND=1 python3 -m pytest -n auto test/test_ops.py || true - name: Test some torch tests (expect failure) run: PYTHONPATH=. python3 -m pytest extra/torch_backend/torch_tests.py -v --tb=no || true