diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 76fa9131b4..615d845d32 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -261,7 +261,7 @@ jobs: - name: Install dependencies run: pip install -e '.[testing${{matrix.backend=='llvm'&&',llvm'||matrix.backend=='cuda'&&',cuda'||matrix.backend=='ptx'&&',cuda'||matrix.backend=='triton'&&',triton'||''}}]' --extra-index-url https://download.pytorch.org/whl/cpu - name: Check Device.DEFAULT - run: python -c "from tinygrad.lazy import Device; assert Device.DEFAULT in ['LLVM','CLANG','CUDA','GPU', 'TRITON'], Device.DEFAULT" + run: python -c "from tinygrad.lazy import Device; assert Device.DEFAULT in ['LLVM','CLANG','CUDA','GPU', 'TRITON'], Device.DEFAULT"; print('Using device = ', Device.DEFAULT) - name: Run pytest (not cuda) 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}}'