George HotzandGitHub c780cd9abb new linearizer with early endrange (#12823) ...

* new linearizer with early endrange

* cleanups

* second stage removal

* not store

* do that later

* end cleanup

* fix globals

* end

* multi end

* fix ends earlier

* work

* do_merge_ends

* mini change

* range_gate

* fix cpu

* test fixups

* ranges on index

* not for ptx

2025-10-21 17:37:48 +08: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

clean up more RANGEIFY flag (#12556)

2025-10-09 03:06:48 -04:00

test_block_reorder.py

Move Ops.SPECIAL arg to src (#11918)

2025-09-04 09:31:44 +02:00

test_conv.py

relax atol for some tests (#12422)

2025-10-02 05:04:44 -04:00

test_device.py

fix up some slow tests that launch python (#12672)

2025-10-14 19:13:55 +08:00

test_disk_cache.py

helpers.diskcache_clear (#4436)

2024-05-05 14:19:01 -04:00

test_disk_tensor.py

feat: assert on bufferview math (#12772)

2025-10-17 14:20:08 -07:00

test_dtype_spec.py

skip test_mean_half_precision_overflow (#12331)

2025-09-29 05:15:04 -04:00

test_dtype.py

nbytes should raise an exception when size is unlimited (#11928)

2025-09-03 07:06:20 -07:00

test_elf.py

Mesa NIR backend (NAK/LLVMpipe) (#12089)

2025-10-15 17:38:33 +08:00

test_gguf.py

fix test_dequantization_mxfp4 (#12123)

2025-09-11 14:22:06 -04:00

test_gradient.py

rename lazydata to uop (#10698)

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

test_graph_rewrite.py

remove st from jit/split_reduceop (#12713)

2025-10-16 12:50:58 +08:00

test_hashing.py

run some hashing test only on METAL (#12554)

2025-10-09 02:39:49 -04:00

test_helpers.py

move get_contraction to helpers [pr] (#12594)

2025-10-10 04:28:57 -04:00

test_indexing.py

delete the ShapeTracker (#12720)

2025-10-16 15:36:22 +08:00

test_kernelize.py

after op (#12801)

2025-10-20 12:27:56 +08:00

test_linalg.py

fix up some slow tests that launch python (#12672)

2025-10-14 19:13:55 +08:00

test_linearizer_rewrite.py

early reduce simplify (#12046)

2025-09-10 21:02:46 +08:00

test_llm_tokenizer.py

clean up the LLM tokenizer (#12653)

2025-10-14 14:22:01 +08:00

test_masked_st.py

move high level tests to unit (#10760)

2025-06-10 12:55:44 -04:00

test_microbenchmarks.py

move files into uop dir (#10399)

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

test_mnist_dataset.py

move high level tests to unit (#10760)

2025-06-10 12:55:44 -04:00

test_pattern_matcher.py

if upat returns self, it's none (#10898)

2025-06-20 12:11:19 -07:00

test_rearrange_einops.py

move high level tests to unit (#10760)

2025-06-10 12:55:44 -04:00

test_rewrite_bottom_up_gate.py

Move Ops.SPECIAL arg to src (#11918)

2025-09-04 09:31:44 +02:00

test_rewrite_map.py

Revert "UOp.factor and add chain sorting (#12413)" (#12492)

2025-10-08 03:20:23 +02:00

test_shm_tensor.py

remove trivial use of RANGEIFY flag (#12550)

2025-10-09 02:29:38 -04:00

test_simple_schedule.py

simple schedule test [pr] (#10853)

2025-06-17 16:19:27 -07:00

test_simplify_valid_idx.py

new linearizer with early endrange (#12823)

2025-10-21 17:37:48 +08:00

test_symbolic_failures.py

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

2025-05-28 16:28:37 -04:00

test_symbolic_shapetracker.py

delete the ShapeTracker (#12720)

2025-10-16 15:36:22 +08:00

test_tar.py

accept filename decorator [pr] (#8049)

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

test_tensor_data.py

test_tesnor_data to unit/ (#12013)

2025-09-04 19:58:27 -04:00

test_tensor_io.py

tar_extract with Tensors (#7853)

2024-12-04 17:03:19 +08:00

test_tensor_uop_representation.py

replace mop arg with vectorized index (#12695)

2025-10-15 20:50:06 +08:00

test_tinyfs.py

feat: Tensor.{load, store} (#12629)

2025-10-13 08:04:41 -07:00

test_tqdm.py

Change rate to int in test_tqdm (#10848)

2025-06-18 08:40:43 -04:00

test_transcendental_helpers.py

TRANSCENDENTAL_SUPPORTED_DTYPES->TRANSCENDENTAL_DTYPES (#11752)

2025-08-20 10:29:36 -04:00

test_uop_resolve.py

move files into uop dir (#10399)

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

test_uop_symbolic.py

UOp symbolic tests use eval to check against string (#12643)

2025-10-13 14:19:42 +02:00

test_uop_vmin_vmax.py

Invalid idx (#12067)

2025-09-12 01:42:02 +02:00

test_upat_compile.py

move files into uop dir (#10399)

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

test_viz.py

viz graph drawing small cleanups (#12830)

2025-10-21 15:51:32 +08:00

test_winograd.py

combine pm_drop_and_clauses (#12660)

2025-10-14 10:09:41 +02:00