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
11,106 commits 280 branches 15 tags 325 MiB
Commit graph

4,667 commits

Author SHA1 Message Date
George Hotz
433581f8ed
 		make POSTOPT=2 the default (#12034) 
* make POSTOPT=2 the default

* more matching tc

* fix winograd

* fix that test

* add matvec to Scheduler

* flip tc sort order

* similar speed

* fix beam on image

* disable slow tests

* slow
 2025-09-05 14:34:05 -07:00
chenyu
3b41a04b96
 		remove test_openpilot in test_onnx (#12037) 
openpilot is tested in compile3
 2025-09-05 16:20:03 -04:00
George Hotz
870f63d9cc
 		add WARP axistype, fix postopt bugs (#12033) 
* postopt is 83% match

* warp is bright CYAN

* beautiful mnist beam works

* fix shutdown bug
 2025-09-05 10:36:55 -07:00
chenyu
4c2d4f683a
 		lower universal_test_unary cos domain (#12032) 
flaky
 2025-09-05 12:19:44 -04:00
chenyu
ce7163e9b4
 		clean up skip slow tests in PYTHON (#12028) 
skip with SKIP_SLOW_TEST and decorators
 2025-09-05 11:35:26 -04:00
chenyu
e0da644171
 		lower sample count in test_multinomial (#12027) 2025-09-05 10:10:28 -04:00
chenyu
9b6f1b86cb
 		add Tensor.maximum in test_dtype_alu (#12025) 
works except nan
 2025-09-05 09:48:39 -04:00
qazal
f461ccf407
 		exclude op2 nan lt in test_dtype_alu (#12024) 
failure: https://github.com/tinygrad/tinygrad/actions/runs/17490320000/job/49679581331?pr=12022#step:6:125
 2025-09-05 14:14:22 +03:00
George Hotz
2b5a73ac65
 		improve test_linearizer (#12016) 
* improve test_linearizer

* tweaks

* simpler

* get_prg

* that one doesn't have to return

* fix postopt bugs

* fix rng
 2025-09-04 20:44:05 -07:00
chenyu
7f3df6ea21
 		exclude nan in test_dtype_alu lt (#12019) 2025-09-04 23:38:37 -04:00
Sieds Lykles
f5404ca53c
 		Divmod combine - associative variations (#12017) 
* add rule and test

* more rules and tests

* add all four variations

* fix test

* test fixed!

* adjust commment

* add new variations

* disable intel tensor core ops count test for bigger_matmul_half
 2025-09-05 03:44:02 +02:00
chenyu
677220ae7e
 		test_tesnor_data to unit/ (#12013) 2025-09-04 19:58:27 -04:00
George Hotz
30eb42a69e
 		fix POSTOPT pad (#11999) 
* fix POSTOPT=1

* fix some tests

* Revert "fix some tests"

This reverts commit 8ee058e206.

* fix padding restrictions

* cuda has two tensor cores

* Set POSTOPT ContextVar to 0 in helpers.py
 2025-09-04 14:28:58 -07:00
qazal
da61b40604
 		some viz tests don't need track_rewrites (#12010) 2025-09-04 23:59:32 +03:00
qazal
be364a1adb
 		viz: add default tracing group (#12009) 
This enables seeing rewrites in unit tests like `VIZ=1 python3 test/test_uop_graph.py TestUOpGraph.test_in_bounds_access_gated_local` that call graph_rewrite directly.

`@track_rewrites` keeps existing as an optional helper to organize larger traces.
 2025-09-04 23:29:56 +03:00
chenyu
52166fd7eb
 		smaller test_ops inputs (#12007) 2025-09-04 16:22:33 -04:00
chenyu
dc8501af30
 		clean up wino tests (#12008) 
removed the one that tests hcopt and added one for backward kernel counts
 2025-09-04 16:14:55 -04:00
chenyu
8c720e8760
 		less iterations for symbolic double for loops (#12006) 2025-09-04 15:09:17 -04:00
George Hotz
70ce29b630
 		test pyrender (#12005) 
* test pyrender

* make them print

* switch to pyrendered
 2025-09-04 11:48:40 -07:00
George Hotz
560df206cc
 		split tc test (#12003) 
* split tc test

* split hand coded opts

* remove some skipped tests

* skips on emulated
 2025-09-04 11:47:56 -07:00
qazal
4996bb668b
 		load all traces before asserting in test_viz (#12004) 2025-09-04 21:34:48 +03:00
George Hotz
9dee724fc4
 		make EMULATE a context var (#12002) 
* make EMULATE a context var

* fix test amx
 2025-09-04 11:15:43 -07:00
George Hotz
09106e4aae
 		refactor and split test_linearizer (#12001) 
* refactor and split test_linearizer

* forget that file

* imports

* remove from docs

* test gen float4
 2025-09-04 10:53:07 -07:00
chenyu
fb71d1e5fd
 		delete some test_search tests (#11998) 
TC_SEARCH_OVER_SHAPE was removed so should the tests
 2025-09-04 11:19:49 -04:00
Sieds Lykles
572a3c15c6
 		Move Ops.SPECIAL arg to src (#11918) 
* initial moving bound to src

* arg to src

* remove import

* fixup linearizer

* arg to src

* fix test_uop_graph

* fix more tests

* fix python renderer

* get const value from const uop

* ssimplify uop estimates

* fix webgpu locals

* fix old test

* gate Ops.SPECIAL in linearizer

* use ssimplify() for local/global_size

* remove toposort gate_parents_instead_of_self

* fix rendering in comment

* cleanup

* rename and add comments

* add BottomUpGate with test
 2025-09-04 09:31:44 +02:00
George Hotz
5cf42dc4db
 		add Scheduler to replace Kernel with POSTOPT=2 (#11924) 
* ** simple kernel to replace Kernel for postopt

* support old

* fix beam

* beaming

* beam on old

* bring tensor cores back

* raise

* postbeam

* test ops passes on mac

* skip that

* postopt default

* gate that

* fix tensor cores

* a few test fixes

* dsp fix

* tc fix

* loop

* support swap

* test_gemv

* fix beam for variable

* test opts from high level stuff

* range annoying

* compile slow

* metal slow

* better beam

* no POSTBEAM

* fix nolocals

* hc opt mostly works

* put that back

* lil

* some work

* fix that

* POSTOPT 2

* fix tests

* no postopt 2

* work

* back

* padded tensors cores

* shift_to

* postopt 0 passes?

* write PADTO

* fix padded tensor cores

* compare hcopt

* 18000 lines

* should pass tests

* fix rangeify

* put types back
 2025-09-03 19:23:30 -07:00
chenyu
b13e071463
 		move test_winograd to unit test (#11993) 2025-09-03 21:47:32 -04:00
chenyu
edc8b99853
 		more tests that pass PTX now (#11992) 2025-09-03 21:18:14 -04:00
chenyu
ed2f45712b
 		remove skip PTX in test_arange (#11991) 
all passes now
 2025-09-03 20:45:19 -04:00
George Hotz
a5f2b4872a
 		use_tensor_cores is a heuristic (#11989) 
* use_tensor_cores is a heuristic

* context
 2025-09-03 17:05:10 -07:00
George Hotz
63e930fec3
 		apply_tensor_cores is a heuristic (#11988) 
* apply_tensor_cores is a heuristic

* delete extra_opts
 2025-09-03 16:39:33 -07:00
chenyu
d0e739453e
 		update many einsum tests (#11981) 
correct the exception testing, and raise ValueError instead of assert when checking args
 2025-09-03 15:40:20 -04:00
b1tg
6d53cac457
 		dtype fuzz: log need input > 0 (#11979) 
Co-authored-by: b1tg <b1tg@users.noreply.github.com>
 2025-09-03 12:10:42 -04:00
Jordan Chalupka
68e83b850f
 		nbytes should raise an exception when size is unlimited (#11928) 
* nbytes should raise an exception when size is unlimited

* adding a test
 2025-09-03 07:06:20 -07:00
Sieds Lykles
86e908db57
 		cast parents of int64 alu to int32 if possible (#11977) 
* add overflows helper

* add rules

* x -> y

* check overflow of u too

* cleaner

* use alu instead of replace to preserve vectorization

* just one rule

* add test
 2025-09-03 11:05:04 +02:00
Sieds Lykles
033184b3cb
 		parse_valid with non const rhs (#11957) 
* const to using vmin/vmax

* add test

* convert to int

* remove left over part of and
 2025-09-03 08:08:46 +02:00
Sieds Lykles
53eff8970a
 		add Ops.GEP to _min_max (#11976) 2025-09-03 07:07:54 +02:00
Sieds Lykles
d1d0960e6e
 		remove intermediate cast using bounds - weaker pattern (#11974) 2025-09-03 06:24:40 +02:00
Sieds Lykles
8a2846b31a
 		assert embedding input is integer dtype (#11963) 
* cast embedding input

* raise error if not using int for index embedding
 2025-09-03 01:44:26 +02:00
George Hotz
1b73993521
 		pyrender to render uops (#11968) 
* pyrender to render uops

* new pyrender style

* pyrender works

* list str

* store render
 2025-09-02 15:44:01 -07:00
chenyu
69dd1817d0
 		raise RuntimeError in merge_dicts instead of assert [pr] (#11965) 2025-09-02 17:18:44 -04:00
qazal
f750c15965
 		viz: add python marker (#11952) 
* viz: add python marker

* remove duplicate
 2025-09-02 23:44:00 +03:00
George Hotz
550cf2ca7f
 		tests from postopt (#11964) 
* tests from postopt

* reraise is fine
 2025-09-02 13:34:17 -07:00
nimlgen
897254ad6c
 		ci: add dev<->cpu copy speeds (#11959) 2025-09-02 15:22:44 +03:00
George Hotz
0dfca4e74b
 		add failing test for rangeify setitem (#11954) 2025-09-01 16:24:35 -07:00
chenyu
6a40216724
 		correct bf16 fuzz input in test_dtype_alu (#11933) 
it was using float16 inputs, now it's uint16 then convert to bf16
 2025-09-01 10:52:26 -04:00
chenyu
965ea59b16
 		test_dtype_alu use AMD_LLVM from helpers (#11950) 2025-09-01 10:03:17 -04:00
b1tg
a9f07c31bc
 		fix amd llvm sqrt (#11936) 
* fix amd llvm sqrt

* lint

---------

Co-authored-by: b1tg <b1tg@users.noreply.github.com>
Co-authored-by: chenyu <chenyu@fastmail.com>
 2025-09-01 09:31:14 -04:00
qazal
0a53e72f70
 		viz: fix trace duration in python test decoder (#11949) 2025-09-01 14:32:25 +03:00
qazal
27c9ed5a84
 		viz: more consistent naming of events (#11948) 
* s/shapes/events in test_viz

* s/bufs/events in the memory packer
 2025-09-01 14:16:47 +03:00