sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 6 Activity
11,106 Commits 336 Branches 15 Tags
python_speed
Commit Graph
11106 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
qazalandGitHub 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
qazalandGitHub 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
qazalandGitHub 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
qazalandGitHub 1f8a8721e9 remove test_unaligns_idxs, UOps don't have order like this [pr] (#10587) 2025-06-01 12:16:14 +03:00
iharandGitHub c45936c4fc replaced '.upper()' which is never needed with '.lower()' which were duplicated (#10586) 2025-05-31 20:58:42 -04:00
iharandGitHub 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
chenyuandGitHub 77c7989fa0 remove a MUL rewrite rule for wgsl (#10582) 
tests are fine without it
 2025-05-31 14:05:49 -04:00
Ahmed HarmoucheandGitHub 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
qazalandGitHub 6af4b02374 use plain dict and list in grouper [pr] (#10580) 2025-05-31 13:09:59 +03:00
chenyuandGitHub 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
chenyuandGitHub 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
nimlgenandGitHub 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
qazalandGitHub e0305e54fc remove custom merge_views rewrite rule for buffer ops [pr] (#10574) 2025-05-30 15:27:13 +03:00
qazalandGitHub de9597a8a9 cleanup kernel.py ShapeTracker replacement [pr] (#10573) 2025-05-30 15:06:01 +03:00
qazalandGitHub 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
chenyuandGitHub 116ffc4e92 cstyle strips paren for AND and OR (#10560) 2025-05-30 07:09:05 -04:00
qazalandGitHub bbf05110a2 use kernelize in TestLinearizer.test_indexing_multireduce [pr] (#10571) 2025-05-30 11:27:09 +03:00
qazalandGitHub 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
qazalandGitHub 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
wozeparrotandGitHub 5e3c4a8431 fix: comma testsig (#10568) 2025-05-29 19:00:07 -07:00
Eitan TurokandGitHub 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 HotzandGitHub b3b43a82c4 remove Tensor.no_grad, it's meaningless now [pr] (#10556) 2025-05-28 22:20:02 -07:00
George HotzandGitHub e4e7b5d7e1 continue work on beautiful cifar (#10555) 2025-05-28 21:42:01 -07:00
George HotzandGitHub 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 HotzandGitHub 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 HotzandGitHub 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 LyklesGitHubchenyu
 		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 <[email protected]>
 2025-05-28 16:28:37 -04:00
chenyuandGitHub 74cf5dbd9e mlperf system updates (#10550) 
standardized processor and accelerator names
 2025-05-28 16:15:46 -04:00
George HotzandGitHub 98f3d1c26d enumerate cases of Tensors in the JIT (#10548) 2025-05-28 11:51:27 -07:00
nimlgenandGitHub d1d9e729fd am_smi: mem usage (#10547) 2025-05-28 16:53:31 +03:00
chenyuandGitHub 23e41f523a sdxl also run with cached search (#10546) 2025-05-28 06:51:56 -04:00
chenyuandGitHub fffdc4d31c workflow to run sdxl with search (#10543) 2025-05-27 17:25:41 -04:00
qazalandGitHub d1f0043331 use store_val helper in test_schedule asserts [pr] (#10540) 2025-05-27 21:48:06 +03:00
George HotzandGitHub 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
qazalandGitHub 271110bb5a s/src[0]/buf in lowerer.py [pr] (#10539) 2025-05-27 21:08:54 +03:00
qazalandGitHub f0042629d1 replace arg in merge_view [pr] (#10537) 2025-05-27 21:00:24 +03:00
qazalandGitHub 617ecc1a7b fixup grouper process replay [pr] (#10538) 2025-05-27 20:46:55 +03:00
George HotzandGitHub a07caaca0d handle stride 0 variable reshape (#10536) 2025-05-27 10:00:24 -07:00
George HotzandGitHub 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
qazalandGitHub 142f6ba873 move merge_views to grouper swizzler [pr] (#10531) 2025-05-27 16:33:26 +03:00
qazalandGitHub c03e9c8995 fixup typing for @diskcache [pr] (#10529) 2025-05-27 15:30:49 +03:00
George HotzandGitHub 41e3d07d7f view gradient is tricky (#10528) 
* view gradient is tricky

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

* grads to base
 2025-05-26 19:09:59 -07:00
chenyuandGitHub 76eb130d8c hotfix: BenchEvent MLPERF_RUN is mlperf_run (#10526) 2025-05-26 20:19:37 -04:00
uuuvnandGitHub 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
chenyuandGitHub 51dc7eedb0 correct use AM for resnet run_and_time (#10524) 2025-05-26 15:33:11 -04:00
chenyuandGitHub c1919ad55f use AM for resnet run_and_time (#10523) 2025-05-26 14:50:49 -04:00
geohot e9bb2052cf hotfix: update readme 2025-05-26 10:28:16 -07:00
qazalGitHubgeohot
 		6d07087fe1 remove contiguous from MSELECT 2 (#10522) 
* remove contiguous from MSELECT

* test_shrink_on_shard_axis

---------

Co-authored-by: George Hotz <[email protected]>
 2025-05-26 19:19:01 +03:00
geohotstanandGitHub 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