github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-28 00:08:16 -05:00
Code Issues Packages Projects Releases Wiki Activity
10,633 Commits 207 Branches 14 Tags
multi_range
Commit Graph

10633 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
George Hotz
37fa38d272 Revert "switch beautiful_mnist to use new optimizer [pr] (#8231)" (#8233) 
This reverts commit e9ee39df22.
 2024-12-13 19:07:09 -08:00
George Hotz
e9ee39df22 switch beautiful_mnist to use new optimizer [pr] (#8231) 
* switch beautiful_mnist to use new optimizer [pr]

* fix abstractions3 + docs

* fix OptimizerGroup with schedule_step api
 2024-12-13 18:27:16 -08:00
chenyu
e0956c518c move some ifs from merge_dims to reshape [pr] (#8229) 
the third return value is only used in reshape
 2024-12-13 19:56:15 -05:00
George Hotz
e2f87ecf36 start work on new gradient (#7838) 
* start work on new gradient

* more correct

* working tests

* more tests

* work

* add (faliing) gradient test

* add view and reduce gradient

* test_add works, many failing test_ops

* add max and reduce max

* add max and reduce max

* 129 failing

* 108 failed

* better view drawing

* 101 failed

* i got 99 failures

* 94 failures

* it's tons of terrible code, but only 50 tests fail

* only 19 failures

* same 19 but shorter

* minimal doesn't matter

* shorter

* lil simpler

* simpler

* simpler

* simpler

* 13 test failures

* nine tests fail

* all ops tests pass

* add contiguous gradient + fix sched tests

* faster by removing toposort calls

* missed one

* add jax to testing
 2024-12-13 16:45:53 -08:00
chenyu
e371a23c45 more comments and tests to reshape [pr] (#8228) 2024-12-13 18:50:13 -05:00
George Hotz
6d83a96440 retry: use movement ops [pr] (#8225) 
* Revert "Revert "use movement ops [pr] (#8222)" (#8224)"

This reverts commit da19c37f0a.

* fix cast before view
 2024-12-13 15:14:26 -08:00
George Hotz
4679f9fb44 add detach to graph [pr] (#8221) 
* add detach to graph [pr]

* accept failure
 2024-12-13 14:21:32 -08:00
chenyu
62e19649c0 lower test_conv_3x3_256_32_32_256_256 (#8226) 
tiny7 is slow
 2024-12-13 17:15:53 -05:00
George Hotz
da19c37f0a Revert "use movement ops [pr] (#8222)" (#8224) 
This reverts commit 0d26c970ba.
 2024-12-13 14:10:47 -08:00
George Hotz
0d26c970ba use movement ops [pr] (#8222) 
* use movement ops [pr]

* test indexing
 2024-12-13 14:06:01 -08:00
chenyu
eb0e5a14fd reorder and comments to reshape [pr] (#8223) 
something feels wrong... contructing a counter example next
 2024-12-13 17:02:27 -05:00
pkotzbach
c1b79c118f add unit tests for to_dtype (#8217) 
* add unit test for to_dtype

* add unit test for to_dtype

---------

Co-authored-by: pkotzbach <pawkotz@gmail.com>
 2024-12-13 16:21:02 -05:00
George Hotz
8a50868264 touchup function.py [pr] (#8220) 
* touchup function.py [pr]

* remove ALLOWED_READ_IMAGE

* eh, keep it, just change it
 2024-12-13 13:07:00 -08:00
George Hotz
aff112f8ab add new uops to prep for gradient (#8219) 2024-12-13 11:54:26 -08:00
George Hotz
dbe549e462 rename expand to unroll [pr] (#8218) 2024-12-13 11:41:52 -08:00
ignaciosica
0a00187dce add real AMX tests to benchmark (#8216) 
* add real amx to benchmark

* add debug=2 to check tc are triggered
 2024-12-13 14:03:41 -05:00
Ahmed Harmouche
70f6183f34 Remove unnecessary wgsl string rewrite pattern (#8215) 2024-12-13 18:27:29 +01:00
geohotstan
eebb3a1bb9 unique names (#8213) 2024-12-13 12:14:47 -05:00
qazal
1824cbd72c s/lazybufs/tensor_uops [pr] (#8207) 2024-12-13 19:20:02 +08:00
qazal
6d6c34eb1e scheduler local graph_rewrite cleanups [pr] (#8206) 
* scheduler local graph_rewrite cleanups [pr]

* extra merge
 2024-12-13 19:07:09 +08:00
qazal
4a617c84e1 cleanup ctx usage in scheduler upats [pr] (#8205) 2024-12-13 18:01:13 +08:00
qazal
55b8c4e8bf apply_swizzle can apply to any views [pr] (#8204) 2024-12-13 17:58:35 +08:00
Ahmed Harmouche
651f72442c encapsulate the exported webgpu model (#8203) 2024-12-13 10:55:37 +01:00
qazal
5864627abe process replay filter warnings [pr] (#8199) 2024-12-13 17:43:43 +08:00
qazal
c5c0d0277d flatten buffer args, delete dtype [pr] (#8202) 2024-12-13 16:43:47 +08:00
Ahmed Harmouche
5198415bfb No unpack_map in wgsl (#8200) 2024-12-13 08:10:31 +01:00
leopf
fe68dbdb23 GroupOp.Idempotent (#8198) 2024-12-12 20:44:04 -05:00
chenyu
ce41e6572d unit test merge_dim [pr] (#8195) 
looking for better ways to write this. first adding some tests
 2024-12-12 17:55:52 -05:00
chenyu
d47530c0d4 fix device canonicalize for :0 in middle [pr] (#8193) 
replace is wrong because it does not check if `:0` is at the end. use re.sub instead
 2024-12-12 16:32:36 -05:00
chenyu
40a4c603b9 remove more test skip for webgpu [pr] (#8192) 2024-12-12 14:06:35 -05:00
chenyu
d586c7e108 remove had_counter from rand (#8191) 2024-12-12 13:35:39 -05:00
chenyu
2fe98e44cd unneeded isinstance(size, int) in alloc [pr] (#8189) 2024-12-12 13:05:02 -05:00
chenyu
72ff631f8d remove unreachable tensor dtype assert (#8190) 
it would have failed in `to_dtype`. added some tests for it too
 2024-12-12 13:04:49 -05:00
chenyu
2e4c7d4cfb add "tinygrad" to be part of cache_dir [pr] (#8188) 
instead of having sqlite / http download / metal compile to add "tinygrad" separately. also make it non-private since it's used in metal
 2024-12-12 12:09:44 -05:00
Ahmed Harmouche
db76586780 Cast pattern touchup in AMDRenderer [pr] (#8185) 2024-12-12 15:12:14 +01:00
nimlgen
bf7d1fcd2c tiny import fixes in hcq graph (#8184) 2024-12-12 16:30:06 +03:00
Ahmed Harmouche
2f2b1e792c wgsl and ops_webgpu simplifications [pr] (#8182) 
Simplify wgsl and ops_webgpu
 2024-12-12 14:21:58 +01:00
George Hotz
d9a0880d33 delete fuzz uops (not tested) [pr] (#8181) 2024-12-12 01:41:27 -08:00
George Hotz
c77cb57454 remove untested BEAM_COMPARE=1 [pr] (#8180) 2024-12-12 01:35:27 -08:00
Ahmed Harmouche
1b94cc095a Bump back wgpu to latest (#8179) 2024-12-12 09:40:52 +01:00
chenyu
97aaa50f3a remove duplicated UOp in Tensor init types [pr] (#8177) 
and a small comment
 2024-12-11 22:59:35 -05:00
chenyu
d240bdd172 remove upcast_in_mid_reduce_axes [pr] (#8176) 2024-12-11 22:14:28 -05:00
chenyu
64a917b7eb remove LAZYCACHE ContextVar [pr] (#8175) 
also removed from resnet latest script
 2024-12-11 22:02:52 -05:00
chenyu
7047ffd27d tiny gguf_load cleanup [pr] (#8174) 
round_up helper
 2024-12-11 21:32:52 -05:00
George Hotz
151ac5f5a2 remove UPCASTMID [pr] (#8173) 2024-12-11 17:29:01 -08:00
George Hotz
f86e0014b7 delete CAPTURE_BEAM, this should use PR or VIZ infrastructure instead [pr] (#8172) 2024-12-11 16:29:03 -08:00
George Hotz
8a04a3a77a rename LazyBuffer -> UOp [pr] (#8169) 
* rename LazyBuffer -> UOp [pr]

* fix docs
 2024-12-11 16:15:52 -08:00
George Hotz
e0fe867c74 delete beam compare 2 [pr] (#8168) 2024-12-11 16:10:01 -08:00
chenyu
aaa3cc235d unused from __future__ import annotations (#8171) 2024-12-11 19:05:04 -05:00
George Hotz
aae2f4da8d fix process replay [pr] (#8170) 
* empty change [pr]

* store the context in PROCESS_REPLAY_CAPTURE
 2024-12-11 15:58:42 -08:00