github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

torch backend works for ResNet-18 (#9200)

Browse Source 
* torch backend progress, a few more functions

* resnet works

* pillow

* tv
This commit is contained in:
George Hotz
2025-02-22 22:16:23 +08:00
committed by GitHub
parent f92820d30d
commit 97bc723538
4 changed files with 101 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
.github/workflows/test.yml vendored
View File

@@ -153,14 +153,17 @@ jobs:
- name: Setup Environment
uses: ./.github/actions/setup-tinygrad
with:
key: torch-backend
key: torch-backend-pillow-torchvision
deps: testing_minimal
pydeps: "pillow torchvision"
- name: Install ninja
run: |
sudo apt update || true
sudo apt install -y --no-install-recommends ninja-build
- name: Test one op
run: PYTHONPATH=. FORWARD_ONLY=1 TINY_BACKEND=1 python3 test/test_ops.py TestOps.test_add
- name: Test ResNet-18
run: PYTHONPATH=. python3 extra/torch_backend/example.py
- name: Test Ops with TINY_BACKEND (expect failure)
run: PYTHONPATH=. TINY_BACKEND=1 pytest test/test_ops.py || true
- name: Test beautiful_mnist in torch with TINY_BACKEND (expect failure)