From 5c6edfb0649097eb645167ae54bed61e34946fc3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Szymon=20O=C5=BC=C3=B3g?= <58388001+SzymonOzog@users.noreply.github.com> Date: Tue, 22 Aug 2023 10:54:07 +0200 Subject: [PATCH] use triton tests --- .github/workflows/test.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index efbcd5be9c..4a2812245b 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -263,7 +263,7 @@ jobs: - name: Check Device.DEFAULT run: python -c "from tinygrad.lazy import Device; assert Device.DEFAULT in ['LLVM','CLANG','CUDA','GPU'], Device.DEFAULT" - name: Run pytest (not cuda) - if: matrix.backend!='cuda' && matrix.backend!='ptx' + if: matrix.backend!='cuda' && matrix.backend!='ptx' && matrix.backend!='triton' run: python -m pytest -n=auto test/ -k '${{matrix.backend=='llvm'&&'not (test_nn.py and test_conv_transpose2d)'||'test'}}' -m 'not exclude_${{matrix.backend}}' - name: Run pytest (cuda) if: matrix.backend=='cuda'