George HotzandGitHub e13f2a3092 multi is O(1) (#10183) ...

* multi is O(1)

* allreduce

* no new uops needed

* junk

* something

* simple

* that's really what i want

* closer

* inject _device_num

* pretty print

* cleanups

* this

* early dnum

* ops allreduce is good

* ish

* device is the tuple and this is fine

* simpler

* progress

* copy_multi

* work

* more tests

* more tests pass

* work

* no None axis

* tests

* no none multi

* type fixes

* pre commit passes

* lil

* remove this

* mlperf dataloader on mac

* that test was wrong

* unbind

* support DEBUG=2

* realize

* only unbind bound vars

* don't include fixedvars

* graph test

* one test

* fixedvars in hcq

* new ring reduce

* ring reduce

* simpler ring

* mselect

* mselect doesn't work

* Revert "mselect doesn't work"

This reverts commit c78b77bd7d.

* Revert "mselect"

This reverts commit bb2e430ac3.

* simpler

* fixups

* no optional

* fix jit

* move things around

* cleanup multi

* simpler multi

* simpler reshape

2025-05-16 23:14:23 -07:00

..

__init__.py

more test cleanups (#2631)

2023-12-05 16:17:57 -08:00

test_allreduce.py

multi is O(1) (#10183)

2025-05-16 23:14:23 -07:00

test_attention.py

move cast to before softmax in attention (#9213)

2025-02-24 17:24:59 -05:00

test_block_reorder.py

toposort is a function [pr] (#10004)

2025-04-23 16:25:03 +01:00

test_device.py

make DISABLE_COMPILER_CACHE a ContextVar [pr] (#9983)

2025-04-22 10:32:54 -04:00

test_disk_cache.py

helpers.diskcache_clear (#4436)

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

test_disk_tensor.py

setitem tests (some failing) from kernelize (#9940)

2025-04-20 18:47:55 +08:00

test_elf.py

CLANG -> CPU (#9189)

2025-02-20 18:03:09 -05:00

test_gguf.py

add a missing unittest.main() [pr] (#8274)

2024-12-16 14:28:10 -05:00

test_gradient.py

allow VIEW(BUFFER) in Tensor UOps [pr] (#9210)

2025-02-24 13:06:15 +01:00

test_graph_rewrite.py

fix tests for rewrite [pr] (#10167)

2025-05-05 19:19:49 -07:00

test_helpers.py

fix get reduce contraction with test (#9834)

2025-04-10 22:24:21 +08:00

test_linearizer_rewrite.py

use Ops.REDUCE (#9721)

2025-04-04 10:14:34 +08:00

test_microbenchmarks.py

toposort is a function [pr] (#10004)

2025-04-23 16:25:03 +01:00

test_pattern_matcher.py

fix minimum length in pattern matcher (#9736)

2025-04-04 14:57:01 +08:00

test_qcom.py

BufferSpec and ProgramSpec [pr] (#7814)

2024-11-21 12:18:05 +08:00

test_rewrite_map.py

multi stage graph_rewrite_map (#9803)

2025-04-09 15:59:45 +08:00

test_shapetracker_math.py

[bounty] Stride is flip (#8792)

2025-01-31 11:34:10 +09:00

test_shapetracker.py

rewriter -> devectorizer [pr] (#9147)

2025-02-18 12:42:08 +08:00

test_shm_tensor.py

line reduction [pr] (#7296)

2024-10-25 17:05:09 +07:00

test_simplify_valid_idx.py

add check (#10257)

2025-05-12 11:03:01 -04:00

test_tar.py

accept filename decorator [pr] (#8049)

2024-12-05 11:40:59 +08:00

test_tensor_io.py

tar_extract with Tensors (#7853)

2024-12-04 17:03:19 +08:00

test_tensor_uop_representation.py

flip Ops.COPY order [pr] (#10122)

2025-05-01 00:26:24 -04:00

test_tqdm.py

add rounding to tqdm unit scale (#9507)

2025-03-19 12:08:46 -04:00

test_transcendental_helpers.py

Fix dtype comparisons in vectorized transcendental + tests (#9794)

2025-04-16 08:06:12 -04:00

test_uop_resolve.py

rename MathTraits to maximum [pr] (#7356)

2024-10-29 16:43:04 +08:00

test_uop_symbolic.py

Use cmod/cdiv in sym_infer (#10258)

2025-05-12 09:07:28 -04:00

test_uop_vmin_vmax.py

update uop symbolic tests (#10228)

2025-05-09 01:55:53 -04:00

test_upat_compile.py

fast pattern matcher [pr] (#9737)

2025-04-14 15:24:41 +01:00

test_verify_ast.py

toposort is a function [pr] (#10004)

2025-04-23 16:25:03 +01:00

test_view.py

Small bug in _reshape_mask (#8538)

2025-01-08 22:11:24 -05:00

test_viz.py

do not require self for track_rewrites [pr] (#10302)

2025-05-14 18:23:32 +03:00