George Hotz 838cd078bc use atomics for embedding backward (#14400) ...

* embedding is slow

* failing

* float is fine

* null

* it fails

* simplify embedding with broadcasting

* ATOMIC_ADD incoming

* min change

* simpler test

* better test

* fix test

* real test

* simpler

* cleanups

* types and names

* _zero_kernel

* grad multi

* hack

* none

* multi unshard

* more for call

* don't tag in call

* good

* call_multi

* call_multi wow claude is useless

* embedding backward mutli test

* test passes

* fix as_param

* shape_to_shape_arg

* add clip

* before cast

* fix spec=2, use atomics

2026-01-30 18:10:59 +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

allow Tensor setitem input to be list/tuple (#14432)

2026-01-29 21:26:58 -05:00

test_attention.py

fast tinygrad.apps.llm (#13685)

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

test_autogen.py

Revert "move is_dtype_supported logic to renderer (#14188)" (#14237)

2026-01-20 12:19:14 +09:00

test_call.py

use atomics for embedding backward (#14400)

2026-01-30 18:10:59 +08: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

failed test case for disk setitem (#14426)

2026-01-29 14:54:19 -05:00

test_dtype_spec.py

tinygrad changes from ucode (#14336)

2026-01-26 11:30:18 +08: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

Revert "add retry to fetch (#14370)" (#14385)

2026-01-28 09:35:37 +08:00

test_indexing.py

decompose long to ints where unsupported, try 2 (#14383)

2026-01-27 23:20:43 -05:00

test_linalg.py

IGNORE_OOB -> CHECK_OOB [pr] (#14374)

2026-01-27 12:20:59 -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

remove llm generate always reset start_pos (#14276)

2026-01-21 16:54:30 -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

add wrapper class for the -0.0 != 0.0 issue (#14339)

2026-01-26 16:52:37 +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

Add ContextVar to disable the scheduler cache (#14257)

2026-01-27 19:55:29 +08: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

comment out fold_where_closure (#14316)

2026-01-24 10:15:42 -05:00

test_uop_vmin_vmax.py

improve min/max for AND (#14356)

2026-01-26 15:44:18 -05: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

IGNORE_OOB -> CHECK_OOB [pr] (#14374)

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

test_viz.py

make tests faster (#13734)

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

test_winograd.py

fix winograd padding order (#14294)

2026-01-22 23:00:14 -05:00