nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity
9,797 Commits 334 Branches 15 Tags
260da2017c437d34bebfd39d15bd383362ba831a
Commit Graph
732 Commits
Author SHA1 Message Date
geohot 260da2017c fix unit test dtypes 2025-08-13 12:15:12 -07:00
George HotzandGitHub d2521d828a transcendental+idiv+threefry are uop decompositions (#11636) 
* transcendental+idiv+threefry are uop decompositions [pr]

* threefry decomp

* fix randomness tests

* fix webgpu

* unneeded now

* fix

* move prematcher

* all cast should probably be cast_vec
 2025-08-13 09:37:12 -07:00
Sieds LyklesandGitHub 4c3982c44e Take sign out of mod (#11631) 
* Add rule and test

* fix tests
 2025-08-12 18:44:36 +02:00
George HotzandGitHub ca41b5e38b skip_0 in graph rewrite [pr] (#11627) 
* skip_0 in graph rewrite [pr]

* no track_rewrites on test

* use dict instead of set
 2025-08-11 18:29:04 -07:00
George HotzandGitHub 996c907c0b rewrite not ready + children machinery (#11607) 
* rewrite not ready + children machinery

* it doesn't like track rewrites
 2025-08-10 15:28:30 -07:00
qazalandGitHub 960cc6533a pass through name function args in track_rewrites (#11572) 2025-08-08 02:28:52 +03:00
George HotzandGitHub 82be8abfd2 move opt under codegen (#11569) 2025-08-07 14:19:17 -07:00
George HotzandGitHub 6ed2dfd187 delete the arange dim mismatch restriction (#11568) 
* delete the arange dim mismatch restriction

* skip that test race
 2025-08-07 13:46:17 -07:00
George HotzandGitHub 9764c6cdee fix mismatch reduce, try 2 (#11560) 
* fix mismatch reduce, try 2

* fix heuristic

* delete that test

* don't start allowing ones
 2025-08-07 07:57:58 -07:00
George HotzandGitHub a1aa5670aa Revert "fix mismatch reduce (#11547)" (#11549) 
This reverts commit 49d21a9055.
 2025-08-06 22:43:15 -07:00
George HotzandGitHub 49d21a9055 fix mismatch reduce (#11547) 
* fix mismatch reduce

* cleanups

* fix shape

* fix mypy

* resolve
 2025-08-06 21:12:51 -07:00
George HotzandGitHub 21570545d3 move view pushing to codegen, try 2 (#11534) 
* move view pushing to codegen, try 2

* fix up some linearizer tests

* fix test search

* fix test schedule

* delete that test

* fix test arange

* fix a few tests

* update tests

* push views

* ebs cleanup

* fix local/reg

* test and lint

* fix more tests

* test cleanups

* skipped that one
 2025-08-06 15:58:38 -07:00
George HotzandGitHub 80d9cced07 more test cleanups (#11544) 
* more test cleanups

* revert that
 2025-08-06 15:05:21 -07:00
qazalandGitHub 846a2826ab viz: remove TracingKey.fmt (#11482) 
* viz: remove TracingKey.fmt

* remove from test too
 2025-08-05 00:00:03 +03:00
leopfGitHubGeorge Hotz
 		4f0ee4e982 BPE tokenizer (#11415) 
* BPE works

* refactor tok

* oops

* basic tests

* fix eval

* smaller diff

* fix error

* proper vocab decoding

* use regex for splitting

* escape ucatrange

* full compat

---------

Co-authored-by: George Hotz <[email protected]>
 2025-08-04 09:52:38 -07:00
chenyuandGitHub e0106b6b25 1/(x*c) -> (1/c)*(1/x) (#11491) 
example: 2*(2*a).reciprocal() -> a.reciprocal()

# TODO: bounds for reciprocal
# TODO: should z3 work?
 2025-08-03 23:35:46 -04:00
chenyuandGitHub 66be747908 few more dtype cast convinience methods (#11480) 2025-08-02 15:47:09 -04:00
chenyuandGitHub e22e5da9a5 move some test_dtype tests to unit (#11479) 2025-08-02 15:25:00 -04:00
qazalandGitHub fa66d9772d viz: show const node when it's root (#11456) 2025-08-01 01:01:58 +03:00
chenyuandGitHub d5fc6af4a2 remove unused ShapeTracker.consecutive [pr] (#11426) 2025-07-29 18:36:19 -04:00
chenyuandGitHub 88c338bfcc add kernelize to keccak for each data block (#11370) 
* add kernelize to keccak for each data block

test_long works now. this prevents internal uops from growing propotional to data length and eventually too deep

* this?

* hash stuff

* gate test

* mv
 2025-07-25 16:07:20 -04:00
chenyuandGitHub 82e6de7fc6 more keccak reference tests (#11329) 2025-07-23 22:06:39 -04:00
George HotzandGitHub e14b4fefa5 ranges on store (#11334) 
* ranges on store

* fix store spec

* fix that

* fix gates

* fix tests

* fix ptx
 2025-07-22 21:00:50 -07:00
chenyuandGitHub 4535908679 update keccak test_long (#11331) 
it should compare with arg "shake_128"
 2025-07-22 16:08:01 -04:00
qazalandGitHub 6668d6d241 fix word_wrap with newlines in input string [pr] (#11319) 2025-07-22 12:03:13 +03:00
George HotzandGitHub 842184a1ab rename kernelize to schedule, try 2 (#11305) 2025-07-21 11:18:36 -07:00
wozeparrotandGitHub 30ce16a424 feat: failing test for long keccak (#11292) 2025-07-21 12:49:23 -04:00
nimlgenandGitHub 188ed38315 replace from_mv with lightweight mv_address (#11280) 2025-07-19 13:50:51 +03:00
quortusandGitHub 924bc7c9ae Fix test_uop_spec (#11259) 2025-07-16 11:02:31 +03:00
Alisher ZhubanyshevandGitHub 4ef6b46b34 hcq: reduce launch overhead (#11193) 
* nv: improve mmio creation speed

* add memoryview test

* fix indents

* move mv bench to `test_helpers`, remove comparison
 2025-07-13 19:25:50 +03:00
chenyuandGitHub 73caa5dd1b remove Kernel.membufs [pr] (#11200) 2025-07-12 14:48:47 -04:00
qazalandGitHub d3ec63a5c3 viz: add base class for unittests (#11178) 2025-07-11 13:58:03 +03:00
chenyuandGitHub 7db07e5f2c don't narrow range of CAST on bool/unsigned (#11156) 2025-07-09 22:20:09 -04:00
George HotzandGitHub 53ae153404 tc should be in opt (#11148) 
* tc should be in opt [pr]

* fix import
 2025-07-09 14:12:21 -07:00
qazalandGitHub 5c1d215b41 viz: add Graph stream (#11144) 
* viz: stack an event for the entire batch

* multi

* whitespace

* work

* multi graph, Graph gets its own row
 2025-07-09 20:56:46 +03:00
qazalandGitHub 3dfc0ff887 move cpu_profile and shared ProfileEvents from device.py to helpers [pr] (#11126) 
* move cpu_profile and shared ProfileEvents to helpers [pr]

* TestProfiler.test_cpu_profile

* update test_viz.py

* TestProfiler.test_profile_multiops ordering, it's different streams now
 2025-07-08 12:14:03 +03:00
Sieds LyklesGitHubqazal
 		584fd6af5a Fix division by zero and mask bug in add views (#11088) 
* merge view infinite loop test

* adjust condition in `x//d -> x//(-d)*-1`

* Fix division by zero in add views

* adjust offset end

* fix typo in comment

* add target to test_merge_views_variable

* fix view incorrectly being masked

* ssimplify strides and offset of the new view to canonicalize

* remove print in test

---------

Co-authored-by: qazal <[email protected]>
 2025-07-07 10:05:47 -07:00
qazalandGitHub 81781dc12b viz: renames and spacing changes to tracing (#11102) 2025-07-05 18:40:39 +03:00
qazalandGitHub 4fcfaa0ef7 viz: switch to TracingKey (#11100) 
* viz: switch to TracingKey

* tuple

* order is name, keys, fmt

* add test_tracing_key
 2025-07-05 17:46:18 +03:00
qazalandGitHub 3d8569f6d8 hotfix: infinite loop in tracking pattern matcher (#11094) 
* failing test

* fix that

* given matchers
 2025-07-04 19:55:26 +03:00
qazalandGitHub b695e8c4d6 viz: remove support for naming with self (#11076) 2025-07-03 17:29:14 +03:00
Sieds LyklesandGitHub 53985297bd add test, fix rewrite rule and raise error on division by zero (#11073) 2025-07-03 08:25:06 -04:00
qazalandGitHub a919b8325b add test_kernel_info (#11054) 
* add test_kernel_info

* reorder
 2025-07-02 19:48:12 +03:00
chenyuandGitHub 4626e9c172 is_numpy_ndarray helper [pr] (#11050) 2025-07-02 09:12:53 -04:00
qazalandGitHub 8b0871ac31 viz: test for no lockup on infinite loop (#11041) 
* viz: add test infinite loop fallback

* assert

* continue til the end

* work

* bring that back

* fallback to nop
 2025-07-01 17:44:20 +03:00
b1tgGitHubb1tg
 		fcbefde8f5 fix DiskDevice reuse (#11039) 
* fix DiskDevice reuse

* fix mypy and DiskDevice.count

* mypy

* add test

---------

Co-authored-by: b1tg <[email protected]>
 2025-07-01 10:29:21 -04:00
geohot cccfe6b422 hotfix: test_no_inf_loop_bottom_up 2025-06-30 14:21:45 -07:00
George HotzandGitHub b829331219 infinite loop detect in fixed_point_rewrite [pr] (#11038) 2025-06-30 08:57:29 -07:00
George HotzandGitHub cb531dba42 detect infinite loop in graph rewrite [pr] (#11036) 2025-06-30 08:15:13 -07:00
qazalandGitHub 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