github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-20 20:38:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

increase speed of torch mnist: use gradient api (#9282)

Browse Source
This commit is contained in:
George Hotz
2025-02-27 11:57:41 +08:00
committed by GitHub
parent a0764f0dc0
commit 387ea41e99
2 changed files with 17 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -170,7 +170,7 @@ jobs:
- name: Test one op in torch tests
run: PYTHONPATH=. DEBUG=2 python3 extra/torch_backend/torch_tests.py TestTinyBackendPRIVATEUSE1.test_unary_log_tiny_float32
- name: Test beautiful_mnist in torch with TINY_BACKEND
run: PYTHONPATH=. TARGET_EVAL_ACC_PCT=96.0 TINY_BACKEND=1 python3 examples/other_mnist/beautiful_mnist_torch.py
run: PYTHONPATH=. LLVM=1 TARGET_EVAL_ACC_PCT=96.0 TINY_BACKEND=1 python3 examples/other_mnist/beautiful_mnist_torch.py
- name: Test Ops with TINY_BACKEND (expect failure)
run: PYTHONPATH=. LLVM=1 LLVMOPT=0 TINY_BACKEND=1 python3 -m pytest -n auto test/test_ops.py || true
- name: Test some torch tests (expect failure)