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

11,106 commits

This branch
This branch
All branches
Author SHA1 Message Date
qazal
00822603ec
 		allow stacking of VIEW UOps [pr] (#10532) 
* allow stacking of VIEW UOps [pr]

* merge_views is first

* simpler

* loc for pr, this needs a helper

* keep

* diff [pr]

* formatting
 2025-06-01 23:27:04 +03:00
qazal
3cc73a0172
 		simpler process replay main loop [pr] (#10588) 
* simpler process replay main loop [pr]

* use logging

* default to 1
 2025-06-01 15:03:21 +03:00
qazal
dc882d3d7d
 		merge process replay and viz captures [pr] (#10581) 
* refactoring

* test script

* work

* more work

* diff

* repr splits lines correctly

* that

* add location

* add location

* also don't need name_override

* k.copy

* [pr]

* name_override 2

* err
 2025-06-01 12:30:10 +03:00
qazal
1f8a8721e9
 		remove test_unaligns_idxs, UOps don't have order like this [pr] (#10587) 2025-06-01 12:16:14 +03:00
ihar
c45936c4fc
 		replaced '.upper()' which is never needed with '.lower()' which were duplicated (#10586) 2025-05-31 20:58:42 -04:00
ihar
88f38d3fcc
 		remove '_metaop' because it is an old wrapper around 'UOp.metaop' with no additional functionality anymore (#10583) 2025-05-31 14:06:39 -04:00
chenyu
77c7989fa0
 		remove a MUL rewrite rule for wgsl (#10582) 
tests are fine without it
 2025-05-31 14:05:49 -04:00
Ahmed Harmouche
35eb4d357a
 		[webgpu] Fix atomic shared mem load inside loop (#10530) 
* Disable shared mem atomics on webgpu

* allow_any_len in load pattern matcher to fix temp load inside loop
 2025-05-31 09:29:02 -04:00
qazal
6af4b02374
 		use plain dict and list in grouper [pr] (#10580) 2025-05-31 13:09:59 +03:00
chenyu
4ab3391e6f
 		set -o pipefail for mlperf run_and_time (#10577) 
also run the 5.1 script in ci cron job
 2025-05-30 16:36:44 -04:00
chenyu
baf482d314
 		copy mlperf stuff to 5.1 (#10576) 
5.0 is finalized, new changes go to 5.1
 2025-05-30 16:12:39 -04:00
nimlgen
883bb4541c
 		am: reserve address space (#10564) 
* am: reserve address space

* f

* cc

* errno

* fix

* always has cpu mapping
 2025-05-30 19:31:03 +03:00
qazal
e0305e54fc
 		remove custom merge_views rewrite rule for buffer ops [pr] (#10574) 2025-05-30 15:27:13 +03:00
qazal
de9597a8a9
 		cleanup kernel.py ShapeTracker replacement [pr] (#10573) 2025-05-30 15:06:01 +03: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
chenyu
116ffc4e92
 		cstyle strips paren for AND and OR (#10560) 2025-05-30 07:09:05 -04:00
qazal
bbf05110a2
 		use kernelize in TestLinearizer.test_indexing_multireduce [pr] (#10571) 2025-05-30 11:27:09 +03:00
qazal
7051bf3fd5
 		fixup hardcoded asts ptr dtype and constants [pr] (#10570) 
* fixup hardcoded asts ptr dtype and constants [pr]

* use kernelize for test_kernel_count
 2025-05-30 09:38:32 +03:00
qazal
066196415f
 		UOp.valid and const_like work with just shapes [pr] (#10569) 
* UOp.valid and const_like work with just shapes [pr]

* pm_quant left

* pm_quant
 2025-05-30 08:55:06 +03:00
wozeparrot
5e3c4a8431
 		fix: comma testsig (#10568) 2025-05-29 19:00:07 -07:00
Eitan Turok
c07f13c438
 		Docs for masked_fill (#10558) 
* add docs

* fix doc examples

* add to docs

* fix typo
 2025-05-29 03:49:02 -07:00
George Hotz
b3b43a82c4
 		remove Tensor.no_grad, it's meaningless now [pr] (#10556) 2025-05-28 22:20:02 -07:00
George Hotz
e4e7b5d7e1
 		continue work on beautiful cifar (#10555) 2025-05-28 21:42:01 -07:00
George Hotz
e140f8f0d8
 		linearizer test_failure_61 (#10552) 
* enumerate cases of Tensors in the JIT

* optional fused optimizers

* add fused optimizer test

* move that there

* ugh

* work on beautiful_cifar

* speed close to hlb_cifar

* test_failure_61

* just the failure
 2025-05-28 21:30:50 -07:00
George Hotz
871df1436a
 		more beautiful cifar (#10551) 
* enumerate cases of Tensors in the JIT

* optional fused optimizers

* add fused optimizer test

* move that there

* ugh

* work on beautiful_cifar

* speed close to hlb_cifar

* schedule to corealize all

* one line sched step

* less lines
 2025-05-28 20:48:20 -07:00
George Hotz
ee12e801a3
 		optional fused optimizers (#10549) 
* enumerate cases of Tensors in the JIT

* optional fused optimizers

* add fused optimizer test

* move that there

* ugh
 2025-05-28 13:50:30 -07: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
chenyu
74cf5dbd9e
 		mlperf system updates (#10550) 
standardized processor and accelerator names
 2025-05-28 16:15:46 -04:00
George Hotz
98f3d1c26d
 		enumerate cases of Tensors in the JIT (#10548) 2025-05-28 11:51:27 -07:00
nimlgen
d1d9e729fd
 		am_smi: mem usage (#10547) 2025-05-28 16:53:31 +03:00
chenyu
23e41f523a
 		sdxl also run with cached search (#10546) 2025-05-28 06:51:56 -04:00
chenyu
fffdc4d31c
 		workflow to run sdxl with search (#10543) 2025-05-27 17:25:41 -04:00
qazal
d1f0043331
 		use store_val helper in test_schedule asserts [pr] (#10540) 2025-05-27 21:48:06 +03:00
George Hotz
5b268121d4
 		remove becomes map (#10533) 
* remove becomes map

* add comment and delete dead code

* multi is a view
 2025-05-27 11:47:11 -07:00
qazal
271110bb5a
 		s/src[0]/buf in lowerer.py [pr] (#10539) 2025-05-27 21:08:54 +03:00
qazal
f0042629d1
 		replace arg in merge_view [pr] (#10537) 2025-05-27 21:00:24 +03:00
qazal
617ecc1a7b
 		fixup grouper process replay [pr] (#10538) 2025-05-27 20:46:55 +03:00
George Hotz
a07caaca0d
 		handle stride 0 variable reshape (#10536) 2025-05-27 10:00:24 -07:00
George Hotz
0515622d95
 		the schedule graph is the tensor graph (#10534) 
* the schedule graph is the tensor graph

* gate type_verify on debug

* relax that spec

* unmasked check is okay
 2025-05-27 09:23:57 -07:00
qazal
142f6ba873
 		move merge_views to grouper swizzler [pr] (#10531) 2025-05-27 16:33:26 +03:00
qazal
c03e9c8995
 		fixup typing for @diskcache [pr] (#10529) 2025-05-27 15:30:49 +03:00
George Hotz
41e3d07d7f
 		view gradient is tricky (#10528) 
* view gradient is tricky

* explicit
 2025-05-26 22:28:30 -07:00
George Hotz
ab4ca5da29
 		remove gradient nonsense [pr] (#10527) 
* remove gradient nonsense [pr]

* grads to base
 2025-05-26 19:09:59 -07:00
chenyu
76eb130d8c
 		hotfix: BenchEvent MLPERF_RUN is mlperf_run (#10526) 2025-05-26 20:19:37 -04:00
uuuvn
c29c46853f
 		Very basic mock sqtt (#10512) 
This mockgpu sqtt emulation will just ignore basically everything and end
up with a 0x1000 size trace full of zeroes, but just testing for things
like register rename is better than nothing i guess
 2025-05-26 14:38:28 -07:00
chenyu
51dc7eedb0
 		correct use AM for resnet run_and_time (#10524) 2025-05-26 15:33:11 -04:00
chenyu
c1919ad55f
 		use AM for resnet run_and_time (#10523) 2025-05-26 14:50:49 -04:00
George Hotz
e9bb2052cf hotfix: update readme 2025-05-26 10:28:16 -07:00
qazal
6d07087fe1
 		remove contiguous from MSELECT 2 (#10522) 
* remove contiguous from MSELECT

* test_shrink_on_shard_axis

---------

Co-authored-by: George Hotz <geohot@gmail.com>
 2025-05-26 19:19:01 +03:00
geohotstan
602a145f8f
 		Add Tensor.unfold (#10518) 
* yoinked 10272

* eitanturok's fixes

* hmmm should size be sint?

* add test
 2025-05-26 11:15:44 -04:00