b1tg 891a73befc llm: fix chunked prefill (#15182) ...

* llm: fix chunked prefill

* less lines

---------

Co-authored-by: b1tg <b1tg@users.noreply.github.com>

2026-03-07 22:08:31 +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

update assign tests to also test the expected behavior (#15132)

2026-03-04 11:34:43 -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

fix symbolic shape call + chunked prefill (#15149)

2026-03-05 14:02:26 +08: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_disk_tensor.py

update non-contiguous buffer error message [pr] (#15131)

2026-03-04 11:13:26 -05:00

test_dtype_spec.py

everything has dtype.long now (#14661)

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

test_function.py

assign fusion tests on detach and contiguous_backward (#15092)

2026-03-02 15:21:51 -05:00

test_getitem_ops.py

improve consecutive Tensor indexing (#14208)

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

test_gguf.py

update mxfp4 tests to use the same patterns as the others (#15177)

2026-03-06 13:21:40 -05:00

test_gradient.py

don't remove_bufferize for Invalid (#15053)

2026-02-27 15:16:09 -05:00

test_hashing.py

fix tests that failed locally on mac (#13872)

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

test_hcq_graph.py

hcq: fix usb<->cpu mappings (#14827)

2026-02-17 18:04:18 +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_linalg.py

add contiguous back to svd (#15074)

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

test_llm_moe.py

add olmoe support to llm (#13792)

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

test_llm_server.py

llm: fix chunked prefill (#15182)

2026-03-07 22:08:31 +08:00

test_masked_tensor.py

delete test_tensor_uop (#13951)

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

test_metal_graph.py

metal uint32 icb offset overflow (#15156)

2026-03-06 00:54:39 +03: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 more tests to test/null, split some existing ones (#14512)

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

test_setitem_schedule.py

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

2026-02-21 16:33:55 +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_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

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_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