sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 8 Activity
7,294 Commits 336 Branches 15 Tags
accc186c8b3dd0d72cc2af642b4bb3d1e14754ca
Commit Graph
7294 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
chenyuandGitHub accc186c8b remove a leading 1 check in _reshape_mask [pr] (#8327) 
the only possible mask for it is either (0, 0) or (0, 1). so the logic is no-op
 2024-12-18 19:30:10 -05:00
chenyuandGitHub 8a8eaa1ed9 minor change to _reshape_mask [pr] (#8324) 
formatting before logic change
 2024-12-18 16:29:12 -05:00
George HotzandGitHub 6608ba316d add size of the buffer to the ptr dtype (#8322) 2024-12-18 12:46:35 -08:00
George HotzandGitHub 52243b258c move flops_mem to renderer [pr] (#8320) 2024-12-18 12:13:17 -08:00
chenyuandGitHub d2ee304337 minor cleanup to _reshape_mask [pr] (#8321) 
removed usused mask check, and combined if blocks
 2024-12-18 15:09:33 -05:00
chenyuandGitHub b4bb8de7f4 remove Sigmoid from function.py [pr] (#8318) 2024-12-18 13:23:38 -05:00
George HotzandGitHub 8f95b578f6 use Estimates class [pr] (#8319) 
* use Estimates class [pr]

* frozen dataclass
 2024-12-18 10:19:32 -08:00
chenyuGitHubGeorge Hotz
 		63f195729d add gguf_load to doc [pr] (#8314) 
Co-authored-by: George Hotz <[email protected]>
 2024-12-18 12:44:09 -05:00
George HotzandGitHub bd9c015b09 tests from grad uop path [pr] (#8313) 2024-12-18 09:25:05 -08:00
geohot 6a1987f9f9 hotfix: detach is not a metaop 2024-12-18 09:23:42 -08:00
qazalGitHubGeorge Hotz
 		fddaeb6344 scheduler deduping spec and asserts [pr] (#8307) 
Co-authored-by: George Hotz <[email protected]>
 2024-12-18 09:21:41 -08:00
leopfandGitHub c5ae66215a [pr] move has_valid into pm_lowerer (#8308) 
* [pr] move has_valid into pm_lowerer

* simpler
 2024-12-18 09:05:18 -08:00
qazalandGitHub 69eb55a529 bugfix: dtype mismatch in TestAssign.test_double_assign_alt [pr] (#8310) 2024-12-18 22:29:07 +08:00
George HotzandGitHub 32df46cd73 test const pattern [pr] (#8304) 
* test const pattern [pr]

* add model to test_tiny
 2024-12-17 23:34:17 -08:00
Jyotirmaya MahantaandGitHub 45f2fb82d5 add failing tests for merge views (#8306) 
* add failing tests for merge views

* assert is not none

* make linter happy
 2024-12-18 01:27:38 -05:00
George HotzandGitHub 801e199196 change buffer to not be pointer [pr] (#8302) 2024-12-17 16:47:51 -08:00
chenyuandGitHub 4e2d98638d redundant shape simplify in __unsafe_resize [pr] (#8301) 
also done in View.create.
 2024-12-17 19:00:45 -05:00
chenyuandGitHub a9f46ebf70 un1d -> unravel [pr] (#8300) 
numpy/torch has a similar function called `unravel_index`
 2024-12-17 18:08:37 -05:00
chenyuandGitHub 66b92b6061 unused case where View adds invalid masked view [pr] (#8299) 
code path does not hit this, and we might want to assert if it ever hits
 2024-12-17 17:31:01 -05:00
chenyuandGitHub 513d765c9e remove duplicated canonicalize mask in __unsafe_resize [pr] (#8296) 
also more comments and type annotation
 2024-12-17 16:54:39 -05:00
qazalandGitHub 673a76398a fix viz with untracked graph_rewrite calls (#8298) 
* fix viz with untracked graph_rewrite calls

* mark as green
 2024-12-17 23:37:53 +02:00
qazalandGitHub 5977a3d8a6 regression test viz failure when there's no tracked context (#8297) 
* regression test viz failure when there's no tracked context

* test inner rewrite locations, keep notes
 2024-12-17 23:23:27 +02:00
nimlgenandGitHub 777d2aec05 metal profiler + cpu_profile (#8291) 
* metal + cpu_profile

* gpt example

* linter + revert gpt2 for now

* a bit of readme

* linter

* unrelated

* tests

* linter

* b
 2024-12-18 00:06:56 +03:00
geohotstanGitHubchenyu
 		32c995a5da move to_python_const from onnx_ops to onnx (#8158) 
* move to_python_const out

* move more over

* try deleting alternative gather implementation

* Revert "try deleting alternative gather implementation"

This reverts commit d46b30b717.

* add types to onnx ops

* better debug msg

* improve some com.microsoft too

---------

Co-authored-by: chenyu <[email protected]>
 2024-12-17 14:12:06 -05:00
George HotzandGitHub 21b085b8ed add device UOp [pr] (#8290) 2024-12-17 09:11:51 -08:00
nimlgenandGitHub af87e4b53c viz profiler (#8287) 
* only hcq

* fix get_metadata

* linter

* oops

* tiny

* linter

* time

* print pm

* hmm

* nits
 2024-12-17 20:00:53 +03:00
geohot 0794af97db consts do not realize 2024-12-17 08:53:53 -08:00
George HotzandGitHub 4764a4c172 Revert "TIP 3 - Tensor realization spec tests (#8288)" (#8289) 
This reverts commit c0d4346b5a.
 2024-12-17 08:36:11 -08:00
qazalandGitHub c0d4346b5a TIP 3 - Tensor realization spec tests (#8288) 2024-12-18 00:04:50 +08:00
qazalandGitHub f16188b8c0 viz simplify tracked rewrites [pr] (#8285) 2024-12-17 23:18:08 +08:00
qazalandGitHub fd23738d9d start viz better typing, unsupport bottom_up=True [pr] (#8284) 
* start viz refactor

* delete bottom_up tracking

* more cleanup

* early continue
 2024-12-17 19:52:30 +08:00
qazalandGitHub 856c068172 make viz work with uop mutability (#8283) 
* make viz work with uop mutability

* keep pickle_buffers local
 2024-12-17 19:17:41 +08:00
qazalandGitHub 775746fa08 swizzle / const notes, new asserts [pr] (#8281) 2024-12-17 17:02:38 +08:00
chenyuandGitHub df7ba102c3 minor changes to views add [pr] (#8279) 
naming / style / comments before logic change
 2024-12-16 23:53:59 -05:00
George HotzandGitHub e3731766c9 add a test for UOp representation as Tensor [pr] (#8278) 2024-12-16 19:41:29 -08:00
chenyuandGitHub 3195bd0d12 more test examples to merge views [pr] (#8277) 
these have masks in self and masks in the merged views
 2024-12-16 20:44:35 -05:00
chenyuandGitHub 6e2e56c0ff unit test for view add when self has a mask [pr] (#8276) 2024-12-16 20:07:35 -05:00
geohot cda34ccadf hotfix: time.time -> time.perf_counter 2024-12-16 11:32:49 -08:00
chenyuandGitHub 2bb298f38d add a missing unittest.main() [pr] (#8274) 2024-12-16 14:28:10 -05:00
leopfandGitHub befa8641c8 [pr] GroupOp.Associative (#8239) 
* [pr] GroupOp.Associative

* two_stage_associative_folding as lambda
 2024-12-16 08:42:26 -08:00
nimlgenandGitHub a2a4ff30dc hcq better timout haandling (#8269) 2024-12-16 13:44:55 +03:00
chenyuandGitHub f05fd118a2 few minor code cleanups [pr] (#8267) 2024-12-15 23:44:51 -05:00
chenyuandGitHub 9789a83064 hotfix DEBUG in speed_v_theoretical.py conv (#8266) 
infinite loop with manual DEBUG set `DEBUG=2 python test/external/speed_v_theoretical.py -k conv`

```
  File "/Users/chenyu/code/tinygrad/tinygrad/helpers.py", line 95, in __ge__
    def __ge__(self, x): return self.value >= x
                                ^^^^^^^^^^^^^^^
  [Previous line repeated 4984 more times]
RecursionError: maximum recursion depth exceeded in comparison
```
 2024-12-15 19:44:45 -05:00
uuuvnandGitHub da2245a458 Fix double => half cast on clang (#8265) 2024-12-15 11:24:05 -08:00
qazalandGitHub 1d21651823 free toposort cache after it goes out of scope [pr] (#8264) 2024-12-15 19:50:42 +02:00
George HotzandGitHub 53603c4ec1 simple ignore mop noops [pr] (#8263) 2024-12-15 09:32:42 -08:00
qazalandGitHub e1518f1e38 minimal failing test for UOp.toposort gc [pr] (#8261) 2024-12-15 19:30:56 +02:00
qazalandGitHub 67e66ac1ab hotfix: schedule_uop in process replay (#8260) 
* hotfix: schedule_uop in process replay

* notes
 2024-12-15 21:24:54 +08:00
qazalandGitHub 58b224a40f schedule_uop api refactor [pr] (#8259) 2024-12-15 19:50:00 +08:00
qazalandGitHub ef1346ab39 simplify to_uop more, toward deleting early bufferize (#8254) 2024-12-15 17:41:31 +08:00