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

58 commits

Author SHA1 Message Date
George Hotz
fb61f3519f
 		remove assign contiguous hack (#12659) 
* remove assign contiguous hack

* remove bad contiguous usage in torch backend

* assign
 2025-10-14 16:42:14 +08:00
chenyu
cf8232ec6a
 		clean up more RANGEIFY flag (#12556) 2025-10-09 03:06:48 -04:00
chenyu
ae51bdd06a
 		remove trivial use of RANGEIFY flag (#12550) 
some tests need update still
 2025-10-09 02:29:38 -04:00
qazal
22674798df
 		assert correctness in test_permuted_assignment [pr] (#12483) 2025-10-07 11:42:22 +03:00
qazal
4b60121498
 		fix bmnist torch with RANGEIFY=1 (#12442) 
* fix bmnist torch with RANGEIFY=1

* alt

* test and comment

* this was always wrong

* simple failing test for rangeify

* simple upat to match the old behavior
 2025-10-05 12:34:27 +03:00
qazal
865d5796f8
 		add a test for untested Tensor.assign behavior (#12448) 
* add a test for untested Tensor.assign behavior

* better
 2025-10-04 12:44:56 +03: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
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
George Hotz
32e9949052
 		rename lazydata to uop (#10698) 2025-06-08 08:42:22 -07:00
qazal
660c034da6
 		KERNEL op try 3 (#9061) 
* work

* tolerate shape, maybe this is ASSIGN(RESHAPE(BUF), KERNEL)

* err, it's not ASSIGN(BUF, KERNEL), it's ASSIGN(VIEW(BUF), KERNEL)

* burn the boats

* assign slightly works

* assign works

* cleanup + var_vals can exist

* fine image + fix metadata

* metadata, without making everything 30% slower

* diff pruning

* faster assign schedule

* add_buffer_ops stage

* add kernel_spec back

* add viz display

* more strict kernel_spec
 2025-02-17 14:47:54 +01:00
George Hotz
fb698920f1
 		revert scheduler change (#9019) 
* Revert "cleanup ast rewriter [pr] (#9012)"

This reverts commit bf0bcb2d5a.

* Revert "kernel op cleanups + use ScheduleItem [pr] (#9009)"

This reverts commit c52cd2b437.

* Revert "construct the schedule sink 2 (#8925)"

This reverts commit cfd3db7862.
 2025-02-11 11:34:12 +08:00
qazal
cfd3db7862
 		construct the schedule sink 2 (#8925) 
* work

* delete preload

* fix metadata

* this can keep existing

* assign pruning

* dedup early

* bfs

* cycle asserts

* move assign check

* once
 2025-02-10 22:23:02 +08:00
FICTURE7
66306b5321
 		Fix disk tensor assignment (#8855) 
* Add test for disk tensor assignment failure

* Fix disk tensor assignment

---------

Co-authored-by: qazal <77887910+Qazalin@users.noreply.github.com>
 2025-02-02 13:50:34 +02:00
qazal
1fce864a6d
 		delete multi output support (#8822) 
* delete multioutput for now

* test_schedule

* test_assign too

* linter

* 515 for sd

* update tests and ctx

* update that assign check
 2025-01-30 22:45:50 -05:00
qazal
ae241e96db
 		fix half4 on qcom and gpu (#8573) 
* add test_setitem_half

* this fixes comma benchmark
 2025-01-12 06:23:05 -05:00
qazal
fddaeb6344
 		scheduler deduping spec and asserts [pr] (#8307) 
Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
 2024-12-18 09:21:41 -08:00
qazal
69eb55a529
 		bugfix: dtype mismatch in TestAssign.test_double_assign_alt [pr] (#8310) 2024-12-18 22:29:07 +08:00
geohotstan
8100109c9d
 		Add replicate mode to Tensor.pad (#7608) 
* base implementation

* add tests

* actually remove the assertionerror test

* actually only have reflect for this pr

* change the 4 if-else one liner

* maybe use a lambda

* fix

* maybe a lil cleaner

* fix tests

* complete

* small change

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
 2024-11-18 10:55:38 -05:00
qazal
7a9f3dea54
 		assert a schedule double realize (#7178) 
* assert this

* maybe use lazycache

* Revert "maybe use lazycache"

This reverts commit 7368102906.

* set enable_cache=True

* assert 1 schedule
 2024-10-21 14:16:21 +03:00
chenyu
b36a7273c6
 		RUF018 assignment-in-assert [run_process_replay] (#6172) 
assertion should not have side effect or `-O` breaks.

initially just wanted to fix the one in rearrange, but it also made some long lines less long
 2024-08-19 00:34:52 -04:00
qazal
a4a23c40a0
 		test masked assign views (#4599) 
* possible masked

* not contiguous mask
 2024-05-15 15:06:48 +03:00
qazal
77aa8659f5
 		use assign_targets in LazyOp creation (#4568) 
* start

* correct error

* this is possible

* document it
 2024-05-13 10:24:35 +03:00
qazal
b0fa97e176
 		assert error detail in test_assign (#4567) 
* use regex assert

* that shouldnt raise
 2024-05-13 09:56:05 +03:00
qazal
4e1135a0bc
 		assign buffer read/write tests (#4565) 
* simple tests

* more tests
 2024-05-13 09:43:36 +03:00
qazal
249cadd106
 		fusing crossing diamond assign (#4403) 
* refactor scheduler parents search

* assign target

* unit test

* can't chase this
 2024-05-04 15:19:48 +03:00
qazal
9a47ed0705
 		test crossing diamond assigns (#4298) 2024-04-25 21:52:05 +03:00
George Hotz
967638f0d5
 		update docs, remove corealize (#4264) 
* update docs, remove corealize

* handle 0 line count

* tensor schedule
 2024-04-23 12:05:29 +04:00
qazal
a9bc7c1c49
 		unify assign tests (#4247) 2024-04-22 11:01:15 +03:00
chenyu
bdbcac67f1
 		assign jit test case with other tensor as input (#4098) 
hmm it works
 2024-04-06 14:41:14 -04:00
chenyu
c71627fee6
 		move GlobalCounter to helpers (#4002) 
break circular import between ops and buffer
 2024-03-30 00:30:30 -04:00
George Hotz
60639cccac hotfix: RuntimeError for assign 2024-03-27 11:18:48 -07:00
qazal
9fb573d73c
 		DAG cycle asserts (#3955) 
* assert cycles

* these are cycle errors

* flip to positive
 2024-03-27 11:09:59 -07:00
qazal
d8fafca13a
 		assign regression (#3907) 
* infra

* track mutations

* assign levels

* add seen back

* add test

* infra 2.0

* add assign targets

* dont need levels

* delete

* Update test_assign.py

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>
 2024-03-24 15:12:31 -07:00
George Hotz
54dc48aa47
 		fix assign (#3878) 
* fix assign

* remove terrible optimizer hack

* oops, not realized assigns
 2024-03-22 11:48:48 -07:00
George Hotz
86ee36e697
 		preschedule all (#3875) 2024-03-22 11:20:06 -07:00
George Hotz
4c4d3cb3e3
 		restrict assignment to base (#3809) 
* restrict assignment to base

* add some restrictions there

* more restrictions
 2024-03-18 15:33:06 -07:00
George Hotz
d8296d4a3f
 		simple assign tests (#3807) 2024-03-18 13:57:01 -07:00
George Hotz
0183a05f0a
 		test assign (#3798) 
* Reapply "add failing assign test (#3796)" (#3797)

This reverts commit 1e1beb888c.

* no realized check
 2024-03-18 08:58:04 -07:00
George Hotz
1e1beb888c
 		Revert "add failing assign test (#3796)" (#3797) 
This reverts commit 2dea12832c.
 2024-03-18 08:55:36 -07:00
George Hotz
2dea12832c
 		add failing assign test (#3796) 
* that was a hack

* tests to reveal the issue

* add assign for realized assign
 2024-03-18 08:47:30 -07:00
George Hotz
641f347232
 		simple LoadOps.ASSIGN (#3745) 
* simple LoadOps.ASSIGN

* skip that test

* don't assign in onnx ops gemm

* track cache usage

* recreate the lazybuffer to avoid the cache

* fix contigs

* skip that test

* lol

* better letters
 2024-03-14 20:44:34 -07:00
George Hotz
d52d0b0efb test_assign_kv_cache 2024-03-14 16:17:20 -07:00
George Hotz
3527c5a9d2
 		add Tensor.replace (#3738) 
* add Tensor.replace

* fix dtypes in that test

* should be replace

* and mixtral
 2024-03-14 13:34:14 -07:00
George Hotz
56b914fc8c hotfix: test_assign_contiguous 2024-03-13 17:49:54 -07:00
George Hotz
838afbc351
 		assign tests (#3728) 2024-03-13 17:04:55 -07:00
xarkes
28a8b72024
 		Remove Interpreted device & remaining CPU/TORCH ref (#3423) 
* Remove Interpreted device & remaining CPU/TORCH ref

* Oops

* supports_device was useful

* Fix doc wording

---------

Co-authored-by: chenyu <chenyu@fastmail.com>
 2024-02-16 00:30:21 -05:00
George Hotz
a280cfe169
 		move dtypes to dtype.py (#2964) 
* move dtypes to dtype.py

* fix urllib
 2024-01-01 14:58:48 -08:00
George Hotz
1765849937
 		new lazy, benchmark (#2878) 
* lazy rewrite, try 2

* min fix tests

* pass contig test

* put broken pads back

* move that to realize

* no contig child fixes array packing

* so wrong

* now that's correct

* base children

* fix bind issues

* disable to_image_idx

* fix tests

* that failure shouldn't break other tests

* more fixes

* fix torch

* skip failing tests in CI

* 1e-7

* half is broken

* 1e-6 margin of error
 2023-12-20 14:33:21 -08:00
Christopher Mauri Milan
7f01dd04f0
 		Apply ruff linting rules to tests (#2473) 
* everything except F821

* enable F821 with noqa

* dumb fix

* fix remaining imports and (former) lambdas

* replace _ with noqa to avoid gc
 2023-11-27 21:24:06 -08:00
George Hotz
9e07824542
 		move device to device.py (#2466) 
* move device to device.py

* pylint test --disable R,C,W,E --enable E0611

* fix tests
 2023-11-27 11:34:37 -08:00