sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 8 Activity
14,284 Commits 336 Branches 15 Tags
8b8c4df66ea03412a676067ae32ca7979f17023a
Commit Graph
11 Commits
Author SHA1 Message Date
chenyuandGitHub 983ad3bd95 fix torch backend batchnorm backward (#17201) 2026-07-25 13:15:09 -04:00
chenyuandGitHub 9f78504304 checked cast in torch backend unwrap (#17199) 2026-07-25 12:34:08 -04:00
chenyuandGitHub 9052db678f remove allow_shape_mismatch in Tensor.replace (#14536) 
move all logic to torch_backend and not hacking Tensor method
 2026-02-04 12:38:18 -05:00
Roelof van DijkGitHubGeorge Hotz
 		0dc2ff431d fix: revive torch backend (#13280) 
* fix: revive torch backend

* as_strided view vs copy

* Revert "as_strided view vs copy"

This reverts commit 82a61223f2.

* add extra tests (move inplace, add fusion tests)

* better fusion with inplace_op

* no optimizer hooks (break mnist training fusion)

* split off fusion tests in separate file, assert on resnet fusion

fix: remove comments

* cleanup, reduce diff

* reduce diff

* better fusion and identity checks

---------

Co-authored-by: George Hotz <[email protected]>
 2025-11-19 15:26:50 -08:00
George HotzandGitHub 32e9949052 rename lazydata to uop (#10698) 2025-06-08 08:42:22 -07:00
Priyank PatelandGitHub e2d9322d21 torch backend: partial fix for strided related test fails (#9642) 
* partial fix for strided related test fails

* cleanup

* fix lint
 2025-03-31 05:45:18 -04:00
Priyank PatelandGitHub 4714c4f9ad torch backend multigpu - add devices and tests (#9414) 
* add multi-device support and tests

* simplify
 2025-03-12 11:33:11 +08:00
George HotzandGitHub 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
albanDandGitHub f2dd9c1562 simplify c++ code (#9221) 2025-02-24 11:04:41 +08:00
George HotzandGitHub 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 HotzandGitHub 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