2,910 commits

Author	SHA1	Message	Date
chenyu	40d7535eeb	clean up DTYPES_DICT [pr] (#7845)	2024-11-22 10:01:34 -05:00
qazal	9828277c03	view doesn't have buffer, fix the tests [pr] (#7841) ... * view doesn't have buffer, fix the tests [pr] * need assigns	2024-11-22 20:41:55 +08:00
chenyu	69e382216d	fix wino conv output dtype for half inputs (#7829)	2024-11-21 12:13:54 -05:00
geohotstan	cf1ec90ad4	add inverse trig functions to Tensor (#7805) ... * implement inverse trig functions * guess we should still test nans? * magnitude as variable name :D * reorder onnx_ops ops * approximation -> x for consistency * address feedback * simpler acos * improvement? * actually just have asin depend on atan * actually this is nicer * remove a comment --------- Co-authored-by: chenyu <chenyu@fastmail.com>	2024-11-21 09:13:36 -05:00
qazal	5399ff6d06	add UOp.const_with_shape [pr] (#7825) ... * add UOp.const_with_shape [pr] * lines	2024-11-21 21:13:23 +08:00
qazal	e378aeb94e	assert view degrade to const tests post scheduler graph_rewrite [pr] (#7822) ... * assert view degrade to const tests post scheduler graph_rewrite [pr] * low pri, probably tricky, todo	2024-11-21 19:00:41 +08:00
qazal	75c082b883	move CONST/BIND -> VALID to matchers (#7818) ... * delete special const * move CONST/BIND -> VALID to matchers * unittests * fix FUSE_ARANGE=1 * split into two upats * the right way to access view	2024-11-21 16:07:01 +08:00
George Hotz	e9ae2ccd09	_prg to match _buf [pr] (#7816)	2024-11-21 12:44:48 +08:00
George Hotz	c5d458ce02	BufferSpec and ProgramSpec [pr] (#7814) ... * BufferSpec and ProgramSpec [pr] * delete preallocate, it's unused * Revert "delete preallocate, it's unused" This reverts commit dcfcfaccde.	2024-11-21 12:18:05 +08:00
George Hotz	9df5a62c5e	unify to HWQueue [pr] (#7812) ... * unify to HWCommandQueue [pr] * all is HWQueue	2024-11-21 10:33:08 +08:00
chenyu	11cea00090	lower vs_theoretical conv tflops threshold for nv (#7811) ... less flaky	2024-11-20 20:03:49 -05:00
ignaciosica	fc3154a7b3	metal bf16 tc support [pr] (#7408) ... * add bf16 tc for metal * hotfix: spacing * fix tolerance and skip metal bf16 in ci * hotfix: check for dtype_out * hotfix: add check for tc.dtype_out is bf16 back * hotfix: add parens	2024-11-20 14:39:08 -05:00
geohotstan	66a069ee25	add replicate mode to Tensor.pad (#7802) ... * base implementation * add tests * actually remove the assertionerror test * good	2024-11-20 08:39:58 -05:00
George Hotz	eb0bb7dc0b	final dname to device [pr] (#7806) ... * final dname to device [pr] * oops, fix nv	2024-11-20 20:20:28 +08:00
George Hotz	bc977fec53	dname -> device [pr] (#7804) ... * dname -> device [pr] * a few more * only one left	2024-11-20 17:57:14 +08:00
ttomsa	9adeb1041c	fix advanced setitem with 1 in shape (#7797) ... * fix advanced setitem with 1 in shape * linter	2024-11-19 20:04:59 -05:00
ttomsa	170ece6605	fix advanced setitem overlap with 0 (#7793) ... * fix advanced setitem overlap with 0 * fix comment	2024-11-19 16:03:55 -05:00
Gaétan Lepage	159c0bf25e	test_kernel_cache_in_action: fix test (#7792)	2024-11-19 13:34:56 -05:00
Eitan Turok	56017c52a0	Raise error when model architecture does not match state dict (#7772) ... * init * style * style * style * fix test	2024-11-20 00:11:54 +08:00
George Hotz	d71fe7faa5	rename allocator methods to not conflict [pr] (#7788) ... * rename allocator methods to not conflict [pr] * forgot those * transfer + offset	2024-11-20 00:10:29 +08:00
geohotstan	aeaf574a05	add failure test for setitem bug (#7786) ... * add failure test * rename * improve tests * improve tests and no need numpy	2024-11-19 08:54:21 -05:00
qazal	1e31b5ba6b	hotfix: ctx doesn't impact process replay [pr] (#7785)	2024-11-19 20:17:01 +08:00
chenyu	26200574dc	load_state_dict test cases when model and data shard differently (#7774) ... current behavior is weird... when model is sharded and state_dict is not, load shards the state_dict and model shard axis does not change. but if model and state_dict are sharded differently, model shard axis becomes the state_dict axis after load. it should either always use model shard axis or always use state_dict shard	2024-11-18 16:08:24 -05:00
Francis Lata	a1c1b9547f	Context manager support for tqdm (#7770) ... * add context manager support * add test case for context manager usage	2024-11-18 14:12:03 -05: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
chenyu	66d7d5af50	fix Tensor(MultiLazyBuffer) with different dtype should fail (#7757) ... similar to Tensor(LazyBuffer) as we don't cast implicitly	2024-11-17 21:05:45 -05:00
chenyu	df817297b6	fix passing acc_dtype="" to Tensor.prod should fail (#7750) ... similar to sum	2024-11-17 11:38:13 -05:00
chenyu	55707fd00d	fix passing sum_acc_dtype="" to Tensor.sum should fail (#7748)	2024-11-17 10:58:41 -05:00
qazal	99024b922b	to_uop one path for all ops part 1 (#7745) ... * flat meta ops * one path for everything * add tests * view is always base * just run	2024-11-17 20:12:44 +08:00
chenyu	a15a900415	fix Tensor.meshgrid for 1D input and check indexing (#7740)	2024-11-16 23:39:30 -05:00
geohotstan	72a41095bc	add Tensor.meshgrid (#7714) ... * initial implementation and test * some other places that can use meshgrid * revert the onnx_ops change * add to docs * revert interpolate too * update * improve edge case test * might as well test grad * add to test can improve docs --------- Co-authored-by: chenyu <chenyu@fastmail.com>	2024-11-16 23:06:47 -05:00
chenyu	f1efd84c92	fix repeat_interleave with negative dim (#7734)	2024-11-16 10:15:29 -05:00
chenyu	e3105675fb	cond.where(True, False) is cond (#7733)	2024-11-16 09:44:17 -05:00
ignaciosica	597a239e28	Remove UnaryOps, BinaryOps, TernaryOps, MetaOps [pr] (#7725) ... * remove unaryops * remove ternaryops * remove metaops * hotfix * remove binaryops * hotfix: test_pattern_matcher --------- Co-authored-by: qazal <77887910+Qazalin@users.noreply.github.com>	2024-11-16 20:56:56 +08:00
chenyu	22da31b223	clean up Tensor.dot (#7728) ... more docs (similar to numpy) and removed many confusing `-min(n2, 2)`	2024-11-15 18:21:15 -05:00
chenyu	4338c450ac	fix max_pool2d for int tensor with padding (#7726) ... padding inf messed output dtype	2024-11-15 16:22:11 -05:00
chenyu	aeb1301bab	enable a few tests that work now (#7721) ... should mark the ones that are expected to work with expectedFailure, and delete and ones that are not expected to work	2024-11-15 14:30:52 -05:00
qazal	bddee26114	Ops.VALID cleanup, move recursive tests [pr] (#7713)	2024-11-15 20:22:46 +08:00
qazal	703a255301	use the method_cache in test_schedule [pr] (#7712) ... * use the method_cache in test_schedule [pr] * need half	2024-11-15 19:20:47 +08:00
qazal	88f760cc32	test_two_sum doesn't need del (#7711)	2024-11-15 18:50:08 +08:00
George Hotz	9b1605eef9	Revert "objdump intel syntax (#7605)" (#7707) ... This reverts commit 8f8e375f27.	2024-11-15 12:13:04 +08:00
ttomsa	8f8e375f27	objdump intel syntax (#7605) ... * objdump intel syntax * test for objdump intel syntax * add disassemble to ClangCompiler and LLVMCompiler. Use just llvm-objdump * linter	2024-11-15 11:32:23 +08:00
chenyu	9fb396f660	test_ops maxpool2d -> max_pool2d (#7696) ... and avgpool2d -> avg_pool2d for better grepping the tests	2024-11-14 10:39:12 -05:00
geohotstan	f8056a74d6	combine pad2d with pad (#7677) ... * I have pad2d, I have pad, uuh~, pad2dpad~ * fix some small things * strategically placed cast hack * fix more * fix more more * tests * periods	2024-11-14 17:56:02 +08:00
qazal	0914c2fec9	add TestLinearizerFailures test_failure_56 and test_failure_57 (#7682) ... * add test_failure_56 and test_failure_57 * so it's only METAL=1	2024-11-14 12:00:33 +08:00
chenyu	333f5f9f8b	Tensor.bitwise_not (#7688) ... implemented with xor in tensor for now to not add another op. also used it in Tensor.min to fix dtype int on -2**31	2024-11-13 16:31:52 -05:00
chenyu	fb933b79a6	add test case for nll_loss with input > 2D (#7685) ... * failed test case for nll_loss with input > 2D * fixed * add more	2024-11-13 14:34:07 -05:00
geohotstan	9c41c376d3	add Tensor.nll_loss (#7683) ... * move nll_loss to new branch * make nll_loss examples practical * self *is* * add to docs * small	2024-11-13 13:12:13 -05:00
chenyu	3c6fe4b79a	fix Tensor.bitwise_and and Tensor.bitwise_or to support bool (#7684)	2024-11-13 13:10:39 -05:00
chenyu	3d82f8e340	simpler rand_like (#7680)	2024-11-13 12:28:41 -05:00