20 commits

Author	SHA1	Message	Date
chenyu	33b635d23a	Tensor.train -> TRAINING [PR] (#16705) ... * Tensor.train -> TRAINING [PR] * doc	2026-06-22 15:13:22 -04:00
chenyu	243446b44f	remove CONST(DEVICE) from const_like (#16488)	2026-06-03 14:04:51 -04:00
chenyu	8a4203638a	make full with buffer=False deviceless (#16483) ... affects arange and eye	2026-06-03 12:35:59 -04:00
George Hotz	20242fdf1d	update test + spec from shrink_in_render (#16467) ... * update test + spec from shrink_in_render * cast	2026-06-01 19:24:43 -07:00
chenyu	31424cda71	Tensor.requires_grad -> is_param (#16325) ... for optimizer	2026-05-21 19:39:57 -04:00
chenyu	73ea36f4ac	full(buffer=True) (#16311) ... make full a buffer with flag to turn off	2026-05-21 16:34:44 -04:00
Christopher Milan	172f9493e1	move is_dtype_supported to renderer (#16226)	2026-05-20 21:19:37 -04:00
George Hotz	3f2d401464	all tests pass with NOOPT=1 (#16257) ... * all tests pass with NOOPT=1 * fix a few more * noopt 100% pass * noopt 100% pass	2026-05-18 20:39:51 -07:00
chenyu	8631b6f17d	remove use of requires_grad in test/ (#16237)	2026-05-16 17:21:07 -04:00
nimlgen	4164666c72	programinfo (#15942) ... * programinfo * fix * m * x * x * changes * x * fix * rm	2026-04-27 23:12:03 +03:00
nimlgen	d3378010ee	schedule() -> schedule_linear() in tests (batch 1) (#15915) ... * schedule_with_vars -> linear_with_vars in tests * tests batch 1 * batch 2 * estimate_uop * simpler * rm	2026-04-24 23:40:53 +03:00
chenyu	c24da99d56	avg_pool2d, max_pool2d to mixin (#15903) ... * avg_pool2d, max_pool2d to mixin * fix * just dtype * that	2026-04-23 23:36:17 -04:00
chenyu	7c9bc29e44	Tensor method raise if arg is on different device (#15879) ... instead of implicit `to`. this matches torch	2026-04-22 22:20:22 -04:00
chenyu	0191cc73dc	update arange range check (#15794) ... it was not checking negative steps correctly	2026-04-17 16:07:50 -04:00
chenyu	1aa04eab08	simple CreationMixin (#15567) ... start with full_like, zeros_like, ones_like	2026-04-01 23:00:56 -04:00
chenyu	f5c0794df2	fix Tensor.const_like (#15565) ... used to always return a 0-d tensor, now returns an expanded Tensor based on self.shape and matches UOp	2026-04-01 08:35:19 -04:00
nimlgen	d4836ddbb0	canonicalize device from tuple (#15348) ... * will it ifx ci? * test * um	2026-03-18 20:35:52 +08:00
chenyu	94926d00d8	fix rand > uint32.max (#15330) ... need to keep low and high as 1D tensor. `PYTHONPATH=. LLAMA3_SIZE=405B python3 examples/mlperf/models/flat_llama.py` works now	2026-03-17 22:00:01 -04:00
chenyu	557134e1c7	model/test fix that failed with WEBGPU=1 DEBUG=2 (#14706)	2026-02-12 09:08:16 -05:00
George Hotz	d59e6e7a37	move more tests to test/null, split some existing ones (#14512) ... * move more tests to test/null, split some existing ones * null work * null work * move more * fixes * move PIL * PIL in CLIP * don't move that	2026-02-03 20:20:20 +08:00