mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-09 15:08:02 -05:00
4eae4b0ce6
* unify adreno autogen with mesa * gen pm4 * TestTiny::test_plus works * add a6xx enums * IMAGE=2 TestTiny::test_gemm works * remove adreno from CI * cleanup
177 lines
8.3 KiB
YAML
177 lines
8.3 KiB
YAML
name: Autogen
|
|
env:
|
|
# increment this when downloads substantially change to avoid the internet
|
|
CACHE_VERSION: '13'
|
|
CAPTURE_PROCESS_REPLAY: 1
|
|
GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
|
PYTHONPATH: ${{ github.workspace }}
|
|
|
|
on:
|
|
push:
|
|
branches:
|
|
- master
|
|
pull_request:
|
|
paths:
|
|
- 'tinygrad/runtime/autogen/**/*'
|
|
- 'tinygrad/runtime/support/autogen.py'
|
|
workflow_dispatch:
|
|
paths:
|
|
- 'tinygrad/runtime/autogen/**/*'
|
|
- 'tinygrad/runtime/support/autogen.py'
|
|
|
|
jobs:
|
|
autogen:
|
|
name: In-tree Autogen
|
|
runs-on: ubuntu-24.04
|
|
timeout-minutes: 15
|
|
steps:
|
|
- name: Checkout Code
|
|
uses: actions/checkout@v4
|
|
- name: Setup Environment
|
|
uses: ./.github/actions/setup-tinygrad
|
|
with:
|
|
opencl: 'true'
|
|
amd: 'true'
|
|
cuda: 'true'
|
|
llvm: 'true'
|
|
webgpu: 'true'
|
|
mesa: 'true'
|
|
pydeps: 'pyyaml mako'
|
|
- name: Install autogen support packages
|
|
run: sudo apt-get install -y --no-install-recommends libclang-20-dev llvm-20-dev hip-dev libusb-1.0-0-dev
|
|
- name: Verify OpenCL autogen
|
|
run: |
|
|
mv tinygrad/runtime/autogen/opencl.py /tmp/opencl.py.bak
|
|
python3 -c "from tinygrad.runtime.autogen import opencl"
|
|
diff /tmp/opencl.py.bak tinygrad/runtime/autogen/opencl.py
|
|
- name: Verify CUDA autogen
|
|
run: |
|
|
mv tinygrad/runtime/autogen/cuda.py /tmp/cuda.py.bak
|
|
mv tinygrad/runtime/autogen/nvrtc.py /tmp/nvrtc.py.bak
|
|
mv tinygrad/runtime/autogen/nvjitlink.py /tmp/nvjitlink.py.bak
|
|
mv tinygrad/runtime/autogen/nv_570.py /tmp/nv_570.py.bak
|
|
mv tinygrad/runtime/autogen/nv.py /tmp/nv.py.bak
|
|
python3 -c "from tinygrad.runtime.autogen import cuda, nvrtc, nvjitlink, nv_570, nv"
|
|
diff /tmp/cuda.py.bak tinygrad/runtime/autogen/cuda.py
|
|
diff /tmp/nvrtc.py.bak tinygrad/runtime/autogen/nvrtc.py
|
|
diff /tmp/nvjitlink.py.bak tinygrad/runtime/autogen/nvjitlink.py
|
|
diff /tmp/nv_570.py.bak tinygrad/runtime/autogen/nv_570.py
|
|
diff /tmp/nv.py.bak tinygrad/runtime/autogen/nv.py
|
|
- name: Verify AMD autogen
|
|
run: |
|
|
mv tinygrad/runtime/autogen/comgr.py /tmp/comgr.py.bak
|
|
mv tinygrad/runtime/autogen/hsa.py /tmp/hsa.py.bak
|
|
mv tinygrad/runtime/autogen/hip.py /tmp/hip.py.bak
|
|
mv tinygrad/runtime/autogen/amd_gpu.py /tmp/amd_gpu.py.bak
|
|
mv tinygrad/runtime/autogen/sqtt.py /tmp/sqtt.py.bak
|
|
mv tinygrad/runtime/autogen/rocprof.py /tmp/rocprof.py.bak
|
|
mv tinygrad/runtime/autogen/am/am.py /tmp/am_am.py.bak
|
|
mv tinygrad/runtime/autogen/am/pm4_soc15.py /tmp/am_pm4_soc15.py.bak
|
|
mv tinygrad/runtime/autogen/am/pm4_nv.py /tmp/am_pm4_nv.py.bak
|
|
mv tinygrad/runtime/autogen/am/sdma_4_0_0.py /tmp/am_sdma_4_0_0.py.bak
|
|
mv tinygrad/runtime/autogen/am/sdma_5_0_0.py /tmp/am_sdma_5_0_0.py.bak
|
|
mv tinygrad/runtime/autogen/am/sdma_6_0_0.py /tmp/am_sdma_6_0_0.py.bak
|
|
mv tinygrad/runtime/autogen/am/smu_v13_0_0.py /tmp/am_smu_v13_0_0.py.bak
|
|
mv tinygrad/runtime/autogen/am/smu_v14_0_2.py /tmp/am_smu_v14_0_2.py.bak
|
|
python3 -c "from tinygrad.runtime.autogen import comgr, hsa, hip, amd_gpu, sqtt, rocprof; from tinygrad.runtime.autogen.am import am, pm4_soc15, pm4_nv, sdma_4_0_0, sdma_5_0_0, sdma_6_0_0, smu_v13_0_0, smu_v14_0_2"
|
|
diff /tmp/comgr.py.bak tinygrad/runtime/autogen/comgr.py
|
|
diff /tmp/hsa.py.bak tinygrad/runtime/autogen/hsa.py
|
|
diff /tmp/hip.py.bak tinygrad/runtime/autogen/hip.py
|
|
diff /tmp/amd_gpu.py.bak tinygrad/runtime/autogen/amd_gpu.py
|
|
diff /tmp/sqtt.py.bak tinygrad/runtime/autogen/sqtt.py
|
|
diff /tmp/rocprof.py.bak tinygrad/runtime/autogen/rocprof.py
|
|
diff /tmp/am_am.py.bak tinygrad/runtime/autogen/am/am.py
|
|
diff /tmp/am_pm4_soc15.py.bak tinygrad/runtime/autogen/am/pm4_soc15.py
|
|
diff /tmp/am_pm4_nv.py.bak tinygrad/runtime/autogen/am/pm4_nv.py
|
|
diff /tmp/am_sdma_4_0_0.py.bak tinygrad/runtime/autogen/am/sdma_4_0_0.py
|
|
diff /tmp/am_sdma_5_0_0.py.bak tinygrad/runtime/autogen/am/sdma_5_0_0.py
|
|
diff /tmp/am_sdma_6_0_0.py.bak tinygrad/runtime/autogen/am/sdma_6_0_0.py
|
|
diff /tmp/am_smu_v13_0_0.py.bak tinygrad/runtime/autogen/am/smu_v13_0_0.py
|
|
diff /tmp/am_smu_v14_0_2.py.bak tinygrad/runtime/autogen/am/smu_v14_0_2.py
|
|
- name: Verify Linux autogen
|
|
run: |
|
|
mv tinygrad/runtime/autogen/libc.py /tmp/libc.py.bak
|
|
mv tinygrad/runtime/autogen/kfd.py /tmp/kfd.py.bak
|
|
mv tinygrad/runtime/autogen/io_uring.py /tmp/io_uring.py.bak
|
|
mv tinygrad/runtime/autogen/ib.py /tmp/ib.py.bak
|
|
mv tinygrad/runtime/autogen/pci.py /tmp/pci.py.bak
|
|
mv tinygrad/runtime/autogen/vfio.py /tmp/vfio.py.bak
|
|
python3 -c "from tinygrad.runtime.autogen import libc, kfd, io_uring, ib, pci, vfio"
|
|
diff /tmp/libc.py.bak tinygrad/runtime/autogen/libc.py
|
|
diff /tmp/kfd.py.bak tinygrad/runtime/autogen/kfd.py
|
|
diff /tmp/io_uring.py.bak tinygrad/runtime/autogen/io_uring.py
|
|
diff /tmp/ib.py.bak tinygrad/runtime/autogen/ib.py
|
|
diff /tmp/pci.py.bak tinygrad/runtime/autogen/pci.py
|
|
diff /tmp/vfio.py.bak tinygrad/runtime/autogen/vfio.py
|
|
- name: Verify LLVM autogen
|
|
run: |
|
|
mv tinygrad/runtime/autogen/llvm.py /tmp/llvm.py.bak
|
|
python3 -c "from tinygrad.runtime.autogen import llvm"
|
|
diff /tmp/llvm.py.bak tinygrad/runtime/autogen/llvm.py
|
|
- name: Verify WebGPU autogen
|
|
run: |
|
|
mv tinygrad/runtime/autogen/webgpu.py /tmp/webgpu.py.bak
|
|
python3 -c "from tinygrad.runtime.autogen import webgpu"
|
|
diff /tmp/webgpu.py.bak tinygrad/runtime/autogen/webgpu.py
|
|
- name: Verify Qualcomm autogen
|
|
run: |
|
|
mv tinygrad/runtime/autogen/kgsl.py /tmp/kgsl.py.bak
|
|
mv tinygrad/runtime/autogen/qcom_dsp.py /tmp/qcom_dsp.py.bak
|
|
python3 -c "from tinygrad.runtime.autogen import kgsl, qcom_dsp"
|
|
diff /tmp/kgsl.py.bak tinygrad/runtime/autogen/kgsl.py
|
|
diff /tmp/qcom_dsp.py.bak tinygrad/runtime/autogen/qcom_dsp.py
|
|
- name: Verify libusb autogen
|
|
run: |
|
|
mv tinygrad/runtime/autogen/libusb.py /tmp/libusb.py.bak
|
|
python3 -c "from tinygrad.runtime.autogen import libusb"
|
|
diff /tmp/libusb.py.bak tinygrad/runtime/autogen/libusb.py
|
|
- name: Verify mesa autogen
|
|
run: |
|
|
mv tinygrad/runtime/autogen/mesa.py /tmp/mesa.py.bak
|
|
python3 -c "from tinygrad.runtime.autogen import mesa"
|
|
diff /tmp/mesa.py.bak tinygrad/runtime/autogen/mesa.py
|
|
- name: Verify libclang autogen
|
|
run: |
|
|
cp tinygrad/runtime/autogen/libclang.py /tmp/libclang.py.bak
|
|
REGEN=1 python3 -c "from tinygrad.runtime.autogen import libclang"
|
|
diff /tmp/libclang.py.bak tinygrad/runtime/autogen/libclang.py
|
|
autogen-mac:
|
|
name: In-tree Autogen (macos)
|
|
runs-on: macos-14
|
|
timeout-minutes: 15
|
|
steps:
|
|
- name: Checkout Code
|
|
uses: actions/checkout@v4
|
|
- name: Setup Environment
|
|
uses: ./.github/actions/setup-tinygrad
|
|
with:
|
|
llvm: 'true'
|
|
- name: Verify macos autogen
|
|
run: |
|
|
mv tinygrad/runtime/autogen/metal.py /tmp/metal.py.bak
|
|
LIBCLANG_PATH=/opt/homebrew/opt/llvm@20/lib/libclang.dylib python3 -c "from tinygrad.runtime.autogen import metal"
|
|
diff /tmp/metal.py.bak tinygrad/runtime/autogen/metal.py
|
|
autogen-comgr-3:
|
|
name: In-tree Autogen (comgr 3)
|
|
runs-on: ubuntu-24.04
|
|
timeout-minutes: 15
|
|
steps:
|
|
- name: Checkout Code
|
|
uses: actions/checkout@v4
|
|
- name: Setup Environment
|
|
uses: ./.github/actions/setup-tinygrad
|
|
- name: Install autogen support packages
|
|
run: |
|
|
wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | gpg --dearmor | sudo tee /etc/apt/keyrings/rocm.gpg > /dev/null
|
|
sudo tee /etc/apt/sources.list.d/rocm.list <<EOF
|
|
deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/6.4 $(lsb_release -cs) main
|
|
EOF
|
|
echo -e 'Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600' | sudo tee /etc/apt/preferences.d/rocm-pin-600
|
|
sudo apt -qq update || true
|
|
sudo apt-get install -y --no-install-recommends libclang-20-dev comgr
|
|
- name: Verify comgr (3) autogen
|
|
run: |
|
|
mv tinygrad/runtime/autogen/comgr_3.py /tmp/comgr_3.py.bak
|
|
python3 -c "from tinygrad.runtime.autogen import comgr_3"
|
|
diff /tmp/comgr_3.py.bak tinygrad/runtime/autogen/comgr_3.py
|