32 commits

Author	SHA1	Message	Date
chenyu	1bd4551ee1	invalid clone try 2 [PR] (#16648)	2026-06-17 19:44:35 -04:00
George Hotz	53a1226a49	STACK 0 is dtype void (#16650) ... * STACK 0 is dtype void * spec for stack * fix gemm group + END shape * bump	2026-06-17 16:28:32 -07:00
chenyu	f0998e9bba	Revert "invalid clone is anonymous buffer" (#16613) (#16633)	2026-06-16 08:27:48 -04:00
qazal	7d2b0b697d	simple failing test for invalid extra E kernel (#16632) ... * simple failing test for invalid extra E kernel * 6 kernels	2026-06-16 17:57:44 +09:00
chenyu	efd03d7153	invalid clone is anonymous buffer [PR] (#16613)	2026-06-15 20:14:26 -04:00
qazal	34481830f1	rangeify: fix cost function for AFTER(out, CALL) (#16559) ... * simple failing test * fix rangeify cost function * new ops count	2026-06-10 17:30:50 +09:00
chenyu	bb407d8b3c	fix transform_precompiled_call for MULTI (#16510) ... based on my understanding for https://github.com/tinygrad/tinygrad/pull/16084	2026-06-04 20:09:58 -04:00
chenyu	6da785562b	test_custom_kernel_precompile_multidevice (#16401) ... add a test to show what invalids need	2026-05-27 11:19:16 -04:00
chenyu	8631b6f17d	remove use of requires_grad in test/ (#16237)	2026-05-16 17:21:07 -04:00
wozeparrot	a613bcfc6d	allow after on contiguous in spec (#16169) ... * feat: allow after on contiguous * feat: add test	2026-05-12 13:11:44 -07:00
George Hotz	8294d105a7	Update the spec in spec.py to match the current state (#16132) ... * start work on specv2 * more spec * more spec * fix amd emulator * more spec * more * fix test_uop_graph * move those * spec=2 * skip those questionable tests * ptx fix * more spec=2 * store * allow custom function in tensor * spec 2 * fix beam search for tensor cores * delete the old specs * fix import	2026-05-11 20:07:47 -07:00
wozeparrot	4d1a9dca41	fix: don't copy precompiled custom kernel outputs (#16084)	2026-05-07 14:02:38 -07:00
wozeparrot	419d525553	feat: handle multioutput kernel grads (#16028)	2026-05-02 22:31:45 -07:00
chenyu	9192c93b7e	Tensor.invalid -> Tesnor.invalids (#15849) ... matches ones and zeros, and to not share name with UOp.invalid	2026-04-21 11:19:51 -04:00
chenyu	10c262ced8	update tests that use UOp.size (#15753)	2026-04-15 21:58:27 -04:00
wozeparrot	457508d5a0	llama: save more 2 (#15681)	2026-04-11 01:03:36 -07:00
George Hotz	3b75d8a7a2	fix double after bug in rangeify (#15381)	2026-03-20 14:53:46 +08:00
George Hotz	9d95321be3	set allow_implicit=False by default (#15319) ... * set allow_implicit=False by default * modernize beautiful mnist	2026-03-17 17:14:38 +08:00
George Hotz	584ec75aa2	precompile backward (#15311) ... * add precompile backward support * cleanups * fix * compact grad * split v not split * simpler * no NOOPT	2026-03-17 15:28:40 +08:00
George Hotz	3ff03be413	call always has tuple (#15297) ... * call always has tuple * fix pre-commit and simplify * update * fix * move that assert * tuple * fix multi * cleanups * fix merge	2026-03-17 10:58:46 +08:00
chenyu	3e2b7803e6	view assign replaces at buffer identity (#15298) ... matches what functions capture	2026-03-16 19:58:38 -04:00
George Hotz	476276f4b4	support grads on tuples (#15287) ... * support grads on tuples * simpler * grad_fxn works * cleanups * unused	2026-03-16 17:39:34 +08:00
George Hotz	08662bc4ab	add TUPLE/GETTUPLE, simple tests pass (#15286) ... * simple tuple stuff passes * resolved	2026-03-16 15:06:02 +08:00
chenyu	14d1c5fdfd	assign fusion tests on detach and contiguous_backward (#15092)	2026-03-02 15:21:51 -05:00
George Hotz	bb84e389cf	functions for llama trainer (#15045) ... * functions for llama trainer * function there * axis match * fix multi * lil cleaner * there's a bug with HK_FLASH_ATTENTION * training functions * for commit	2026-02-28 12:15:18 +08:00
George Hotz	010d2790ce	fix multi minimal (#15044)	2026-02-27 14:31:58 +08:00
George Hotz	fe3ee8c27e	fix symbolic shapes in calls (#15021) ... * fix symbolic shapes in calls * fix after in the big graph * real tests	2026-02-26 17:17:18 +08:00
George Hotz	2655655a0c	call gradient creates a call (#15020) ... * function creates a full subgraph * tests * fix var * fix tests * implict assign/contig * move kv init	2026-02-26 14:15:29 +08:00
chenyu	ed9d475a12	assign tests with test_function (#15015)	2026-02-25 16:15:59 -05:00
George Hotz	0d35b67f2c	revert realize to only be buffers (#15008) ... * revert realize to only be buffers * fix that * broken attention * Revert "broken attention" This reverts commit a23c3cd96c. * and that	2026-02-25 22:43:06 +08:00
George Hotz	68831cd852	add more tests to test_function (#15003) ... * add more tests to test_function * add function to llm * function decorator on llm * works * symbolic fixups * minimum change * implicit inputs * don't actually update llama yet	2026-02-25 18:42:06 +08:00
George Hotz	e3fa9896b7	start function and add walk rewrite (#14992) ... * start function and add walk rewrite * work * add function on feed_forward * llm progress * stuff * none of that	2026-02-25 13:56:27 +08:00