4,667 commits

Author	SHA1	Message	Date
George Hotz	4e2895f8d2	safe changes from new dtype branch [pr] (#7397) ... * safe changes from new dtype branch [pr] * only image test on GPU	2024-10-30 17:18:48 +08:00
George Hotz	27995a2a04	vcount + cleanups (#7393) ... * Revert "Revert "Restore vcount [pr] (#7390)" (#7392)" This reverts commit 4ca53db604. * ugh bugfix [pr] * uops_to_dtypes function * fixups * varnames * fix mypy * just 4,8 * tests	2024-10-30 12:50:15 +08:00
George Hotz	4ca53db604	Revert "Restore vcount [pr] (#7390)" (#7392) ... This reverts commit 1058f9c9ff.	2024-10-30 11:40:25 +08:00
George Hotz	1058f9c9ff	Restore vcount [pr] (#7390) ... * Revert "Revert "add vcount to PtrDtype (#7388)"" This reverts commit 399a5219dd. * Revert "Revert "add tests to vcount stuff [pr] (#7389)"" This reverts commit cc8d6dbdf3. * no ptr	2024-10-30 11:27:55 +08:00
George Hotz	cc8d6dbdf3	Revert "add tests to vcount stuff [pr] (#7389)" ... This reverts commit 1b7084899b.	2024-10-30 10:56:49 +08:00
George Hotz	1b7084899b	add tests to vcount stuff [pr] (#7389)	2024-10-30 10:54:54 +08:00
chenyu	f389e1a8a0	test more special values for sin/cos/tan [pr] (#7386)	2024-10-29 21:13:37 -04:00
chenyu	6bf38c35e5	clean up transcendental frexp [pr] (#7384) ... also added some unit tests for frexp	2024-10-29 18:51:37 -04:00
chenyu	d3c192b056	Device method cleanup [pr] (#7375)	2024-10-29 12:49:47 -04:00
qazal	51c0c8d27e	cachable small graph rewrite (#7371)	2024-10-29 22:28:13 +08:00
George Hotz	2cfc7b6695	Index everywhere 2 (#7363) ... * indexing everywhere [pr] * fix tests	2024-10-29 19:29:40 +08:00
qazal	7149eabb34	assert set equality in TestTensorMetadata [pr] (#7364)	2024-10-29 19:29:29 +08:00
qazal	0ebdb136e8	revert metadata with graph_rewrite (#7353) (#7362) ... This reverts commit 540e4179e7.	2024-10-29 19:16:31 +08:00
George Hotz	0af1212164	use assertEqual with new style uops [pr] (#7360)	2024-10-29 18:43:21 +08:00
George Hotz	572499c71a	add indexing to ops_python (#7358) ... * add indexing to ops_python * fix image	2024-10-29 18:11:03 +08:00
qazal	540e4179e7	global UOp to Metadata mapping + inverse DEBUG=2 metadata order [pr] (#7353) ... * add ctx.buf_metadata [pr] * revert metadata insertion order * lint rename	2024-10-29 17:12:00 +08:00
George Hotz	2fdfcffe4c	improve ci speed [pr] (#7357)	2024-10-29 17:00:35 +08:00
George Hotz	b647fa7514	rename MathTraits to maximum [pr] (#7356)	2024-10-29 16:43:04 +08:00
George Hotz	3989bd2682	idiv + reciprocal [pr] (#7354) ... * idiv + reciprocal * remove upcast from div * fix docs	2024-10-29 15:54:19 +08:00
George Hotz	d9d4dd6756	faster ci [pr] (#7348)	2024-10-29 14:01:44 +08:00
George Hotz	4cb236a495	index in cstyle (#7328) ... * index only in cstyle * fix prefix dtypes * fix tests * global indexing * Revert "global indexing" This reverts commit 4d507e8abb. * fix image * fix image * ptx tests * fix CUDA dtype rendering	2024-10-29 13:06:26 +08:00
George Hotz	4fe1945df6	llvm if load (#7345) ... * llvm if load * unneeded line * local llvm CI	2024-10-29 11:33:22 +08:00
chenyu	6021bf87f4	unify T = TypeVar("T") (#7342)	2024-10-28 18:43:44 -04:00
chenyu	c398f2467c	test uop mul min/max do not have nan in 0*inf (#7340)	2024-10-28 17:52:01 -04:00
chenyu	0843734927	clean up nan handling in transcendental (#7332) ... * clean up nan handling in transcendental * skip remu crash	2024-10-28 16:21:49 -04:00
Sieds Lykles	75dcd98e79	Fix calculation of vmin and vmax in multiplication when one src is negative and the other src has negative min and positive max (#7333) ... Co-authored-by: chenyu <chenyu@fastmail.com>	2024-10-28 16:01:46 -04:00
qazal	e46edc22aa	use unittest helpers in TestTensorMetadata [pr] (#7329) ... * use unittest helpers in TestTensorMetadata [pr] * fix that * 5 args	2024-10-28 18:38:30 +08:00
chenyu	cb5702f170	tiny cleanup to transcendental xexp2 (#7326) ... also added test for exp and log of nan and inf	2024-10-27 21:54:20 -04:00
chenyu	4c855ae692	unit test transcendental helpers (#7325) ... added a test to run UOps with const inputs. seems to have issue with both payne_hanek_reduction and cody_waite_reduction	2024-10-27 19:55:00 -04:00
qazal	8d9459f281	always run process replay with contextvars (#7323) ... * always run process replay with contextvars [pr] * not the last two * extra * no pr	2024-10-27 20:44:42 +02:00
talati	d4d201d87b	fixing branch condition on UOps.IF in the ptx renderer (#7315) ... * fixing branch condition on UOps.IF in the ptx renderer * ptx works --------- Co-authored-by: Nick Talati <nick.talati@quantworks.com> Co-authored-by: qazal <77887910+Qazalin@users.noreply.github.com> Co-authored-by: qazal <qazal.software@gmail.com>	2024-10-27 14:27:38 +02:00
qazal	a410b46c1d	unskip test_gated_store_with_if [pr] (#7319)	2024-10-27 14:03:12 +02:00
Maximilian Wolf	3c992250d5	Failing test: different behavior on different devices (#7193) ... * add minimal failing test * more tiny makes linter happy * tinyfy * no walrus in assert * a tiny bit simpler * minimal * better place, better name, expected failure * skip devices with correct behavior	2024-10-27 09:53:58 +08:00
chenyu	d66fe7a66f	fix simplify_valid (#7313) ... the simplex should compare with valid bound, not its vmin	2024-10-26 14:21:12 -04:00
chenyu	0a4d01f6d4	disable simplify_valid (#7312) ... fixed test_failure_55. will reenable it later after fixing the bug	2024-10-26 12:42:48 -04:00
nimlgen	293714610a	capture beam log runtime errors (#7311)	2024-10-26 13:59:45 +03:00
nimlgen	3c62315aa8	add resnet pf (#7310) ... * add resnet pf * all platforms	2024-10-26 13:20:32 +03:00
chenyu	e7cd21c5e3	remove custom render in test_simplify_valid_idx (#7303) ... use UOp render to compare	2024-10-25 10:20:26 -04:00
George Hotz	aadf688aeb	order flipper as *normal* rewrite rule (#7300) ... * instant isn't actually used [pr] * order flipper as *normal* rewrite rule * fix inf loop * need simplify now	2024-10-25 21:28:30 +08:00
George Hotz	3c31497f55	instant isn't actually used [pr] (#7299) ... * instant isn't actually used [pr] * tolerance bump	2024-10-25 21:01:29 +08:00
George Hotz	199a991237	line reduction [pr] (#7296)	2024-10-25 17:05:09 +07:00
George Hotz	4812801aa6	try for canonical order (#7286) ... * try for canonical order * cmp better * disable bad tests * flip const order * fix test * fix tests * different fix for NOOP * metaclass here * fix tests * narrower scope	2024-10-25 16:04:54 +08:00
George Hotz	004af512e6	try all matches in the function (#7288)	2024-10-25 14:17:04 +08:00
chenyu	90f720d703	limit idiv by neg bound to only if s0 is non-negative [pr] (#7277) ... also updated the tests when div by negative const	2024-10-24 15:46:50 -04:00
chenyu	b777cfdcba	update test_max_simplify_and_cancel (#7270) ... it's fixed and no longer dumb	2024-10-24 10:29:05 -04:00
George Hotz	9a3d498d9c	with commutative hack, uops can change. fix that (#7266) ... * with commutative hack, uops can change. fix that * simpler	2024-10-24 18:50:23 +08:00
qazal	d482d927a8	hotfix: nobody uses [run_process_replay] [pr] (#7264)	2024-10-24 13:37:29 +03:00
George Hotz	e4631a47f4	symbolic arange support (#7252) ... * symbolic arange support WIP [pr] * smin/smax from old try * pad2d symbolic works * real test * sym arange * symbolic arange test passes * double mod is single mod * lol that's not right * more tests * Update ops.py	2024-10-24 17:55:53 +08:00
George Hotz	532b7b018c	add smin/smax (#7253) ... * add smin/smax * don't create var with var * better test errors * add failing test * enable shape simplification * fix tests * Update view.py * simpler and simplify	2024-10-24 16:10:49 +08:00
George Hotz	b1a30677fe	add some tiny tests that should pass everywhere [pr] (#7254)	2024-10-24 14:38:46 +08:00