82 commits

Author	SHA1	Message	Date
George Hotz	32e9949052	rename lazydata to uop (#10698)	2025-06-08 08:42:22 -07:00
George Hotz	a07caaca0d	handle stride 0 variable reshape (#10536)	2025-05-27 10:00:24 -07:00
George Hotz	411392dfb7	move files into uop dir (#10399) ... * move files into uop dir [pr] * tinygrad.uop is a thing * fix uop docs, no pr * fix viz	2025-05-18 11:38:28 -07:00
George Hotz	df3b320f46	rewriter -> devectorizer [pr] (#9147)	2025-02-18 12:42:08 +08:00
Ankit Avinash	7647cd8428	[bounty] Stride is flip (#8792) ... * replace stride with flip * Complete replacing stride with flip clean flip function in view.py fix tests * fix tests for multi shapetracker * fix tests for fuzz shapetracker * fix tests for fuzz shapetracker * debug * debug * fix * fix * fix --------- Co-authored-by: George Hotz <geohot@gmail.com> Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>	2025-01-31 11:34:10 +09:00
George Hotz	98d01a059d	rename uopgraph to rewriter [pr] (#8682)	2025-01-19 17:03:12 -08:00
eliotgolding	0289fbb1c2	limit real_size to the size of first View of ShapeTracker (#8628) ... * fix real_size * add fuzzer; typing * spacing --------- Co-authored-by: chenyu <chenyu@fastmail.com>	2025-01-16 16:27:39 -05:00
chenyu	ef3752625b	add test case of realize_size with 0 in shape (#8011)	2024-12-03 09:19:50 -05:00
chenyu	74b4d1c1e1	rewrite idx again in real_strides after uop_given_valid (#7600) ... uop_given_valid does not guarantee output to be flat. fixed one last real_strides test.	2024-11-08 14:30:32 -05:00
chenyu	c6189e38c1	simplify_valid in real_strides (#7599) ... improved one more real_strides. after finishing the last one will think about always applying these in to_indexed_uops	2024-11-08 10:45:22 -05:00
chenyu	4378b100ad	make UOp.range arg a tuple [pr] (#7583) ... * make UOp.range arg a tuple [pr] so render works on output of ShapeTracker.to_indexed_uops * fix	2024-11-07 11:58:09 -05:00
chenyu	bb7b5362be	uop_given_valid in real_strides (#7231) ... simplified idx allows deriving more strides	2024-11-07 09:41:16 -05:00
George Hotz	c8bf09b7d4	s/UOps/Ops (#7500) ... * s/UOps/Ops [pr] * fix	2024-11-03 11:26:10 +08:00
chenyu	e90bbe6bbc	failed test cases for 3+ views shapetracker strides (#7226)	2024-10-22 18:49:13 -04:00
chenyu	08a3b97ddc	more generic lt_folding (#7171) ... * more generic lt_folding instead of checking gcd for all uop, check the gcd of the ones that have const_factor() > 1 and still can simplify if others are smallish * fixed that stride too	2024-10-21 09:41:02 -04:00
chenyu	bd8ecf7fd6	remove NumNode (#7035)	2024-10-13 16:42:19 -04:00
George Hotz	85a45164fb	remove pyint [pr] (#7016) ... * remove pyint * bump time on tp [pr] * dont truncate in const fold * remove dead code * Revert "dont truncate in const fold" This reverts commit 29c81db0f7. * remove define_var	2024-10-12 22:36:24 +08:00
George Hotz	a71bb09ec3	remove symbolic file [pr] (#7012)	2024-10-12 18:44:44 +08:00
George Hotz	5ae2de9845	UOp.variable (#7010) ... * UOp.variable [pr] * fix tests * clean * improve name rendering * last bug	2024-10-12 18:20:44 +08:00
George Hotz	a0cb16ac61	node cleanup + local metal test speed [pr] (#6880) ... * node cleanup [pr] * fix tests, including the double one on metal * no time tqdm tests	2024-10-04 18:14:23 +08:00
George Hotz	738a5794a9	last update for new symbolic [pr] (#6877)	2024-10-04 14:58:51 +08:00
George Hotz	eaa1e0eeeb	rename constant_folder to sym [run_process_replay] (#6780)	2024-09-27 14:54:54 +08:00
chenyu	26ebb7cab4	don't use div_folding in lt_folding (#6666) ... * don't use div_folding in lt_folding valids 35 -> 13 * fails the same as before	2024-09-23 01:50:18 -04:00
George Hotz	52c2c4df9c	fix match of sz 0 + dedup kernel ast [run_process_replay] (#6663) ... * fix match of sz 0 [run_process_replay] * empty graph rewrite to dedup st	2024-09-23 11:56:53 +08:00
George Hotz	67a03e72bb	remove expr_idxs [run_process_replay] (#6567) ... * remove expr_idxs [run_process_replay] * goodbye that test	2024-09-17 18:34:51 +08:00
chenyu	00578a021b	re:6125 switch real_size to use uops [run_process_replay] (#6138) ... * switch real_size to use uops [run_process_replay] * enough to pass --------- Co-authored-by: George Hotz <geohot@gmail.com>	2024-08-19 13:20:24 -04:00
chenyu	f7950fc2b6	add E275 missing-whitespace-after-keyword linting rule (#6149) ... requires space after keywords like `assert`, `not`, `return`, `else`	2024-08-17 16:44:34 -04:00
George Hotz	88edc2902d	axis_is_masked with graph_rewrite [run_process_replay] (#6144)	2024-08-17 10:28:49 -07:00
George Hotz	d9cb45af09	only axis is masked [run_process_replay] (#6123)	2024-08-16 21:01:17 -07:00
George Hotz	94aa5f11b5	Revert "use vmax for real_size [run_process_replay] (#6120)" (#6122) ... This reverts commit a6e3211444.	2024-08-16 20:33:19 -07:00
George Hotz	a6e3211444	use vmax for real_size [run_process_replay] (#6120) ... * use vmax for real_size [run_process_replay] * axis is masked	2024-08-16 20:17:23 -07:00
Roelof van Dijk	975b811ad9	names shadowing builtins (#5179) ... Co-authored-by: chenyu <chenyu@fastmail.com>	2024-06-27 08:15:01 -04:00
chenyu	1ffa5ec492	unit test ShapeTracker.consecutive (#4800)	2024-06-01 10:10:51 -04:00
Obada Khalili	e4befa41d7	Fix in _reshape_mask (#4332) ... * handle reshape with remainder in _reshape_mask * remove trailing whitespce * use helper_test_op to generate tensors from shapes * test in shapetracket too * remove whitespace * revert property name in other class tests	2024-04-28 11:57:39 -04:00
chenyu	f6c8032e5d	assert if expr_idxs return might be outside of int32 (#4157)	2024-04-12 14:18:35 -04:00
chenyu	d459956966	move TestGetContraction to test_helpers (#3313) ... also cleaned long lines in test_shapetracker and enabled the line length check	2024-02-04 06:05:01 -05:00
George Hotz	d2aab65958	remove unused expr node (#3170) ... * remove unused expr node * still works * simple expr_idxs * fixup typing	2024-01-18 14:18:43 -08:00
George Hotz	f0c178b7e9	move get_contraction to helpers (#3162) ... * move get_contraction to helpers * move simplify * lines * to_movement_ops is not generic	2024-01-17 19:13:11 -08:00
George Hotz	655c6f61d3	St real size (#3046) ... * track the size in the lazybuffer * shapetracker real size * lint	2024-01-08 14:44:53 -08:00
George Hotz	c003be7309	Revert "track size in shapetracker" (#3043) ... * Revert "track size in shapetracker (#3026)" This reverts commit a8ba1ac08f. * st.size	2024-01-08 13:13:39 -08:00
George Hotz	a8ba1ac08f	track size in shapetracker (#3026) ... * track size in shapetracker * shapetracker adapter * size is an int * create Buffer with st.size * only compare the views for the jit * fix webgpu	2024-01-05 20:15:53 -08:00
Peter Cawley	dae8976889	Fix reshape merging with masks (#2877)	2023-12-20 14:00:58 -08:00
chenyu	5a739e8c20	update one skipped pad_reshape test that was fine (#2860) ... * update one skipped pad_reshape test that was fine had a typo * this one passed	2023-12-19 23:25:52 -05:00
chenyu	ad233d557f	disable reshape merging with masks (#2858) ... fuzzer found a bug, and it's not complete	2023-12-19 19:06:16 -05:00
George Hotz	877c78b4ce	lazy tests (#2796) ... * tests * mini sd is very mini	2023-12-16 08:24:21 -08:00
George Hotz	6d6eb9302d	ruff checks the max line length is 150 (#2734) ... * ruff checks the max line length is 150 * fix tensor.py * a lot more * done	2023-12-12 17:34:47 -08:00
chenyu	b931a20882	minor shapetracker cleanup (#2652)	2023-12-06 11:43:52 -05:00
Amrit Sahu	71d989b476	adding test to cover #2644 failure (#2645)	2023-12-06 11:00:30 -05:00
chenyu	dd8b4632a4	regression test for reshape fix #2616 (#2620)	2023-12-05 11:46:33 -05:00
chenyu	c257a0dd99	minor reshape cleanups (#2619) ... * minor reshape cleanups * mea culpa	2023-12-05 11:23:17 -05:00