George Hotz e13f2a3092 multi is O(1) (#10183) ... * multi is O(1) * allreduce * no new uops needed * junk * something * simple * that's really what i want * closer * inject _device_num * pretty print * cleanups * this * early dnum * ops allreduce is good * ish * device is the tuple and this is fine * simpler * progress * copy_multi * work * more tests * more tests pass * work * no None axis * tests * no none multi * type fixes * pre commit passes * lil * remove this * mlperf dataloader on mac * that test was wrong * unbind * support DEBUG=2 * realize * only unbind bound vars * don't include fixedvars * graph test * one test * fixedvars in hcq * new ring reduce * ring reduce * simpler ring * mselect * mselect doesn't work * Revert "mselect doesn't work" This reverts commit c78b77bd7d. * Revert "mselect" This reverts commit bb2e430ac3. * simpler * fixups * no optional * fix jit * move things around * cleanup multi * simpler multi * simpler reshape		2025-05-16 23:14:23 -07:00
..
__init__.py	more test cleanups (#2631)	2023-12-05 16:17:57 -08:00
test_allreduce.py	multi is O(1) (#10183)	2025-05-16 23:14:23 -07:00
test_attention.py	move cast to before softmax in attention (#9213)	2025-02-24 17:24:59 -05:00
test_block_reorder.py	toposort is a function [pr] (#10004)	2025-04-23 16:25:03 +01:00
test_device.py	make DISABLE_COMPILER_CACHE a ContextVar [pr] (#9983)	2025-04-22 10:32:54 -04:00
test_disk_cache.py	helpers.diskcache_clear (#4436)	2024-05-05 14:19:01 -04:00
test_disk_tensor.py	setitem tests (some failing) from kernelize (#9940)	2025-04-20 18:47:55 +08:00
test_elf.py	CLANG -> CPU (#9189)	2025-02-20 18:03:09 -05:00
test_gguf.py	add a missing unittest.main() [pr] (#8274)	2024-12-16 14:28:10 -05:00
test_gradient.py	allow VIEW(BUFFER) in Tensor UOps [pr] (#9210)	2025-02-24 13:06:15 +01:00
test_graph_rewrite.py	fix tests for rewrite [pr] (#10167)	2025-05-05 19:19:49 -07:00
test_helpers.py	fix get reduce contraction with test (#9834)	2025-04-10 22:24:21 +08:00
test_linearizer_rewrite.py	use Ops.REDUCE (#9721)	2025-04-04 10:14:34 +08:00
test_microbenchmarks.py	toposort is a function [pr] (#10004)	2025-04-23 16:25:03 +01:00
test_pattern_matcher.py	fix minimum length in pattern matcher (#9736)	2025-04-04 14:57:01 +08:00
test_qcom.py	BufferSpec and ProgramSpec [pr] (#7814)	2024-11-21 12:18:05 +08:00
test_rewrite_map.py	multi stage graph_rewrite_map (#9803)	2025-04-09 15:59:45 +08:00
test_shapetracker.py	rewriter -> devectorizer [pr] (#9147)	2025-02-18 12:42:08 +08:00
test_shapetracker_math.py	[bounty] Stride is flip (#8792)	2025-01-31 11:34:10 +09:00
test_shm_tensor.py	line reduction [pr] (#7296)	2024-10-25 17:05:09 +07:00
test_simplify_valid_idx.py	add check (#10257)	2025-05-12 11:03:01 -04:00
test_tar.py	accept filename decorator [pr] (#8049)	2024-12-05 11:40:59 +08:00
test_tensor_io.py	tar_extract with Tensors (#7853)	2024-12-04 17:03:19 +08:00
test_tensor_uop_representation.py	flip Ops.COPY order [pr] (#10122)	2025-05-01 00:26:24 -04:00
test_tqdm.py	add rounding to tqdm unit scale (#9507)	2025-03-19 12:08:46 -04:00
test_transcendental_helpers.py	Fix dtype comparisons in vectorized transcendental + tests (#9794)	2025-04-16 08:06:12 -04:00
test_uop_resolve.py	rename MathTraits to maximum [pr] (#7356)	2024-10-29 16:43:04 +08:00
test_uop_symbolic.py	Use cmod/cdiv in sym_infer (#10258)	2025-05-12 09:07:28 -04:00
test_uop_vmin_vmax.py	update uop symbolic tests (#10228)	2025-05-09 01:55:53 -04:00
test_upat_compile.py	fast pattern matcher [pr] (#9737)	2025-04-14 15:24:41 +01:00
test_verify_ast.py	toposort is a function [pr] (#10004)	2025-04-23 16:25:03 +01:00
test_view.py	Small bug in _reshape_mask (#8538)	2025-01-08 22:11:24 -05:00
test_viz.py	do not require self for track_rewrites [pr] (#10302)	2025-05-14 18:23:32 +03:00