chenyu e7c2df9113 improve consecutive Tensor indexing (#14208) ...

* improve consecutive Tensor indexing

instead of O(idx_counts*src_dims), it can just be O(idx_counts)

* test correctness

2026-01-18 15:14:33 -05: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_assign.py

move test_assign to unit (#14165)

2026-01-15 17:10:13 -05:00

test_attention.py

fast tinygrad.apps.llm (#13685)

2025-12-14 21:05:21 -05:00

test_autogen.py

autogen: fix enum (#14171)

2026-01-16 01:30:11 -05:00

test_conv.py

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

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

test_device.py

fix misspellings (#13976)

2026-01-02 10:37:38 -05:00

test_disk_cache.py

helpers.diskcache_clear (#4436)

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

test_disk_tensor.py

remove dtypes.index_like and dtypes.fields [pr] (#14207)

2026-01-18 11:49:01 -05:00

test_dtype_spec.py

test fixups / speedups / var_vals refactor (#13812)

2025-12-23 12:05:59 -05:00

test_dtype.py

simpler InvalidType [pr] (#13957)

2026-01-01 13:55:51 -05:00

test_elf.py

add optional compiler in Renderer (#13817)

2025-12-23 17:58:46 -05:00

test_getitem_ops.py

improve consecutive Tensor indexing (#14208)

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

test_gguf.py

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

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

test_gradient.py

gradient of COPY (#13760)

2025-12-19 13:33:59 -05:00

test_graph_rewrite.py

add kernel spec (#12911)

2025-10-25 11:49:20 +08:00

test_hashing.py

fix tests that failed locally on mac (#13872)

2025-12-29 11:23:38 -05:00

test_helpers.py

new ceildiv that fixed symbolic conv (#13944)

2026-01-01 09:02:41 -05:00

test_indexing.py

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

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

test_linalg.py

Fix SVD rank‑1 Jacobi rotation when tau == 0 (#13945)

2026-01-01 00:30:18 -05:00

test_linearizer_rewrite.py

early reduce simplify (#12046)

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

test_llm_moe.py

add olmoe support to llm (#13792)

2025-12-22 10:41:35 -04:00

test_llm_server.py

test fixups / speedups / var_vals refactor (#13812)

2025-12-23 12:05:59 -05:00

test_llm_tokenizer.py

clean up the LLM tokenizer (#12653)

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

test_masked_tensor.py

delete test_tensor_uop (#13951)

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

test_microbenchmarks.py

improve microbenchmarks (#13492)

2025-11-29 10:15:22 -08:00

test_mnist_dataset.py

move high level tests to unit (#10760)

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

test_pattern_matcher.py

SPEC=2 is full spec, SPEC=1 is default (#12910)

2025-10-25 11:10:43 +08:00

test_rearrange_einops.py

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

2026-01-10 00:03:02 -05: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_schedule_cache.py

scheduler: sched_cache bugfix for different Tensor.custom_kernel schedules (#14161)

2026-01-15 14:59:14 +09:00

test_shm_tensor.py

mark slow tests as slow instead of as CI (#13736)

2025-12-17 10:29:57 -04:00

test_simplify_valid_idx.py

faster image load (#13977)

2026-01-04 13:09:59 -05:00

test_symbolic_failures.py

update test_fuzz_failure to not contruct full UOp (#13960)

2026-01-01 15:09:58 -05:00

test_symbolic_tensor.py

delete test_tensor_uop (#13951)

2026-01-01 09:25:05 -05: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

remove kernelize to prepare for refactors (#13463)

2025-11-26 14:18:50 -08:00

test_tqdm.py

update tqdm for edge case (#13956)

2026-01-01 11:37:26 -05:00

test_transcendental_helpers.py

add loads at the end (#12988)

2025-10-30 10:42:19 +08:00

test_uop_repr.py

fix indentation in UOp pretty_print for repeated references (#13857)

2025-12-29 10:46:16 -05:00

test_uop_resolve.py

update test_uop_resolve (#13777)

2025-12-20 12:40:59 -05:00

test_uop_symbolic.py

test case of NOOP store load folding (#13997)

2026-01-03 14:39:26 -05:00

test_uop_vmin_vmax.py

limit AND const min max to ints [pr] (#12918)

2025-10-25 16:07:52 -04:00

test_upat_compile.py

SPEC=2 is full spec, SPEC=1 is default (#12910)

2025-10-25 11:10:43 +08:00

test_validate_oob.py

consolidate IGNORE_OOB=0 tests (#13937)

2025-12-31 15:24:20 -05:00

test_viz.py

make tests faster (#13734)

2025-12-17 09:39:44 -04:00

test_winograd.py

combine pm_drop_and_clauses (#12660)

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