diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 83b514d645..555b221da2 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -79,7 +79,7 @@ jobs: run: clang -O2 recognize.c -lm -o recognize - name: Test EfficientNet run: curl https://media.istockphoto.com/photos/hen-picture-id831791190 | ./recognize | grep hen - - name: Use as an external pacakge + - name: Use as an external package run: | mkdir $HOME/test_external_dir cd $HOME/test_external_dir @@ -283,13 +283,15 @@ jobs: sudo ninja install - 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 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/Triton-Nightly/pypi/simple/ - - name: Check Device.DEFAULT - run: python -c "from tinygrad.ops import Device; assert Device.DEFAULT in ['LLVM','CLANG','CUDA','GPU'], Device.DEFAULT" + - name: Check Device.DEFAULT and print some source + run: | + python -c "from tinygrad.ops import Device; assert Device.DEFAULT in ['LLVM','CLANG','CUDA','GPU'], Device.DEFAULT" + DEBUG=4 FORWARD_ONLY=1 python3 test/test_ops.py TestOps.test_add - 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}}' - name: Run ONNX (not cuda) - if: matrix.backend!='cuda' && matrix.backend!='ptx' && matrix.backend!='triton' + if: matrix.backend!='cuda' && matrix.backend!='ptx' && matrix.backend!='triton' && matrix.backend!='clang' run: python -m pytest -n=auto test/external/external_test_onnx_backend.py - name: Run pytest (cuda) if: matrix.backend=='cuda' @@ -299,7 +301,7 @@ jobs: run: python -m pytest -n=auto test/ -k 'not (half or test_efficientnet_safetensors) and not (test_conv2d and test_tensor.py)' -m 'not exclude_cuda' --ignore=test/external --ignore=test/models - name: Run pytest (triton) if: matrix.backend=='triton' - run: python -m pytest -v -n=auto test/ -k 'not (half or test_efficientnet_safetensors) and not (test_conv2d and test_tensor.py)' -m 'not exclude_cuda' --ignore=test/external --ignore=test/models + run: python -m pytest -n=auto test/ -k 'not (half or test_efficientnet_safetensors) and not (test_conv2d and test_tensor.py)' -m 'not exclude_cuda' --ignore=test/external --ignore=test/models testunicorn: name: ARM64 unicorn Test