5,542 commits

Author	SHA1	Message	Date
nimlgen	11e1a2b89f	cleaner and faster run_linear (#15987) ... * cleaner and faster run_linear * x * assert for now * x * x * sym_infer * remove sink	2026-04-30 20:15:22 +03:00
qazal	58b34e71bd	failing test for llama useless copies (#15989)	2026-05-01 00:55:29 +09:00
George Hotz	0f7e296f5b	fix some indexing edge cases (#15988)	2026-04-30 08:05:30 -07:00
qazal	55915584e5	viz: fix cfg for emulated amd on the null device (#15976) ... * simple failing when i test it end to end * pass * linter * assemble	2026-04-30 05:18:09 +09:00
nimlgen	dfd2d07005	remove CompiledRunner (#15970) ... * rm usage of CompiledRunner * more tests * last * linter * sink * remove * linter	2026-04-29 22:45:48 +03:00
qazal	a37b605523	remove arch from asm kernel class (#15977) ... * rm arch from kernel * update other tests * update abstractions4.py	2026-04-30 03:39:52 +09:00
chenyu	654e611a29	_bits_to_rand to mixin (#15972)	2026-04-29 13:47:25 -04:00
George Hotz	5f441ecffc	unify reduce + reduce_axis (#15973) ... * unify reduce + reduce_axis * fix all tests * lil cleanups	2026-04-29 10:29:56 -07:00
nimlgen	7787f76dcc	get_runner -> get_runtime (#15967) ... * get_runner -> get_runtime * do not use get_runner * fix * remove get_tunner * remove * fix * x	2026-04-29 18:29:49 +03:00
chenyu	fb188c3c23	UOp.bitcast noop early return (#15968) ... matches Tensor	2026-04-29 09:41:40 -04:00
chenyu	c4bea54e9c	_threefry_random_bits to mixin (#15959) ... start RandMixin	2026-04-28 19:13:57 -04:00
Nino Risteski	5eb1fd5d3c	cleanup: untrack wait Metal buffers (#15954)	2026-04-28 12:54:59 -07:00
nimlgen	77965a22e5	local optimize as rewrite (#15953) ... * local optimize as rewrite * better * x * slighly rename * fix * ugh * remove * x * remove * not weak	2026-04-28 22:51:04 +03:00
qazal	54f00e1013	sqtt: correct rdna4 structs (#15948)	2026-04-28 07:35:50 +09:00
qazal	c58fd85a99	sqtt: add needs_rocprof decorator (#15947) ... * sqtt: add needs_rocprof decorator * version string	2026-04-28 06:22:50 +09:00
chenyu	77f9125c21	move Tensor.pad to OpMixin (#15946)	2026-04-27 16:56:04 -04:00
nimlgen	4164666c72	programinfo (#15942) ... * programinfo * fix * m * x * x * changes * x * fix * rm	2026-04-27 23:12:03 +03:00
chenyu	fe38d6de94	_pad_circular and _pad_reflect_replicate to mixin (#15944)	2026-04-27 16:07:05 -04:00
qazal	8c174bdad4	viz/sqtt: correct exec pipes (#15885) ... * wmma * p2 * test * left * work * pickle * handwritten failing tests * start work * test the pipes * empirical evidence * update rdna4 enum types * VALU pipe 1 * TRANSCENDENTAL pipe * transcendental function units * reorder * wmma pipe * cleanup and notes * smaller * work * diff cleanup * pickle * use se:1 * int	2026-04-28 05:05:49 +09:00
nimlgen	96165ff0d1	validate_with_cpu as rewrite (#15938) ... * validate_with_cpu as rewrite * compil * x * linter * moved * fix	2026-04-26 19:58:53 +03:00
nimlgen	117e9e22dd	estimates from graph (#15937) ... * estimates from graph * test * x	2026-04-26 18:22:53 +03:00
nimlgen	bb652352c7	remove execitem (#15932) ... * remove execitem * f * x	2026-04-25 19:33:04 +03:00
nimlgen	e0ff6cc15c	remove old schedule (#15930) ... * remove old schedule * tests * r * x	2026-04-25 16:46:36 +03:00
qazal	9a23de7d27	viz/cli: unify profile and rewrites, -s ALL default (#15931) ... * work * workg * better * cleanup * better defaults * --ls * better * work * update llama * update	2026-04-25 22:31:24 +09:00
nimlgen	a5e9ea7a60	remove schedule batch 4 (#15927) ... * remove schedule batch 4 * fini	2026-04-25 12:36:55 +03:00
nimlgen	d2ab6ea7a6	remove schedule batch 3 (#15924) ... * remove shcedule batch 3 * batch 6 * batch 7	2026-04-25 11:53:16 +03:00
nimlgen	3c8a2db870	remove schedule() from tests batch 2 (#15923) ... * remove schedule() from tests batch 2 * batch 4	2026-04-25 10:44:41 +03:00
Denys Melnyk	1fdcb13bfb	webgpu: fix weight lookup in export_model after compile_net key change (#15919) ... * fix lookup site in export_model_webgpu after refactoring webgpu (sd): fix export_model weight lookup after compile_net changes fix lookup site in export_model_webgpu after refactoring * add regression test	2026-04-25 10:04:55 +03:00
Christopher Milan	57fbaa3d49	amd: fallback to llvm when comgr is not available (#15914)	2026-04-24 23:30:16 -04: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	b501ba3e42	nll_loss to mixin (#15918)	2026-04-24 15:50:31 -04:00
chenyu	2f9fdb4a37	scatter to mixin (#15917)	2026-04-24 15:37:37 -04:00
nimlgen	f2751955cb	remove linear_to_schedule from tests (#15912) ... * remove linear_to_schedule from tests * x	2026-04-24 20:02:10 +03:00
chenyu	03a7604f76	sort argsort topk allclose to mixin (#15910)	2026-04-24 10:20:46 -04:00
chenyu	7a1adfd2aa	update Tensor.allclose to return Tensor (#15904) ... matches jax	2026-04-24 08:27:17 -04:00
qazal	5eb641395a	viz/cli: select kernel events in -s DEV (#15909) ... * simple test * pass	2026-04-24 21:03:34 +09:00
nimlgen	c0f77c2e1c	hcq graph to linear (#15888) ... * hcq * f * f * linter	2026-04-24 12:42:49 +03:00
Christopher Milan	cbf4946ea6	usb: multiple gpus and better error messages (#15900)	2026-04-24 01:57:19 -04: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	08d9106c9f	scatter_reduce and sparse_categorical_crossentropy to mixin (#15902) ... also use `.ne` to fix `# type: ignore[comparison-overlap]`	2026-04-23 21:06:36 -04:00
chenyu	8cc2c69e21	fix isclose mixin (#15898) ... use `.eq` instead of `==`	2026-04-23 20:40:43 -04:00
chenyu	782bc6aece	broadcast in ElementwiseMixin.div [pr] (#15897)	2026-04-23 16:02:43 -04:00
qazal	7745e05a2f	sqtt: update wave end packet names (#15896) ... * sqtt: update wave end packet names * update wavestart and emu	2026-04-24 04:21:22 +09:00
chenyu	11c197955b	interpolate and cross_entropy to mixin (#15895)	2026-04-23 14:59:45 -04:00
chenyu	f0dbc68aa9	gather to mixin (#15891)	2026-04-23 14:00:57 -04:00
chenyu	87223f870e	logcumsumexp, argmax, argmin, sequential to mixin (#15890)	2026-04-23 12:10:42 -04:00
nimlgen	5cf4ad2fb6	fix resolve param (#15889)	2026-04-23 17:41:44 +03:00
George Hotz	0c3260d5d9	rename VECTORIZE to STACK (#15880)	2026-04-23 10:43:42 +08: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	1fc4b3788c	cummax/cummin to mixin (#15877)	2026-04-22 21:25:39 -04:00