George Hotz 5f441ecffc unify reduce + reduce_axis (#15973) ...

* unify reduce + reduce_axis

* fix all tests

* lil cleanups

2026-04-29 10:29:56 -07:00

..

__init__.py

more test cleanups (#2631)

2023-12-05 16:17:57 -08:00

test_allreduce.py

schedule() -> schedule_linear() in tests (batch 1) (#15915)

2026-04-24 23:40:53 +03:00

test_assign.py

unify reduce + reduce_axis (#15973)

2026-04-29 10:29:56 -07:00

test_attention.py

refactor llm into files (#15780)

2026-04-17 12:33:11 +08:00

test_call.py

unskip TestCall.test_call_gemm_uop [pr] (#15786)

2026-04-17 16:18:51 +03:00

test_callify.py

Tensor.callify will be the JIT (#14983)

2026-02-24 18:42:24 +08:00

test_conv.py

update tests, WEBGPU=1 pytest . passes (#14089)

2026-01-10 00:03:02 -05:00

test_cpu.py

programinfo (#15942)

2026-04-27 23:12:03 +03:00

test_disk_tensor.py

Device.count() (#15842)

2026-04-21 16:46:38 -04:00

test_dtype_spec.py

suppress test warnings from numpy (#15688)

2026-04-11 22:33:20 -04:00

test_function.py

Tensor.invalid -> Tesnor.invalids (#15849)

2026-04-21 11:19:51 -04:00

test_getitem_ops.py

improve consecutive Tensor indexing (#14208)

2026-01-18 15:14:33 -05:00

test_gguf.py

llm: glm 4.5 air (#15771)

2026-04-22 22:47:37 +08:00

test_gradient.py

remove gradient materialize_grads (#15367)

2026-03-19 23:36:03 -04:00

test_hashing.py

MOCKGPU interfaces (#15796)

2026-04-17 21:56:29 -04:00

test_hcq_graph.py

validate_with_cpu as rewrite (#15938)

2026-04-26 19:58:53 +03:00

test_helpers.py

move more tests to test/null, split some existing ones (#14512)

2026-02-03 20:20:20 +08:00

test_indexing.py

differentiable setitem (#15054)

2026-02-27 17:25:15 -05:00

test_invalid_tensor.py

remove all run_schedule from tests (#15846)

2026-04-21 12:02:10 +03:00

test_linalg.py

add contiguous back to svd (#15074)

2026-02-28 16:49:26 -05:00

test_llm_mla.py

refactor llm into files (#15780)

2026-04-17 12:33:11 +08:00

test_llm_moe.py

refactor llm into files (#15780)

2026-04-17 12:33:11 +08:00

test_llm_server.py

refactor llm into files (#15780)

2026-04-17 12:33:11 +08:00

test_masked_tensor.py

delete test_tensor_uop (#13951)

2026-01-01 09:25:05 -05:00

test_metal_graph.py

jit: graphing in uops (#15489)

2026-03-27 19:09:02 +03:00

test_objc.py

objc: fix _classmethods_ dispatch flag (#14854)

2026-04-20 09:35:03 +08:00

test_realize_is_realize.py

jit case with Tensor.empty input, realized means allocated (#14930)

2026-02-21 16:33:55 +09:00

test_rearrange_einops.py

move more tests to test/null, split some existing ones (#14512)

2026-02-03 20:20:20 +08:00

test_schedule_cache.py

move schedule into schedule (#15736)

2026-04-15 11:03:25 +08:00

test_setitem_schedule.py

assign into empty works (#15256)

2026-03-13 10:24:29 -04:00

test_shm_tensor.py

pass device into Tensor._frompy (#15385)

2026-03-20 05:09:01 -04:00

test_symbolic_tensor.py

move more tests to test/null, split some existing ones (#14512)

2026-02-03 20:20:20 +08:00

test_system_pci_scan_bus.py

system: fix pci_scan_bus vendor filter (#14885)

2026-02-19 17:23:32 +03:00

test_tar.py

accept filename decorator [pr] (#8049)

2024-12-05 11:40:59 +08:00

test_tensor_data.py

Tensor(uop) uses device from uop (#15340)

2026-03-18 02:56:06 -04:00

test_tensor_io.py

move more tests to test/null, split some existing ones (#14512)

2026-02-03 20:20:20 +08:00

test_tinyfs.py

fix tinyfs (#14974)

2026-02-24 08:50:53 -05:00

test_winograd.py

move more tests to test/null, split some existing ones (#14512)

2026-02-03 20:20:20 +08:00