github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-28 08:17:58 -05:00
Code Issues Packages Projects Releases Wiki Activity
7,293 Commits 209 Branches 14 Tags
8a8eaa1ed9be7d250b33dc628bd7d6e7c5b4f712
Commit Graph

7293 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
chenyu
8a8eaa1ed9 minor change to _reshape_mask [pr] (#8324) 
formatting before logic change
 2024-12-18 16:29:12 -05:00
George Hotz
6608ba316d add size of the buffer to the ptr dtype (#8322) 2024-12-18 12:46:35 -08:00
George Hotz
52243b258c move flops_mem to renderer [pr] (#8320) 2024-12-18 12:13:17 -08:00
chenyu
d2ee304337 minor cleanup to _reshape_mask [pr] (#8321) 
removed usused mask check, and combined if blocks
 2024-12-18 15:09:33 -05:00
chenyu
b4bb8de7f4 remove Sigmoid from function.py [pr] (#8318) 2024-12-18 13:23:38 -05:00
George Hotz
8f95b578f6 use Estimates class [pr] (#8319) 
* use Estimates class [pr]

* frozen dataclass
 2024-12-18 10:19:32 -08:00
chenyu
63f195729d add gguf_load to doc [pr] (#8314) 
Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
 2024-12-18 12:44:09 -05:00
George Hotz
bd9c015b09 tests from grad uop path [pr] (#8313) 2024-12-18 09:25:05 -08:00
George Hotz
6a1987f9f9 hotfix: detach is not a metaop 2024-12-18 09:23:42 -08:00
qazal
fddaeb6344 scheduler deduping spec and asserts [pr] (#8307) 
Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
 2024-12-18 09:21:41 -08:00
leopf
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
qazal
69eb55a529 bugfix: dtype mismatch in TestAssign.test_double_assign_alt [pr] (#8310) 2024-12-18 22:29:07 +08:00
George Hotz
32df46cd73 test const pattern [pr] (#8304) 
* test const pattern [pr]

* add model to test_tiny
 2024-12-17 23:34:17 -08:00
Jyotirmaya Mahanta
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 Hotz
801e199196 change buffer to not be pointer [pr] (#8302) 2024-12-17 16:47:51 -08:00
chenyu
4e2d98638d redundant shape simplify in __unsafe_resize [pr] (#8301) 
also done in View.create.
 2024-12-17 19:00:45 -05:00
chenyu
a9f46ebf70 un1d -> unravel [pr] (#8300) 
numpy/torch has a similar function called `unravel_index`
 2024-12-17 18:08:37 -05:00
chenyu
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
chenyu
513d765c9e remove duplicated canonicalize mask in __unsafe_resize [pr] (#8296) 
also more comments and type annotation
 2024-12-17 16:54:39 -05:00
qazal
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
qazal
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
nimlgen
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
geohotstan
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 <chenyu@fastmail.com>
 2024-12-17 14:12:06 -05:00
George Hotz
21b085b8ed add device UOp [pr] (#8290) 2024-12-17 09:11:51 -08:00
nimlgen
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
George Hotz
0794af97db consts do not realize 2024-12-17 08:53:53 -08:00
George Hotz
4764a4c172 Revert "TIP 3 - Tensor realization spec tests (#8288)" (#8289) 
This reverts commit c0d4346b5a.
 2024-12-17 08:36:11 -08:00
qazal
c0d4346b5a TIP 3 - Tensor realization spec tests (#8288) 2024-12-18 00:04:50 +08:00
qazal
f16188b8c0 viz simplify tracked rewrites [pr] (#8285) 2024-12-17 23:18:08 +08:00
qazal
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
qazal
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
qazal
775746fa08 swizzle / const notes, new asserts [pr] (#8281) 2024-12-17 17:02:38 +08:00
chenyu
df7ba102c3 minor changes to views add [pr] (#8279) 
naming / style / comments before logic change
 2024-12-16 23:53:59 -05:00
George Hotz
e3731766c9 add a test for UOp representation as Tensor [pr] (#8278) 2024-12-16 19:41:29 -08:00
chenyu
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
chenyu
6e2e56c0ff unit test for view add when self has a mask [pr] (#8276) 2024-12-16 20:07:35 -05:00
George Hotz
cda34ccadf hotfix: time.time -> time.perf_counter 2024-12-16 11:32:49 -08:00
chenyu
2bb298f38d add a missing unittest.main() [pr] (#8274) 2024-12-16 14:28:10 -05:00
leopf
befa8641c8 [pr] GroupOp.Associative (#8239) 
* [pr] GroupOp.Associative

* two_stage_associative_folding as lambda
 2024-12-16 08:42:26 -08:00
nimlgen
a2a4ff30dc hcq better timout haandling (#8269) 2024-12-16 13:44:55 +03:00
chenyu
f05fd118a2 few minor code cleanups [pr] (#8267) 2024-12-15 23:44:51 -05:00
chenyu
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
uuuvn
da2245a458 Fix double => half cast on clang (#8265) 2024-12-15 11:24:05 -08:00
qazal
1d21651823 free toposort cache after it goes out of scope [pr] (#8264) 2024-12-15 19:50:42 +02:00
George Hotz
53603c4ec1 simple ignore mop noops [pr] (#8263) 2024-12-15 09:32:42 -08:00
qazal
e1518f1e38 minimal failing test for UOp.toposort gc [pr] (#8261) 2024-12-15 19:30:56 +02:00
qazal
67e66ac1ab hotfix: schedule_uop in process replay (#8260) 
* hotfix: schedule_uop in process replay

* notes
 2024-12-15 21:24:54 +08:00
qazal
58b224a40f schedule_uop api refactor [pr] (#8259) 2024-12-15 19:50:00 +08:00
qazal
ef1346ab39 simplify to_uop more, toward deleting early bufferize (#8254) 2024-12-15 17:41:31 +08:00
qazal
d05e21cb69 replace lazy srcs with the new uop api [pr] (#8255) 
* buf_uop_view function

* srcs shouldn't exist

* fix TestTensorMetadata

---------

Co-authored-by: George Hotz <geohot@gmail.com>
 2024-12-15 17:09:54 +08:00