Fix mmaped in jit (#2225) #7
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Unit Tests | |
| on: | |
| push: | |
| branches: | |
| - master | |
| pull_request: | |
| workflow_dispatch: | |
| jobs: | |
| linter: | |
| name: Linters | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 20 | |
| steps: | |
| - name: Checkout Code | |
| uses: actions/checkout@v3 | |
| - name: Set up Python 3.8 | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: 3.8 | |
| - name: Cache python packages | |
| uses: actions/cache@v3 | |
| with: | |
| path: ${{ env.Python3_ROOT_DIR }}/lib/python3.8/site-packages | |
| key: linting-packages-${{ hashFiles('**/setup.py') }}-3.8 | |
| - name: Install dependencies | |
| run: pip install -e '.[linting,testing]' --extra-index-url https://download.pytorch.org/whl/cpu | |
| - name: Repo line count | |
| run: python sz.py | |
| - name: Lint with pylint | |
| run: python -m pylint --disable=all -e W0311 -e C0303 --jobs=0 --indent-string=' ' **/*.py | |
| - name: Lint with flake8 | |
| run: python -m flake8 . --statistics -j4 | |
| - name: Lint with ruff | |
| run: | | |
| pip3 install --upgrade --force-reinstall ruff | |
| python3 -m ruff . | |
| - name: Lint tinygrad with pylint | |
| run: python -m pylint tinygrad/ | |
| - name: Run mypy | |
| run: python -m mypy | |
| - name: Install SLOCCount | |
| run: sudo apt 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 | |
| - name: Test Docs | |
| run: python docs/abstractions.py | |
| - name: Test Quickstart | |
| run: awk '/```python/{flag=1;next}/```/{flag=0}flag' docs/quickstart.md > quickstart.py && PYTHONPATH=. python quickstart.py | |
| - name: Fuzz Test symbolic | |
| run: python test/external/fuzz_symbolic.py | |
| - name: Fuzz Test shapetracker | |
| run: PYTHONPATH="." python test/external/fuzz_shapetracker.py | |
| - name: Use as an external package | |
| run: | | |
| mkdir $HOME/test_external_dir | |
| cd $HOME/test_external_dir | |
| python -m venv venv | |
| source venv/bin/activate | |
| pip install $GITHUB_WORKSPACE | |
| python -c "from tinygrad.tensor import Tensor; print(Tensor([1,2,3,4,5]))" | |
| testcpuimagenet: | |
| name: CPU and ImageNet to C Tests | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 20 | |
| steps: | |
| - name: Checkout Code | |
| uses: actions/checkout@v3 | |
| - name: Set up Python 3.8 | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: 3.8 | |
| - name: Cache python packages | |
| uses: actions/cache@v3 | |
| with: | |
| path: ${{ env.Python3_ROOT_DIR }}/lib/python3.8/site-packages | |
| key: testing-packages-${{ hashFiles('**/setup.py') }} | |
| - name: Install Dependencies | |
| run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu | |
| - name: Run Pytest | |
| run: python -m pytest -n=auto test/ -k "not (test_efficientnet and models/test_train.py)" --durations=20 | |
| - name: Compile EfficientNet to C | |
| run: PYTHONPATH="." CLANG=1 python examples/compile_efficientnet.py > recognize.c | |
| - name: Compile C to native | |
| run: clang -O2 recognize.c -lm -o recognize | |
| - name: Test EfficientNet | |
| run: curl https://media.istockphoto.com/photos/hen-picture-id831791190 | ./recognize | grep hen | |
| testtorch: | |
| name: Torch Tests | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 20 | |
| steps: | |
| - name: Checkout Code | |
| uses: actions/checkout@v3 | |
| - name: Set up Python 3.11 | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: 3.11 | |
| - name: Cache python packages | |
| uses: actions/cache@v3 | |
| with: | |
| path: ${{ env.Python3_ROOT_DIR }}/lib/python3.11/site-packages | |
| key: testing-packages-${{ hashFiles('**/setup.py') }} | |
| - name: Install Dependencies | |
| run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu | |
| - name: Run Pytest | |
| run: TORCH=1 python -m pytest -n=auto test/ --durations=20 | |
| - name: Run ONNX | |
| run: TORCH=1 python -m pytest -n=auto test/external/external_test_onnx_backend.py | |
| testopencl: | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| task: [optimage, openpilot, onnx, optimization] | |
| name: ${{ matrix.task=='optimage'&&'GPU OPT and IMAGE Tests' || matrix.task=='openpilot'&&'openpilot (OpenCL) Tests' || matrix.task=='onnx'&&'ONNX Tests' || matrix.task=='optimization'&&'Optimization Tests' }} | |
| runs-on: ubuntu-20.04 | |
| timeout-minutes: 20 | |
| steps: | |
| - name: Checkout Code | |
| uses: actions/checkout@v3 | |
| - name: Install OpenCL | |
| run: | | |
| echo "deb [ allow-insecure=yes ] https://apt.repos.intel.com/oneapi all main" | sudo tee /etc/apt/sources.list.d/oneAPI.list | |
| sudo apt update | |
| sudo apt install --allow-unauthenticated -y --no-install-recommends intel-oneapi-runtime-compilers intel-oneapi-runtime-opencl | |
| - name: Set up Python 3.11 | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: 3.11 | |
| - name: Cache python packages | |
| uses: actions/cache@v3 | |
| with: | |
| path: ${{ env.Python3_ROOT_DIR }}/lib/python3.11/site-packages | |
| key: testing-packages-${{ hashFiles('**/setup.py') }} | |
| - name: Install Dependencies | |
| run: pip install -e '.[testing]' --extra-index-url https://download.pytorch.org/whl/cpu | |
| - if: ${{ matrix.task == 'optimage' }} | |
| name: Run Optimizer Test (OPT 2 and 3) | |
| run: | | |
| PYTHONPATH="." OPT=2 GPU=1 python -m pytest -n=auto test/external/external_test_opt.py | |
| PYTHONPATH="." OPT=3 GPU=1 python -m pytest -n=auto test/external/external_test_opt.py | |
| - if: ${{ matrix.task == 'optimage'}} | |
| name: Test WINO=1 | |
| run: GPU=1 DEBUG=2 WINO=1 python3 test/test_ops.py TestOps.test_simple_conv2d | |
| - if: ${{ matrix.task == 'optimage'}} | |
| name: Test GPU IMAGE=1 ops | |
| run: GPU=1 IMAGE=1 python -m pytest -n=auto test/test_ops.py | |
| - if: ${{ matrix.task == 'optimage'}} | |
| name: Test GPU IMAGE=2 ops | |
| run: GPU=1 IMAGE=2 python -m pytest -n=auto test/test_ops.py | |
| - if: ${{ matrix.task == 'openpilot' }} | |
| name: Test openpilot model compile and size | |
| run: | | |
| DEBUG=2 ALLOWED_KERNEL_COUNT=207 VALIDTEST=1 FLOAT16=1 DEBUGCL=1 GPU=1 IMAGE=2 python openpilot/compile.py | |
| python -c 'import os; assert os.path.getsize("/tmp/output.thneed") < 100_000_000' | |
| - if: ${{ matrix.task == 'openpilot' }} | |
| name: Test openpilot model correctness (float32) | |
| run: DEBUGCL=1 GPU=1 IMAGE=2 python openpilot/compile.py | |
| - if: ${{ matrix.task == 'openpilot' }} | |
| name: Test openpilot model correctness (float32, new compiler) | |
| run: DEBUGCL=1 FLOAT16=0 python3 openpilot/compile2.py | |
| - if: ${{ matrix.task == 'openpilot' }} | |
| name: Test openpilot alt model correctness (float32) | |
| run: DEBUGCL=1 GPU=1 IMAGE=2 python openpilot/compile.py https://github.com/commaai/openpilot/raw/3799fe46b3a629e491d4b8498b8ae83e4c88c304/selfdrive/modeld/models/supercombo.onnx | |
| - if: ${{ matrix.task == 'openpilot' }} | |
| name: Test tensor core ops | |
| run: GPU=1 TC=2 python -m pytest -n=auto test/test_ops.py | |
| - if: ${{ matrix.task == 'openpilot' }} | |
| name: Test multigpu | |
| run: | | |
| PYTHONPATH="." python test/external/dist/test_world.py | |
| PYTHONPATH="." python test/external/dist/test_collectives.py | |
| - if: ${{ matrix.task == 'onnx' }} | |
| name: Test ONNX (CPU) | |
| run: CPU=1 python -m pytest -n=auto test/external/external_test_onnx_backend.py --durations=20 | |
| - if: ${{ matrix.task == 'onnx' }} | |
| name: Test ONNX (GPU) | |
| run: GPU=1 python -m pytest -n=auto test/external/external_test_onnx_backend.py --durations=20 | |
| - if: ${{ matrix.task == 'onnx' }} | |
| name: Test ONNX (CLANG) | |
| run: CLANG=1 python -m pytest -n=auto test/external/external_test_onnx_backend.py --durations=20 | |
| - if: ${{ matrix.task == 'optimization' }} | |
| name: Test Action Space | |
| run: PYTHONPATH="." GPU=1 python3 extra/optimization/get_action_space.py | |
| - if: ${{ matrix.task == 'optimization' }} | |
| name: Test Beam Search | |
| run: PYTHONPATH="." GPU=1 IGNORE_BEAM_CACHE=1 python3 -m pytest extra/optimization/test_beam_search.py | |
| testmetalwebgpu: | |
| name: Metal and WebGPU Tests | |
| runs-on: macos-13 | |
| timeout-minutes: 20 | |
| steps: | |
| - name: Checkout Code | |
| uses: actions/checkout@v3 | |
| - name: Set up Python 3.11 | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: 3.11 | |
| - name: Cache python packages | |
| uses: actions/cache@v3 | |
| with: | |
| path: ${{ env.Python3_ROOT_DIR }}/lib/python3.11/site-packages | |
| key: metal-webgpu-testing-packages-${{ hashFiles('**/setup.py') }} | |
| - name: Install Dependencies | |
| run: pip install -e '.[metal,webgpu,testing]' --extra-index-url https://download.pytorch.org/whl/cpu | |
| - name: Test LLaMA compile speed | |
| run: PYTHONPATH="." METAL=1 python test/external/external_test_speed_llama.py | |
| #- name: Run dtype test | |
| # run: DEBUG=4 METAL=1 python -m pytest -n=auto test/test_dtype.py | |
| # dtype test has issues on test_half_to_int8 | |
| - name: Check Device.DEFAULT (METAL) and print some source | |
| run: | | |
| METAL=1 python -c "from tinygrad.ops import Device; assert Device.DEFAULT == 'METAL', Device.DEFAULT" | |
| METAL=1 DEBUG=4 FORWARD_ONLY=1 python3 test/test_ops.py TestOps.test_add | |
| - name: Run metal ops test | |
| run: DEBUG=2 METAL=1 python -m pytest -n=auto test/test_ops.py | |
| - name: Run JIT test | |
| run: DEBUG=2 METAL=1 python -m pytest -n=auto test/test_jit.py | |
| - name: Run symbolic shapetracker test | |
| run: METAL=1 python -m pytest -n=auto test/test_symbolic_shapetracker.py test/test_symbolic_ops.py test/test_symbolic_jit.py | |
| - name: Run ONNX | |
| run: METAL=1 python -m pytest -n=auto test/external/external_test_onnx_backend.py | |
| - name: Run whisper test | |
| run: METAL=1 python -m pytest test/models/test_whisper.py | |
| - name: Check Device.DEFAULT (WEBGPU) and print some source | |
| run: | | |
| WEBGPU=1 python -c "from tinygrad.ops import Device; assert Device.DEFAULT == 'WEBGPU', Device.DEFAULT" | |
| WEBGPU=1 DEBUG=4 FORWARD_ONLY=1 python3 test/test_ops.py TestOps.test_add | |
| - name: Run linearizer and tensor core test | |
| run: METAL=1 python -m pytest -n=auto test/test_linearizer.py | |
| - name: Test tensor core reshape-only ops | |
| run: METAL=1 TC=2 python -m pytest -n=auto test/test_ops.py | |
| #- name: Run webgpu pytest | |
| # run: WEBGPU=1 WGPU_BACKEND_TYPE=Metal python -m pytest -n=auto | |
| - name: Run webgpu dtype tests | |
| run: WEBGPU=1 WGPU_BACKEND_TYPE=Metal python -m pytest -n=auto test/test_dtype.py | |
| - name: Build WEBGPU Efficientnet | |
| run: WEBGPU=1 WGPU_BACKEND_TYPE=Metal python -m examples.compile_efficientnet | |
| - name: Install Puppeteer | |
| run: npm install puppeteer | |
| - name: Run WEBGPU Efficientnet | |
| run: node test/test_webgpu.js | |
| tests: | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| backend: [llvm, clang, gpu, cuda, triton] #, ptx] | |
| name: Tests on (${{ matrix.backend }}) | |
| runs-on: ${{ matrix.backend == 'gpu' && 'ubuntu-20.04' || 'ubuntu-latest' }} | |
| timeout-minutes: 20 | |
| steps: | |
| - name: Checkout Code | |
| uses: actions/checkout@v3 | |
| - name: Set up Python 3.11 | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: 3.11 | |
| - name: Cache python packages | |
| uses: actions/cache@v3 | |
| with: | |
| path: ${{ env.Python3_ROOT_DIR }}/lib/python3.11/site-packages | |
| key: ${{ matrix.backend }}-packages-${{ hashFiles('**/setup.py') }} | |
| - name: Set env | |
| run: printf "${{ matrix.backend == 'llvm' && 'LLVM=1' || matrix.backend == 'clang' && 'CLANG=1' || matrix.backend == 'gpu' && 'GPU=1' || matrix.backend == 'cuda' && 'FORWARD_ONLY=1\nJIT=1\nOPT=2\nCUDA=1\nCUDACPU=1\n' || matrix.backend == 'PTX' && 'FORWARD_ONLY=1\nJIT=1\nOPT=2\nCUDA=1\nCUDACPU=1\nPTX=1' || matrix.backend == 'triton' && 'FORWARD_ONLY=1\nJIT=1\nOPT=2\nCUDA=1\nCUDACPU=1\nTRITON=1\nTRITON_PTXAS_PATH=/usr/bin/ptxas'}}" >> $GITHUB_ENV | |
| - name: Install OpenCL | |
| if: matrix.backend == 'gpu' | |
| run: | | |
| echo "deb [ allow-insecure=yes ] https://apt.repos.intel.com/oneapi all main" | sudo tee /etc/apt/sources.list.d/oneAPI.list | |
| sudo apt update | |
| sudo apt install --allow-unauthenticated -y --no-install-recommends intel-oneapi-runtime-compilers intel-oneapi-runtime-opencl | |
| - name: Install packages (cuda) | |
| if: matrix.backend == 'cuda' || matrix.backend == 'ptx' || matrix.backend == 'triton' | |
| run: | | |
| echo 'Acquire::http::Pipeline-Depth "5";' | sudo tee -a /etc/apt/apt.conf.d/99parallel | |
| 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 | |
| - name: Cache gpuocelot | |
| if: matrix.backend == 'cuda' || matrix.backend == 'ptx' || matrix.backend == 'triton' | |
| id: cache-build | |
| uses: actions/cache@v3 | |
| env: | |
| cache-name: cache-gpuocelot-build | |
| with: | |
| path: ${{ github.workspace }}/gpuocelot/ocelot | |
| key: ubuntu22.04-gpuocelot-18401f4245b27ca4b3af433196583cc81ef84480 | |
| - name: Clone/compile gpuocelot | |
| if: (matrix.backend == 'cuda' || matrix.backend == 'ptx' || matrix.backend == 'triton') && steps.cache-build.outputs.cache-hit != 'true' | |
| run: | | |
| git clone --recurse-submodules https://github.com/gpuocelot/gpuocelot.git ${{ github.workspace }}/gpuocelot | |
| cd ${{ github.workspace }}/gpuocelot/ocelot | |
| git checkout 18401f4245b27ca4b3af433196583cc81ef84480 | |
| mkdir build | |
| cd build | |
| cmake .. -Wno-dev -G Ninja -DOCELOT_BUILD_TOOLS=OFF -DCMAKE_BUILD_ALWAYS=0 | |
| ninja | |
| - name: Install gpuocelot | |
| if: matrix.backend == 'cuda' || matrix.backend == 'ptx' || matrix.backend == 'triton' | |
| run: | | |
| cd ${{ github.workspace }}/gpuocelot/ocelot/build | |
| sudo ninja install -d explain | |
| - 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 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}}' --durations=20 | |
| - name: Run ONNX (only LLVM) | |
| if: matrix.backend == 'llvm' | |
| run: python -m pytest -n=auto test/external/external_test_onnx_backend.py --durations=20 | |
| - name: Run pytest (cuda) | |
| if: matrix.backend=='cuda'||matrix.backend=='ptx'||matrix.backend=='triton' | |
| 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 --durations=20 | |
| #testunicorn: | |
| # name: ARM64 unicorn Test | |
| # runs-on: ubuntu-latest | |
| # timeout-minutes: 20 | |
| # steps: | |
| # - name: Checkout Code | |
| # uses: actions/checkout@v3 | |
| # - name: Set up Python 3.11 | |
| # uses: actions/setup-python@v4 | |
| # with: | |
| # python-version: 3.11 | |
| # - name: Cache python packages | |
| # uses: actions/cache@v3 | |
| # with: | |
| # path: ${{ env.Python3_ROOT_DIR }}/lib/python3.11/site-packages | |
| # key: testing-arm-packages-${{ hashFiles('**/setup.py') }} | |
| # - name: Install cross-assembler | |
| # run: | | |
| # sudo apt update -y | |
| # sudo apt install -y --no-install-recommends gcc-aarch64-linux-gnu | |
| # - name: Install dependencies | |
| # run: pip install -e '.[testing,arm]' --extra-index-url https://download.pytorch.org/whl/cpu | |
| # - name: Test arm | |
| # run: CI=1 ARM64=1 CLANG=1 python -m pytest -n=auto test/ -k 'not (test_nn.py and (test_conv_transpose2d or test_conv2d))' --ignore=test/models --ignore=test/test_speed_v_torch.py --ignore=test/test_net_speed.py --ignore=test/test_specific_conv.py --ignore=test/unit/test_disk_tensor.py |