17 Commits

Author	SHA1	Message	Date
George Hotz	fce432d2e3	Ops.FUSE makes softmax a single kernel (#9808) ... * KERNELIZE makes softmax a single kernel * single kernel works * softmax works * broken * correct * skip that test * kernelize tests * rename to fuse * better reduce_push_add_ones code * correct now * cleanups * oops * return None if we can't push ones * rename + docs * atol fixes group * flash attention broken test	2025-04-09 22:56:28 +08:00
George Hotz	c1c5227acb	preserve size in dtype ptr [pr] (#8898)	2025-02-05 14:38:57 +08:00
George Hotz	56fa5c1191	dsp simulator (#8869) ... * dsp simulator * progress * fix * close on test tiny * working * less waste * line savings * Device DSP compiler * mock DSP at the bottom * DSP tests * docker caching * test update * need load * skip that test for CI DSP * last touch * ugh	2025-02-04 09:45:04 +08:00
Ignacio Sica	ed1b573868	ignore beam cache in test_tiny for stateless beam (#8771)	2025-01-27 12:56:30 -05:00
George Hotz	bfbe81df71	remove cast before view (#8613) ... * remove cast before view * greener * indexing * that passes too * openpilot too * ack --------- Co-authored-by: qazal <qazal.software@gmail.com>	2025-01-14 15:04:58 -05:00
George Hotz	074315ec08	hotfix: simpler test_mnist_model	2024-12-20 10:18:17 -08:00
George Hotz	32df46cd73	test const pattern [pr] (#8304) ... * test const pattern [pr] * add model to test_tiny	2024-12-17 23:34:17 -08:00
qazal	bb606e5bcf	process replayable ops.py changes from delete_lazy [pr] (#7994) ... * process replayable ops.py changes from delete_lazy [pr] * hotfix: seed tiny_jit	2024-12-02 19:38:31 +08:00
George Hotz	0c7477b108	no bool in range [pr] (#7988) ... * no bool in range [pr] * fix llvm * add arg to range spec * fix broken test * forgot this one * hotfix: test_tiny jit is a real test	2024-12-02 19:05:16 +08:00
George Hotz	d53cd92364	fix tests for delete lazy [pr] (#7980)	2024-12-02 12:00:48 +08:00
George Hotz	4f1f823021	add tiny test for randomness + remove ulong buffers (#7648) ... * add tiny test for randomness * Tensor._device_seeds is a Tuple * no tuple, just a 2 element tensor * no more longs * fix tests, and maybe ocelot works now * NV still doesn't work. cleanup rules * test + two more rules	2024-11-12 12:45:52 +08:00
George Hotz	9a7cc04843	fix viz [pr] (#7519) ... * fix viz [pr] * Update serve.py	2024-11-04 15:02:41 +08:00
George Hotz	72a9ac27e9	support image dtype in cloud [pr] (#7482) ... * support image dtype in cloud [pr] * remove outdated osx hack * unused imports	2024-11-02 23:54:27 +08:00
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	4fe1945df6	llvm if load (#7345) ... * llvm if load * unneeded line * local llvm CI	2024-10-29 11:33:22 +08:00
George Hotz	b1a30677fe	add some tiny tests that should pass everywhere [pr] (#7254)	2024-10-24 14:38:46 +08:00
George Hotz	b20b22a738	hotfix: add test_tiny, because many times it's what you want	2024-10-14 16:32:33 +08:00