github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-17 02:48:03 -05:00
Code Issues Packages Projects Releases Wiki Activity
10,420 Commits 204 Branches 14 Tags
split_sub
Commit Graph

403 Commits

Author SHA1 Message Date
George Hotz
061a17ee4d split pm_substitute_recurse 2025-10-06 09:22:01 +08:00
qazal
6ad9a688ed add failing test after "pend substitutes for speed" (#12457) 
* add failing substitute test

* expect_rangeify_fails
 2025-10-05 16:10:04 +03:00
qazal
13a25b2e67 rangeify: don't shape INDEX on kernelize (#12417) 2025-10-02 09:45:37 +03:00
qazal
6fc6b51b59 fix limit_bufs with kernelize (#12415) 2025-10-02 07:49:11 +03:00
George Hotz
89bed28716 split reduceop (#12404) 
* some rangeify tests fixed

* bring split reduceop to rangeify

* fix tests
 2025-10-01 18:45:16 +08:00
qazal
90b1c0dd96 rangeify: test_where_fold kernel count (#12379) 
* rangeify: test_where_fold kernel count

* get these from the index

* replace ranges

* fine

* movement ops

* diff

* better
 2025-10-01 09:35:12 +03:00
nimlgen
2c397eb2a2 rangeify: buf limit (#12336) 
* limit bufs

* g

* fix buffer limit

* um?

* fix

* only these?

* typo

* f

* cleaner
 2025-09-30 14:59:47 +03:00
qazal
4ff7f20b9d rangeify: fix kernelize (#12357) 2025-09-30 10:10:08 +03:00
George Hotz
ab6b0d3a21 enable cleanup_dead_axes (#12351) 
* enable cleanup_dead_axes

* don't mess with user contig

* correct tag behavior

* double reshape isn't correct

* block on assign too

* skip messing with symbolic

* Fix tests

* disable RANGEIFY=2

* test w rangeify
 2025-09-30 14:09:39 +08:00
wozeparrot
2a0caa09c2 push copy to disk (#12348) 2025-09-29 21:55:05 -07:00
qazal
250cb10e8f rangeify permuted assign (#12299) 
* enable RANGEIFY=1 test_assign

* work

* rangeify=0 asserts this ast

* remove that

* beta test, it's correct though

* skip multi

* matches torch/np output

* memcopy without memcopy

* can remove this

* rangeify isn't silently wrong anymore

* diff cleanup

* use UOp toposort instead of global tags

* actual assert TestRangeifyAssign

* step

* work

* this isn't optimizing away now

* some todos

* test fusion schedule

* typo

* dedup idxs

* cleaner

* pre

* work

* diff
 2025-09-29 07:27:57 +03:00
Sieds Lykles
ed90de6583 Revert "Bufferize early, fix "children not making progress" on big graphs (#1…" (#12318) 
This reverts commit 6f1cf717de.
 2025-09-28 19:10:21 +02:00
Sieds Lykles
6f1cf717de Bufferize early, fix "children not making progress" on big graphs (#12308) 
* bufferize children early

* cleaner

* fix types

* lower number of reduceops

* test openpilot
 2025-09-27 04:17:15 +02:00
nimlgen
f5eb46a3d9 fix limit buf metal on non rangeify (#12303) 
* add failure test for limit buf on non rangeify

* correct metal

* correct

* hm
 2025-09-26 11:06:28 +03:00
qazal
6c9d8c7e41 rangeify: simplify noop copy (#12289) 2025-09-24 17:01:23 +03:00
nimlgen
02a7b7fe48 rangeify: fix test_setitem (#12269) 
* rangeify: fix test_setitem

* um?

* better?

* simple where folding

* f

* revert

* x
 2025-09-23 20:42:36 +03:00
nimlgen
b53a266254 rangeify: fix test_optim (#12262) 
* rangeify: fix test_optim

* add to cl?

* these are good now
 2025-09-21 18:08:35 +03:00
qazal
bb59eed82f rangeify: don't tag consts, they are global (#12247) 
* rangeify: don't tag consts, they are global

* don't map movement ops

* sym failing test

* remove that

* update comment

* simpler test

* work
 2025-09-19 15:25:03 +03:00
qazal
825f148469 rangeify: fix copy size mismatch errs (#12232) 
* rangeify: fix copy size mismatch errs

* const folding can happen in sym

assert it

* shippable

* rangeify copy is completely wrong

* pre_bufferize

* tag bufferize

* pre back
 2025-09-18 18:23:32 +03:00
qazal
dbbc261075 rangeify: fix COPY simplifier (#12233) 2025-09-18 14:35:33 +03:00
qazal
525f80e0d2 rangeify: enable putting consts back in the tensor graph (#12225) 
* rangeify: enable putting consts back in the tensor graph

* work

* sym in ci
 2025-09-17 19:45:04 +03:00
qazal
7733c217c5 remove spam comments in test_schedule (#12224) 2025-09-17 18:24:55 +03:00
qazal
d917895569 map out rangeify errors in test_schedule (#12211) 
* map out rangeify errors in test_schedule

* skip that

* add to ci
 2025-09-17 09:10:28 +03:00
qazal
122a50fe8c assert kernel count (#12205) 2025-09-16 14:24:39 +03:00
qazal
02054b53fe remove tests that pre date the uop spec (#12168) 
* remove tests that pre date the uop spec

* const src

* for RANGEIFY=1

* update with bind

* remove import
 2025-09-14 18:47:42 +03:00
George Hotz
bcafa72b7f use tags instead of graph_rewrite_map in rangeify (#12110) 
* use tags instead of graph_rewrite_map in rangeify

* new style, add realize

* metadata works

* simple failure

* fix

* loops

* stuff becomes a NOOP when you remove it

* stuff becomes a NOOP when you remove it

* tags on bufferize

* bmnist works

* locals don't work

* shippable

* fix some tests

* simpler map_realize

* remove const hack

* debuggable test

* broke

* assign test

* straight up bug

* wooo it passes

* sink shouldn't be there

* fix ops

* bmnist

* kv cache ish

* Set RANGEIFY context variable to 0

* should work normal

* better

* types

* hacks to fix test_symbolic

* pm_add_buffers

* tests should pass
 2025-09-14 11:39:01 +08:00
qazal
e80c8a7548 merge TestIndexing with TestSchedule + remove duplicate tests (#12134) 
* merge TestIndexing with TestSchedule

* remove the arange_copy tests

* no FUSE_ARANGE import
 2025-09-12 10:35:14 +03:00
George Hotz
0766616962 isolate the const hacks in the old kernelize (#12126) 
* isolate the const hacks in the old kernelize

* if rangeify, don't waste time
 2025-09-12 08:35:35 +08:00
chenyu
0e266f376c ops_gpu -> ops_cl (#12103) 2025-09-10 15:15:48 -04:00
chenyu
beb5982165 FUSE_ATTENTION (#11884) 2025-08-27 19:59:17 -04:00
chenyu
a67e0917c3 list indexing can normalize in python (#11609) 
* list indexing can normalize in python

list index does not need to be normalized in tensor

* update those
 2025-08-10 20:02:38 -04:00
George Hotz
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
chenyu
7ee3770961 FUSE_ARANGE=1 (#11427) 
* FUSE_ARANGE=1

* fix test

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
 2025-08-07 13:32:34 -04:00
George Hotz
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 Hotz
a1aa5670aa Revert "fix mismatch reduce (#11547)" (#11549) 
This reverts commit 49d21a9055.
 2025-08-06 22:43:15 -07:00
George Hotz
49d21a9055 fix mismatch reduce (#11547) 
* fix mismatch reduce

* cleanups

* fix shape

* fix mypy

* resolve
 2025-08-06 21:12:51 -07:00
George Hotz
6fd1332763 update some tests for less Kernel (#11543) 
* update some tests for less Kernel

* get_program update
 2025-08-06 14:19:59 -07:00
George Hotz
842184a1ab rename kernelize to schedule, try 2 (#11305) 2025-07-21 11:18:36 -07:00
chenyu
54924f9969 type remove Union and Optional [pr] (#11283) 
use `|` for consistency
 2025-07-19 14:05:52 -04:00
George Hotz
8743ca40e2 force reduce to be in axis order (#10837) 
* force reduce to be in axis order

* disable rule causing loop

* disable that rule

* no ra there

* only move non reduce

* fix tests
 2025-06-24 13:00:16 -07:00
qazal
ac39f27ae6 viz: non blocking UOp tracing (#10913) 
* viz: non blocking UOp tracing

* u.arg

* no if Ops.KENREL

* drop replace

* switch to weakref.WeakKeyDictionary

* back

* remove ram usage skips, viz works here

* cache on reconstruct
 2025-06-23 19:59:28 +03:00
George Hotz
b41e0563a3 move stuff to kernelize folder (#10902) 
* move stuff to kernelize folder

* oops, forgot that
 2025-06-20 16:10:20 -07:00
qazal
5839542fc8 viz: one name arg in track_rewrites [pr] (#10873) 
* viz: one name arg in track_rewrites [pr]

* other test
 2025-06-19 03:34:56 +03:00
qazal
96509daaba enable copy folding tests [pr] (#10862) 2025-06-18 13:05:35 +03:00
qazal
84d568d0cc do not import grouper internals in test_schedule [pr] (#10861) 
* fix import

* fix test_multitoutput_ast

* fix test_recursive_swizzle

* test_alu_after_copy

* remove that test
 2025-06-18 12:47:57 +03:00
qazal
8b879b0314 merge TestTensorUOpSpec with the other spec unittests [pr] (#10860) 
* merge TestTensorUOpSpec with the other spec unittests [pr]

* rename to test_uop_spec
 2025-06-18 12:12:08 +03:00
George Hotz
cba6e15937 split grouper and kernelize [pr] (#10854) 2025-06-17 17:54:20 -07:00
George Hotz
a38947b4bb move symbolic and transcendental to uop [pr] (#10771) 2025-06-10 20:51:22 -07:00
George Hotz
81ef879da3 non recursive top_down_rewrite (#10729) 
* non recursive top_down_rewrite

* nicer algorithm

* rewrite bottom up also

* only top down is broken?

* simpler iterative algo

* no recursion errors

* top down and bottom up

* unified rewrite

* simpler rewrite

* clean up comments

* move that comment
 2025-06-09 16:33:04 -07:00
George Hotz
81b9c04574 move high level stuff to unit tests [pr] (#10708) 
* move high level stuff to unit tests [pr]

* process replay on unit tests

* fix pr, less compute

* set omp num threads

* set 200MB buffer size limit

* delete junk

* fix tests

* faster

* move test_indexing to unit

* faster
 2025-06-08 14:05:56 -07:00