diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index dda25bca3f..4a356a00e3 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -36,7 +36,7 @@ jobs: - name: Run mypy run: mypy tinygrad/ --ignore-missing-imports --check-untyped-defs --explicit-package-bases --warn-unreachable - name: Install SLOCCount - run: sudo apt install sloccount + run: sudo apt-get install sloccount - name: Check <5000 lines run: sloccount tinygrad test examples extra; if [ $(sloccount tinygrad | sed -n 's/.*Total Physical Source Lines of Code (SLOC)[ ]*= \([^ ]*\).*/\1/p' | tr -d ',') -gt 5000 ]; then exit 1; fi @@ -111,14 +111,16 @@ jobs: steps: - name: Checkout Code uses: actions/checkout@v3 + - name: Find faster apt mirror + uses: vegardit/fast-apt-mirror.sh@v1 - name: Update packages run: | wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB | gpg --dearmor | sudo tee /usr/share/keyrings/oneapi-archive-keyring.gpg > /dev/null echo "deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main" | sudo tee /etc/apt/sources.list.d/oneAPI.list - sudo apt update + sudo apt-get update - name: Install OpenCL #run: sudo apt-get install -y pocl-opencl-icd - run: sudo apt install -y intel-oneapi-runtime-compilers intel-oneapi-runtime-opencl + run: sudo apt-get install -y intel-oneapi-runtime-compilers intel-oneapi-runtime-opencl - name: Set up Python 3.8 uses: actions/setup-python@v4 with: @@ -222,19 +224,20 @@ jobs: key: ${{ matrix.backend }} - name: Set env run: printf "${{ matrix.backend == 'llvm' && 'ENABLE_METHOD_CACHE=1\nLLVM=1' || matrix.backend == 'clang' && 'CLANG=1\nENABLED_METHOD_CACHE=1' || matrix.backend == 'gpu' && 'GPU=1' || matrix.backend == 'cuda' && 'FORWARD_ONLY=1\nJIT=1\nOPT=2\nCUDA=1\nCUDACPU=1\n'}}" >> $GITHUB_ENV + - name: Find faster apt mirror + uses: vegardit/fast-apt-mirror.sh@v1 - name: Install packages (gpu) if: matrix.backend == 'gpu' run: | wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB | gpg --dearmor | sudo tee /usr/share/keyrings/oneapi-archive-keyring.gpg > /dev/null echo "deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main" | sudo tee /etc/apt/sources.list.d/oneAPI.list - sudo apt update && \ - sudo apt install -y intel-oneapi-runtime-compilers intel-oneapi-runtime-opencl + sudo apt-get update -y && \ + sudo apt-get install -y intel-oneapi-runtime-compilers intel-oneapi-runtime-opencl - name: Install packages (cuda) if: matrix.backend == 'cuda' run: | - export DEBIAN_FRONTEND=noninteractive - sudo apt update -y && \ - sudo apt install -y --no-install-recommends git g++ cmake ninja-build llvm-15-dev zlib1g-dev libglew-dev flex bison libfl-dev libboost-thread-dev libboost-filesystem-dev nvidia-cuda-toolkit-gcc + sudo apt-get update -y && \ + sudo apt-get install -y --no-install-recommends git g++ cmake ninja-build llvm-15-dev zlib1g-dev libglew-dev flex bison libfl-dev libboost-thread-dev libboost-filesystem-dev nvidia-cuda-toolkit-gcc - name: Cache gpuocelot if: matrix.backend == 'cuda' id: cache-build @@ -270,4 +273,4 @@ jobs: 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' - 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 \ No newline at end of file + 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