14 Commits

Author	SHA1	Message	Date
chenyu	c6cf9e8f0c	remove test_svd_nonfull_5_5 (#16217) ... flaky, kinda overlap with test_svd_general	2026-05-15 13:10:02 -04:00
chenyu	38d407fd58	simplify svd more (#16181) ... all the slowness is scheduling	2026-05-12 23:48:22 -04:00
chenyu	2172363be5	don't use Tensor indexing in svd (#16174) ... prepare mixin, also about 4X faster for 8x8 input	2026-05-12 21:56:19 -04:00
chenyu	63c1f00b80	disable test_svd_general again (#16146) ... flaky on CI	2026-05-11 19:24:32 -04:00
chenyu	fbe8be0b8b	style cleanup to Tensor.qr and svd (#16142) ... * style cleanup to Tensor.qr and svd same kernels * more * enable	2026-05-11 17:16:59 -04:00
chenyu	103ea16ec0	add contiguous back to svd (#15074) ... can cause infinite loop	2026-02-28 16:49:26 -05:00
chenyu	db010a31be	IGNORE_OOB -> CHECK_OOB [pr] (#14374) ... flip the meaning	2026-01-27 12:20:59 -05:00
haofei	526fd4ec71	Fix SVD rank‑1 Jacobi rotation when tau == 0 (#13945)	2026-01-01 00:30:18 -05:00
haofei	20777f30b9	Fix QR/SVD NaNs on zero/orthogonal inputs (#13943)	2025-12-31 23:40:09 -05:00
Sieds Lykles	9f39f6391c	shared_codegen_spec and fix index spec (#12967) ... * split shared_codegen_spec and fix index * add VCONST to program_spec and move index to shared_codegen_spec * working ignore_oob=0 * cleanup * fix spec * undo that * move barrier and special earlier * fix more spec issues * more updates * remove special from program_spec * cleanup and fixes * move more to shared * special is not in shared_spec * some comments * dont do bounds check there	2025-10-29 09:14:11 +01:00
George Hotz	db4a359374	fix up some slow tests that launch python (#12672) ... * fix up some slow tests that launch python * svd nonfull in parallel * split test_advancedindex	2025-10-14 19:13:55 +08:00
George Hotz	3b0b3a2e64	fast RANGEIFY (#12504) ... * rtoposort is fast, can replace rangeify with this * fast rangeify * work * fast rangeify works for mnist * should work * progress * pad fix * FAST * tests passing * don't delete those shape ops * put in rangeify map * ending ranges fix * tests * mstack/mselect no hacks * move to indexing.py * touch up tests + add comments * disable failing test * actually make the file readable * failing * error	2025-10-08 19:38:06 +08:00
chenyu	86c5c969ea	linalg cosmetic change (#12356)	2025-09-30 03:00:59 -04:00
George Hotz	1d307f568c	move device tests to test/device + test cleanups (#11735) ... * move device tests to test/device * test speedups * test device * linalg to unit * upd * so pytest just works * more divide and skip * speed * test devectorize * add pillow	2025-08-19 16:02:20 -07:00