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

9,343 commits

This branch
This branch
All branches
Author SHA1 Message Date
George Hotz
383c0bf05e notes 2025-07-01 10:36:05 -07:00
George Hotz
fd2c0e2626 uc warp 2025-06-30 20:54:29 -07:00
George Hotz
5907f2d443
 		Merge branch 'master' into warp_fun 2025-06-30 18:18:52 -07:00
George Hotz
5628e2054c hotfix: if no ranges, return None 2025-06-30 18:07:56 -07:00
George Hotz
fc59db0aaa
 		Merge branch 'master' into warp_fun 2025-06-30 15:54:57 -07:00
George Hotz
0597735f28
 		remove TC=3 not porting this (#11045) 2025-06-30 15:12:49 -07:00
George Hotz
cccfe6b422 hotfix: test_no_inf_loop_bottom_up 2025-06-30 14:21:45 -07:00
George Hotz
752c76ceb7
 		tc3 shape expand [pr] (#11043) 
* tc3 shape expand [pr]

* remove unused stuff in lowerer
 2025-06-30 13:38:14 -07:00
George Hotz
539b17fcbf
 		expand local shape so shapes work [pr] (#11042) 2025-06-30 13:03:31 -07:00
nimlgen
9ea7deb515
 		hcq: select_iface shared (#11033) 
* hcq: select_iface shared

* errs

* sorry

* upprt
 2025-06-30 21:12:39 +03:00
qazal
013085da7d
 		viz: only path "/" serves the UI (#11037) 
The dict used to exist for /profiler and main localhost:8000, we don't
need it anymore.
 2025-06-30 19:10:33 +03:00
George Hotz
b829331219
 		infinite loop detect in fixed_point_rewrite [pr] (#11038) 2025-06-30 08:57:29 -07:00
Nino Risteski
bc15e98f5c
 		clean up unused imports in examples and update CI linting (#11024) 
* clean up unused imports in examples

* enable unused import checking in examples

* lint

* ignore F541 and F841 - focus on unused imports only

* clean up

* restore tinygrad.frontend.torch for TINY_BACKEND

* tiny change
 2025-06-30 08:21:27 -07:00
George Hotz
cb531dba42
 		detect infinite loop in graph rewrite [pr] (#11036) 2025-06-30 08:15:13 -07:00
qazal
710d734ce7
 		viz: don't need PICKLE_BUFFER=0 in capture (#11031) 2025-06-30 16:20:04 +03:00
qazal
2ea4737930
 		viz: fix newlines breaking label colors (#11030) 
* viz: fix newlines breaking label colors

* TestViz.test_colored_label

* TestWordWrap
 2025-06-30 13:39:44 +03:00
George Hotz
5911b71404
 		early support for bidirectional pattern matcher (#11027) 
* early support for bidirectional pattern matcher

* expose it and add a test

* no bottom up arg there

* disable flaky test
 2025-06-29 16:54:07 -07:00
George Hotz
ec1d97191d
 		minor cleanup to lowerer [pr] (#11026) 
* minor cleanup to lowerer [pr]

* add that rule to sym
 2025-06-29 11:01:29 -07:00
George Hotz
256d4403c5 what does order change 2025-06-29 09:26:32 -07:00
George Hotz
ad823a5199
 		Merge branch 'master' into warp_fun 2025-06-29 09:13:16 -07:00
Piyush
454bc3393d
 		redundant code (#11014) 2025-06-29 09:06:10 -07:00
George Hotz
0df4355cd8 upcasted warp experiments 2025-06-29 09:04:37 -07:00
qazal
19b11cb778
 		hotfix: check canvas exists before access (#11022) 2025-06-29 14:44:14 +03:00
chenyu
126fcf4129
 		clean up AMD_LLVM in tests (#11021) 2025-06-28 22:45:47 -04:00
qazal
cb6a66ea84
 		viz: remove per schedule renderMemoryGraph (#11019) 
replaced with per device Buffer viz https://github.com/tinygrad/tinygrad/pull/10960
 2025-06-28 22:09:38 +03:00
qazal
4c8d2a0383
 		buffer viz (#10960) 
* add mem_layout

* ui

* cleanup

* work

* debugLine work and expander

* tooltip style

* real expand device

* wheel does one thing

* diff

* shows llama oom

* add y axis

* mypy chill

* work

* unittests for the memory layout
 2025-06-28 21:50:32 +03:00
qazal
e3d024afa0
 		viz: split into scale, shapes, axes last (#11018) 
* viz: split into scale, shapes, axes last

* set zoom on render
 2025-06-28 19:10:58 +03:00
qazal
508bc68078
 		viz: small fixups from memory graph (#11017) 
* don't need div.id

* tooltip z-index
 2025-06-28 16:34:14 +03:00
qazal
fc3e509822
 		viz: new canvas on first render (#11016) 2025-06-28 16:04:51 +03:00
chenyu
c14c9a8eff
 		llama3 grad clip (#11003) 2025-06-27 19:14:12 -04:00
nimlgen
e53673a0b2
 		amd: sdma queue overrun fix (#11012) 
* amd: sdma queue overrun fix

* add ()

* fix

* bug

* this is correct
 2025-06-28 01:42:03 +03:00
chenyu
f2548afeb5
 		bert grad clipping start with const 0 (#11008) 
saved the init kernels
 2025-06-27 18:02:23 -04:00
chenyu
a6485d00c8
 		very tiny generate_dataset (#11013) 
one minute to gen on my mac
 2025-06-27 17:10:45 -04:00
qazal
382fa6a325
 		viz: support axis colors in UOp nodes (#11009) 
* work

* javascript

* optional defaultColor

* fine
 2025-06-27 23:02:55 +03:00
qazal
44257f25e4
 		bump line count to 14600 (#11010) 2025-06-27 22:48:14 +03:00
George Hotz
be53ef4f0a
 		rename DEFINE_ACC -> DEFINE_REG (#11006) 
* rename DEFINE_ACC -> DEFINE_REG

* add CMPEQ to groupops
 2025-06-27 11:09:25 -07:00
George Hotz
05c35d0db8
 		reorder ops and add comments (#11005) 2025-06-27 10:52:14 -07:00
George Hotz
5a1911b7c4
 		apply the global dims late (#11002) 
* apply the global dims late [pr]

* late gpudims

* tests passing

* remove the random local_dims inc

* simpler
 2025-06-27 09:54:34 -07:00
qazal
4ef10c57f9
 		remove unused test helper (#10999) 2025-06-27 13:48:48 +03:00
qazal
a39343e39f
 		viz: move timeline layout to python (#10998) 
* viz: move timeline layout to python

* DevEvent has a device and a name
 2025-06-27 13:06:00 +03:00
George Hotz
b4eb876d5a
 		kernel.py no longer permutes reduce axis [pr] (#10968) 
* kernel.py no longer permutes reduce axis [pr]

* delete tests that handcode uops

* regen of sops is broken...

* put import back

* just remove that

* disable those tests
 2025-06-26 17:44:58 -07:00
chenyu
6ab5a5cb6c
 		llama3 mlperf train (#10983) 
work in progress. now it can overfit small examples and vram roughly matches
 2025-06-26 20:24:27 -04:00
George Hotz
856759c79c
 		add halide example (#10980) 
* add halide example

* upd halide gemm

* partial works

* touchups
 2025-06-26 16:14:57 -07:00
qazal
1127302c46
 		move perfetto to extra (#10994) 
* move perfetto to extra

* update TestViz and fix tests

* remove perfetto.html from viz directory

* work

* mypy
 2025-06-27 01:53:54 +03:00
qazal
712980e167
 		fix extract_dataset + add tests to CI (#10995) 
* fix extract_dataset + tests

* add CI

* sops.gz itself is same as master

* yml + gzip -c + ge

* don't commit that

* bump limit to 1000

* axis=7

* test_tiny
 2025-06-27 01:51:36 +03:00
chenyu
4572e65f0f
 		remove duplicated move_early logic in UOp.r [pr] (#10993) 2025-06-26 18:33:54 -04:00
Ignacio Sica
579194f523
 		remove some linearize calls from tests 2 [pr] (#10992) 
* refactor count_float4 to take uops as input instead of kernel

* remove some calls to linearize in test_linearizer

* remove some more calls

* remove one more call
 2025-06-26 18:22:27 -03:00
geohotstan
50936b4a18
 		ONNX real float16 (#10694) 
* squash commits

* temp fix for const tensor

* actually realizing float16 can only happen in raw_data

* .float -> cast(float) to rerun CI

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
 2025-06-26 14:05:12 -04:00
qazal
73484b0803
 		viz: generic shape tooltip/click handlers + renames (#10990) 
* viz: generic tooltip

* assign kernel

* labelParts/label

* rect with a fillColor

* line
 2025-06-26 19:14:04 +03:00
qazal
7f79c1388f
 		viz: update y offset calculation (#10987) 
* viz: update y offset calculation

* don't rescale padding
 2025-06-26 12:05:20 +03:00