3 people 0f1ca8eb43 torch_load: fix shared storage slicing (#14771) ...

* faster zip_extract + usage in torch load

* clean zip in torch load

* working zipextract in torchload

* tar_extract in tar path

* faster tar path

* tests passing, cleanup needed

* faster tar with 1MB buffer

* comments

* unify storage_source with all paths

* use bufferedreader in zip path

* fix ruff

* clean

* removed unnecessary string conversion

* fix for tensors that share storage

* less hacky

* shared storage test

* test comment

* linter

---------

Co-authored-by: George Hotz <[email protected]>

2026-02-16 14:30:13 +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_assign.py

fix write-after-read tracking (#14754)

2026-02-14 17:23:05 -05:00

test_attention.py

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

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

test_call.py

call autodiff gradient (#14510)

2026-02-03 13:51:02 +08:00

test_conv.py

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

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

test_disk_tensor.py

torch_load: fix shared storage slicing (#14771)

2026-02-16 14:30:13 +08:00

test_dtype_spec.py

everything has dtype.long now (#14661)

2026-02-10 15:08:50 +08:00

test_getitem_ops.py

improve consecutive Tensor indexing (#14208)

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

test_gguf.py

fix wgsl bitcast (#14600)

2026-02-06 16:57:36 -05:00

test_gradient.py

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

2026-02-03 20:20: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

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

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

test_indexing.py

fix test_numpy_parity_and_backward_2d (#14755)

2026-02-14 17:59:00 -05:00

test_linalg.py

IGNORE_OOB -> CHECK_OOB [pr] (#14374)

2026-01-27 12:20:59 -05:00

test_llm_moe.py

add olmoe support to llm (#13792)

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

test_llm_server.py

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

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

test_masked_tensor.py

delete test_tensor_uop (#13951)

2026-01-01 09:25:05 -05: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 more tests to test/null, split some existing ones (#14512)

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

test_setitem_schedule.py

lazy basic setitem to unrealized Tensor (#14756)

2026-02-14 20:27:03 -05:00

test_shm_tensor.py

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

2025-12-17 10:29:57 -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_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

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

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

test_winograd.py

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

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