github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-16 01:26:29 -05:00
Code Issues Packages Projects Releases Wiki Activity
8,028 Commits 227 Branches 14 Tags
b32595dbbc3b568b8e1db0ea35d5558b26bbe5f3
Commit Graph

19 Commits

Author SHA1 Message Date
George Hotz
b32595dbbc torch examples (#9290) 
* torch, fix examples/mnist

* fix vae torch example

* where out
 2025-02-28 10:16:06 +08:00
chenyu
184030168d fix aten.reflection_pad2d (#9289) 
tested the torch doc example
 2025-02-27 15:53:46 -05:00
chenyu
0de6585df0 fix aten.normal_ arg (#9288) 
should be mean and std.
 2025-02-27 15:36:25 -05:00
chenyu
8ee2b460ee Tensor.var_mean (#9287) 2025-02-27 15:15:31 -05:00
George Hotz
387ea41e99 increase speed of torch mnist: use gradient api (#9282) 2025-02-27 11:57:41 +08:00
Priyank Patel
a0764f0dc0 (bounty) Make mnist training run with torch backend (#9233) 
* yml changes

* torch backend remove meta decomps and add test

* torch backend bump timeout for tests

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
 2025-02-27 11:32:25 +08:00
George Hotz
9088125a6a a lil more torch (#9280) 2025-02-27 11:12:20 +08:00
George Hotz
b6a14911c8 start torch.compile support (#9279) 2025-02-27 10:29:51 +08:00
Francis Lata
86b737a120 leakyrelu to leaky_relu (#9270) 2025-02-26 13:22:08 -05:00
chenyu
aaf0a8069f xor -> bitwise_xor (#9264) 2025-02-26 10:21:14 -05:00
George Hotz
2158dc4849 full fix for as_strided in torch backend (#9257) 
* fixes from chargpt for torch backend

* shrink support

* add stride support

* comment cleanup

* a few more

* work

* import the stream hack

* llvm multi auto
 2025-02-26 22:34:05 +08:00
George Hotz
7780393460 rig up torch's testing framework [pr] (#9254) 
* rig up torch's testing framework [pr]

* support more movement ops

* dec on expand

* fix tests

* work

* fix tests

* a few more

* decomps + opt hook

* installed pytest
 2025-02-26 18:46:22 +08:00
George Hotz
b603af373e run some tests from torch [pr] (#9252) 
* run some tests from torch [pr]

* yml

* wrap_out

* clean up for the new people

* a lil more
 2025-02-26 15:42:22 +08:00
George Hotz
fc32ff80d6 torch and numpy dtype interop [pr] (#9224) 
* torch and numpy dtype interop [pr]

* less lines

* order
 2025-02-24 18:26:49 +08:00
George Hotz
fd731e740a hotfix: add note on backend2.py 2025-02-24 11:23:03 +08:00
albanD
f2dd9c1562 simplify c++ code (#9221) 2025-02-24 11:04:41 +08:00
George Hotz
97bc723538 torch backend works for ResNet-18 (#9200) 
* torch backend progress, a few more functions

* resnet works

* pillow

* tv
 2025-02-22 22:16:23 +08:00
George Hotz
4e6665bda5 different way to write torch backend (#9197) 
* different way to write torch backend

* both backends

* more work

* simpler code

* more work

* test both

* imply unwrap/wrap

* FORWARD_ONLY=1 TINY_BACKEND=1 python3 test/test_ops.py TestOps.test_add works

* ready to start making test_ops work in torch backend

* backward pass, TINY_BACKEND=1 python3 test/test_ops.py TestOps.test_add works

* FORWARD_ONLY=1 TINY_BACKEND=1 python3 test/test_ops.py TestOps.test_simple_conv2d works

* matmul backward is broken with as_strided
 2025-02-22 14:42:26 +08:00
George Hotz
e87be0131e torch backend start (#9191) 
* start torch backend

* progress

* ugh, you need cpp crap

* 1+1 works

* 1+1 works

* becoming a real backend

* ready to merge?
 2025-02-21 16:57:28 +08:00