George Hotz df0f9d6860 add olmoe support to llm (#13792) ...

* add olmoe support to llm

* cleanups

* simpler

* clean

* fix mypy

* lil

* remove dumb assert

2025-12-22 10:41:35 -04: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

fast tinygrad.apps.llm (#13685)

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

test_autogen.py

fix anonymous struct fields (#13610)

2025-12-07 12:56:38 -05:00

test_conv.py

hotfix: disable flaky test

2025-11-13 06:19:28 -08:00

test_device.py

fix local tests for AMD_LLVM (#13738)

2025-12-17 12:23:46 -04:00

test_disk_cache.py

helpers.diskcache_clear (#4436)

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

test_disk_tensor.py

tests passing on tinybox h3 (#13742)

2025-12-17 19:04:34 -04:00

test_dtype_spec.py

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

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

test_dtype.py

clean up can_lossless_cast and add missing pairs [p] (#13793)

2025-12-21 12:18:33 -05:00

test_elf.py

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

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

test_gguf.py

add Q4_K GGUF quantization support (#13700)

2025-12-15 10:17:56 -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

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

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

test_helpers.py

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

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

test_indexing.py

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

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

test_linalg.py

shared_codegen_spec and fix index spec (#12967)

2025-10-29 09:14:11 +01: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

qwen model is working (#13690)

2025-12-15 18:00:34 -04: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

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

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_schedule_cache.py

schedule: cache unbinds for consistent cache keys (#13664)

2025-12-12 17:27:42 -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_simplify_valid_idx.py

add test that shows the validhack regression with bad rewrite order (#13411)

2025-11-21 13:48:30 -05:00

test_symbolic_failures.py

symbolic fuzz failure (#13367)

2025-11-19 14:21:08 -08: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

remove kernelize to prepare for refactors (#13463)

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

test_tqdm.py

symbolic fuzz failure (#13367)

2025-11-19 14:21:08 -08:00

test_transcendental_helpers.py

add loads at the end (#12988)

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

test_uop_resolve.py

update test_uop_resolve (#13777)

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

test_uop_symbolic.py

minor test_uop_symbolic updates (#13551)

2025-12-03 13:17:44 -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_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