George Hotz 81ef879da3 non recursive top_down_rewrite (#10729) ...

* non recursive top_down_rewrite

* nicer algorithm

* rewrite bottom up also

* only top down is broken?

* simpler iterative algo

* no recursion errors

* top down and bottom up

* unified rewrite

* simpler rewrite

* clean up comments

* move that comment

2025-06-09 16:33:04 -07:00

..

__init__.py

more test cleanups (#2631)

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

test_allreduce.py

minimal mstack pr to fix allreduce (#10649)

2025-06-05 15:14:53 -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

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

test_conv.py

move high level stuff to unit tests [pr] (#10708)

2025-06-08 14:05:56 -07: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

ci speed work 1 (#10676)

2025-06-07 16:33:11 -07: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

rename lazydata to uop (#10698)

2025-06-08 08:42:22 -07:00

test_graph_rewrite.py

non recursive top_down_rewrite (#10729)

2025-06-09 16:33:04 -07:00

test_helpers.py

added unit tests for 'argfix' (#10678)

2025-06-07 22:17:10 -04:00

test_indexing.py

move high level stuff to unit tests [pr] (#10708)

2025-06-08 14:05:56 -07:00

test_keccak.py

smaller tests are faster tests [pr] (#10704)

2025-06-08 10:54:19 -07:00

test_linearizer_rewrite.py

use Ops.REDUCE (#9721)

2025-04-04 10:14:34 +08:00

test_microbenchmarks.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

test_pattern_matcher.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

test_qcom.py

BufferSpec and ProgramSpec [pr] (#7814)

2024-11-21 12:18:05 +08:00

test_rewrite_map.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

test_search.py

fix odd number in get_test_global_size (#10671)

2025-06-06 17:31:35 -04:00

test_shapetracker_math.py

[bounty] Stride is flip (#8792)

2025-01-31 11:34:10 +09:00

test_shapetracker.py

rename lazydata to uop (#10698)

2025-06-08 08:42:22 -07:00

test_shm_tensor.py

line reduction [pr] (#7296)

2024-10-25 17:05:09 +07:00

test_simplify_valid_idx.py

fix tensor int floor div (#10327)

2025-05-21 06:46:54 -04:00

test_symbolic_failures.py

[bounty] Z3 symbolic fuzzer [pr] (#10514)

2025-05-28 16:28:37 -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

rename lazydata to uop (#10698)

2025-06-08 08:42:22 -07:00

test_tqdm.py

add rounding to tqdm unit scale (#9507)

2025-03-19 12:08:46 -04:00

test_transcendental_helpers.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

test_uop_resolve.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

test_uop_symbolic.py

More div conditions (#10432)

2025-05-26 07:36:05 -04:00

test_uop_vmin_vmax.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

test_upat_compile.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

test_verify_ast.py

refactor LOAD(DEFINE_GLOBAL, VIEW) in kernels to LOAD(VIEW(DEFINE_GLOBAL)) (#10541)

2025-05-30 14:27:58 +03:00

test_view.py

Small bug in _reshape_mask (#8538)

2025-01-08 22:11:24 -05:00

test_viz.py

move shared viz fields out of serve.py [pr] (#10684)

2025-06-07 17:18:18 +03:00