mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity
9,139 commits 280 branches 15 tags 325 MiB
Commit graph

651 commits

Author SHA1 Message Date
George Hotz
a38947b4bb
 		move symbolic and transcendental to uop [pr] (#10771) 2025-06-10 20:51:22 -07:00
chenyu
14fa62c61d
 		move high level tests to unit (#10760) 
either no need a backend, or running on one to check suffice
 2025-06-10 12:55:44 -04:00
qazal
5d9c274924
 		keep UOp tags if sources are replaced (#10754) 
* keep UOp tags in unified_rewrite

* add failing test, print tag if defined

* remove the repr change
 2025-06-10 08:30:14 -07:00
George Hotz
acf72872b3
 		move view left to the outer graph prereqs + testing (#10725) 
* move view left to the outer graph

* global view right

* dont need that one

* remove comment

* test kernelize

* simple

* split onnx, test sdxl null

* fix testing

* ugh, wrong one

* Update test.yml
 2025-06-09 20:43:25 -07:00
George Hotz
81ef879da3
 		non recursive top_down_rewrite (#10729) 
* non recursive top_down_rewrite

* nicer algorithm

* rewrite bottom up also

* only top down is broken?

* simpler iterative algo

* no recursion errors

* top down and bottom up

* unified rewrite

* simpler rewrite

* clean up comments

* move that comment
 2025-06-09 16:33:04 -07:00
George Hotz
81b9c04574
 		move high level stuff to unit tests [pr] (#10708) 
* move high level stuff to unit tests [pr]

* process replay on unit tests

* fix pr, less compute

* set omp num threads

* set 200MB buffer size limit

* delete junk

* fix tests

* faster

* move test_indexing to unit

* faster
 2025-06-08 14:05:56 -07:00
George Hotz
4e2c3560b4
 		smaller tests are faster tests [pr] (#10704) 
* remove del spam from CI

* more

* preconstruct default buffer spec

* ignore those errors

* check exception

* more exception check

* skip stuff

* smaller tests mean faster tests

* a few more
 2025-06-08 10:54:19 -07:00
George Hotz
32e9949052
 		rename lazydata to uop (#10698) 2025-06-08 08:42:22 -07:00
ihar
40c1479267
 		added unit tests for 'argfix' (#10678) 2025-06-07 22:17:10 -04:00
George Hotz
53ed64e133
 		ci speed work 1 (#10676) 
* skip a few slow tests

* use a venv for python packages

* create venv

* no user, it's in venv

* ignore venv

* venv

* new cache key

* try that

* this

* version the python cache
 2025-06-07 16:33:11 -07:00
qazal
cb61774ab6
 		move shared viz fields out of serve.py [pr] (#10684) 
* move shared viz fields out [pr]

* update javascript

* update test_viz
 2025-06-07 17:18:18 +03:00
qazal
b515d796fb
 		inline viz get_name [pr] (#10682) 
* inline viz get_name [pr]

* changing name_fxn makes this simpler

* waitUntil dom
 2025-06-07 11:16:16 +03:00
chenyu
bdede4924e
 		fix odd number in get_test_global_size (#10671) 
factor might not be a integer if input global_size has an odd number in it
 2025-06-06 17:31:35 -04:00
George Hotz
baba274a76
 		minimal mstack pr to fix allreduce (#10649) 
* minimal mstack pr to fix allreduce

* fix webgpu
 2025-06-05 15:14:53 -07:00
qazal
5b59728c75
 		refactor LOAD(DEFINE_GLOBAL, VIEW) in kernels to LOAD(VIEW(DEFINE_GLOBAL)) (#10541) 
* changes to core tinygrad

* fixups pt1

TC=3
docs/abstractions2.py
IMAGE=2
test_quantize_dsp
test_schedule

* more tests

* green now

* images stay images
 2025-05-30 14:27:58 +03:00
Sieds Lykles
ae02a1e232
 		[bounty] Z3 symbolic fuzzer [pr] (#10514) 
* First version, caught a bug?

* Nicely print failure to reproduce

* Remove that

* Put the assert back

* Change fuzzing to use testing_unit so it has z3

* Test key to match

* Add rule

* Add test

* Add test for edge case 0

* Merge patterns

* update comment

* consistent whitespace

* whitespace

* add condition

* add test

* update comment

* use Variable

* fuzzer using z3_renderer

* Cleaned up printing and debugging

* working new fuzzer

* change some comments and printing

* more formatting

* fuzz failures in seperate file

* fix fstring

* more tests

* naming

* remove added line

* remove comment

* print number of skipped expressions

* use self.assertEqual

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
 2025-05-28 16:28:37 -04:00
George Hotz
a07caaca0d
 		handle stride 0 variable reshape (#10536) 2025-05-27 10:00:24 -07:00
George Hotz
41e3d07d7f
 		view gradient is tricky (#10528) 
* view gradient is tricky

* explicit
 2025-05-26 22:28:30 -07:00
Sieds Lykles
478c76f4b7
 		More div conditions (#10432) 
* add condition

* add test

* use Variable
 2025-05-26 07:36:05 -04:00
Sieds Lykles
c6c7882bdf
 		bugfix: seperate rule for x//d<-c (#10148) 
* Add rule

* Add test

* Add test for edge case 0

* Merge patterns

* update comment

* consistent whitespace

* whitespace

* update comment
 2025-05-26 07:35:41 -04:00
wozeparrot
a18963d9e7
 		feat: use tinygrad useragent (#10488) 2025-05-23 15:44:40 -07:00
qazal
7a762f01ab
 		s/shape_spec/ast_spec [pr] (#10485) 2025-05-23 15:43:54 +03:00
qazal
127a7c8aee
 		assert AST views only exist in the edges (#10484) 
* assert AST views only exist in the edges

* valid without device
 2025-05-23 15:27:09 +03:00
Sieds Lykles
ce6ebfb8ee
 		verify rewrites in test_uop_symbolic (#10430) 
* verify rewrites in test_uop_symbolic

* use global context
 2025-05-23 06:57:29 -04:00
chenyu
8cc2dff4d8
 		only float Tensors have gradient [pr] (#10475) 2025-05-22 21:02:11 -04:00
chenyu
7bfb20757c
 		fix tensor int floor div (#10327) 
* fix tensor int floor div

* test_float_floordiv_scalar
 2025-05-21 06:46:54 -04:00
Sieds Lykles
2b4375f36d
 		Correct divmod folding behind flag (#10433) 
* add flag

* add test

* remove import
 2025-05-21 06:46:13 -04:00
Sieds Lykles
db09676250
 		Dont simplify gate in gate, fix FUSE_ARANGE=1 python test/test_ops.py TestOps.test_scatter_add (#10411) 
* substitute out index

* Add test

* change comment
 2025-05-19 13:16:21 -04:00
George Hotz
411392dfb7
 		move files into uop dir (#10399) 
* move files into uop dir [pr]

* tinygrad.uop is a thing

* fix uop docs, no pr

* fix viz
 2025-05-18 11:38:28 -07:00
George Hotz
6f77b938d7
 		Move getbits tests into test_helpers (#10382) 2025-05-17 17:04:00 -07:00
George Hotz
e13f2a3092
 		multi is O(1) (#10183) 
* multi is O(1)

* allreduce

* no new uops needed

* junk

* something

* simple

* that's really what i want

* closer

* inject _device_num

* pretty print

* cleanups

* this

* early dnum

* ops allreduce is good

* ish

* device is the tuple and this is fine

* simpler

* progress

* copy_multi

* work

* more tests

* more tests pass

* work

* no None axis

* tests

* no none multi

* type fixes

* pre commit passes

* lil

* remove this

* mlperf dataloader on mac

* that test was wrong

* unbind

* support DEBUG=2

* realize

* only unbind bound vars

* don't include fixedvars

* graph test

* one test

* fixedvars in hcq

* new ring reduce

* ring reduce

* simpler ring

* mselect

* mselect doesn't work

* Revert "mselect doesn't work"

This reverts commit c78b77bd7d.

* Revert "mselect"

This reverts commit bb2e430ac3.

* simpler

* fixups

* no optional

* fix jit

* move things around

* cleanup multi

* simpler multi

* simpler reshape
 2025-05-16 23:14:23 -07:00
George Hotz
876d2275a1
 		changes from new multi (#10353) 
* changes from new multi

* revert hcq change
 2025-05-16 13:07:29 -07:00
qazal
043efc6ec4
 		do not require self for track_rewrites [pr] (#10302) 2025-05-14 18:23:32 +03:00
Sieds Lykles
02208565de
 		add check (#10257) 2025-05-12 11:03:01 -04:00
Kirill R.
4c7c139102
 		Use cmod/cdiv in sym_infer (#10258) 
* Use cmod/cdiv in sym_infer

* test

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
 2025-05-12 09:07:28 -04:00
Sieds Lykles
7c4b381fbf
 		Extra simplify valid test [pr] (#10256) 
* add test

* Change the range

* add todo test
 2025-05-12 07:32:03 -04:00
Sieds Lykles
74e40aafa0
 		use cdiv in div and mod folding (#10216) 
* use cdiv

* use cdiv and cmod there as well

* Add tests

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
 2025-05-09 12:37:24 -04:00
Sieds Lykles
8da9c070ca
 		take gcd out of trunc div (#10238) 2025-05-09 12:08:10 -04:00
chenyu
9846435c2e
 		fix test_div_numerator_negative (#10229) 
the simplification was wrong with negative const_factor
 2025-05-09 06:19:59 -04:00
chenyu
cba508c8c3
 		update uop symbolic tests (#10228) 
clean up TODOs and update tests
 2025-05-09 01:55:53 -04:00
chenyu
56def6c319
 		better bound for mod negative number (#10227) 2025-05-09 01:19:47 -04:00
chenyu
99f6d89dfb
 		tighter idiv bound for symbolic denominator (#10226) 2025-05-08 22:38:56 -04:00
Sieds Lykles
2891892834
 		Fold constant variable (#10196) 
* Add rule

* add test and comment

* merge rule
 2025-05-07 11:39:44 -07:00
Sieds Lykles
8386527bb9
 		Take neg out of idiv (#10164) 
* Add rules

* Fix tests

* Move rules lower to prevent recursion
 2025-05-07 11:39:08 -07:00
Sieds Lykles
09544d4556
 		Add rule and test (#10189) 
Co-authored-by: chenyu <chenyu@fastmail.com>
 2025-05-07 10:15:55 -04:00
George Hotz
603c03bef2
 		fix tests for rewrite [pr] (#10167) 
* fix tests for rewrite [pr]

* cleaner

* delete linearize_uop

* clean up the rest
 2025-05-05 19:19:49 -07:00
Sieds Lykles
848c7783a4
 		Sign check in div const div pattern (#10150) 
* Add rule

* Relax the condition

* Add test

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
 2025-05-03 18:04:34 -04:00
George Hotz
ef011ff5f9
 		flip Ops.COPY order [pr] (#10122) 
* flip Ops.COPY order [pr]

* fix copy and support multi device copy in _device
 2025-05-01 00:26:24 -04:00
George Hotz
dd0070daab
 		Revert "flip Ops.COPY order [pr] (#10120)" (#10121) 
This reverts commit 984f09ac74.
 2025-04-30 17:25:21 -04:00
George Hotz
984f09ac74
 		flip Ops.COPY order [pr] (#10120) 2025-04-30 16:50:18 -04:00