George HotzandGitHub bcafa72b7f use tags instead of graph_rewrite_map in rangeify (#12110) ...

* use tags instead of graph_rewrite_map in rangeify

* new style, add realize

* metadata works

* simple failure

* fix

* loops

* stuff becomes a NOOP when you remove it

* stuff becomes a NOOP when you remove it

* tags on bufferize

* bmnist works

* locals don't work

* shippable

* fix some tests

* simpler map_realize

* remove const hack

* debuggable test

* broke

* assign test

* straight up bug

* wooo it passes

* sink shouldn't be there

* fix ops

* bmnist

* kv cache ish

* Set RANGEIFY context variable to 0

* should work normal

* better

* types

* hacks to fix test_symbolic

* pm_add_buffers

* tests should pass

2025-09-14 11:39:01 +08:00

..

device

ops_gpu -> ops_cl (#12103)

2025-09-10 15:15:48 -04:00

external

make method_cache account for compiler (#12156)

2025-09-13 17:00:11 +03:00

mockgpu

mockgpu: no hang if gpuocelot failed (#11915)

2025-08-30 00:44:49 +03:00

models

update test/models (#12142)

2025-09-12 16:43:28 -04:00

opt

ops_gpu -> ops_cl (#12103)

2025-09-10 15:15:48 -04:00

speed

ops_gpu -> ops_cl (#12103)

2025-09-10 15:15:48 -04:00

testextra

break swizzle into three chunks [pr] (#11153)

2025-07-09 15:30:34 -07:00

unit

device: fix envvars (#12159)

2025-09-13 23:38:09 +03:00

web

time viz (#10763)

2025-06-17 19:39:34 +03:00

__init__.py

All devices are equal! (#196)

2020-12-15 23:44:08 -08:00

Dockerfile

new cloud is cloudy [pr] (#7631)

2024-11-11 20:18:04 +08:00

helpers.py

ops_gpu -> ops_cl (#12103)

2025-09-10 15:15:48 -04:00

test_arange.py

early reduce simplify (#12046)

2025-09-10 21:02:46 +08:00

test_assign.py

rename lazydata to uop (#10698)

2025-06-08 08:42:22 -07:00

test_compile_failures.py

unify cpu and llvm (#11982)

2025-09-09 13:54:44 +03:00

test_const_folding.py

clean up TestBitcastConstFolding (#11856)

2025-08-26 15:26:47 -04:00

test_dtype_alu.py

do not use getenv('PTX') in tests (#12095)

2025-09-10 14:04:07 +03:00

test_dtype.py

ops_gpu -> ops_cl (#12103)

2025-09-10 15:15:48 -04:00

test_edgecases.py

add dtypes.index (#12015)

2025-09-06 06:03:44 +02:00

test_fusion_op.py

just schedule in test_recursive_pad [pr] (#8860)

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

test_gc.py

s/lb_refcount/uop_refcount [pr] (#10865)

2025-06-18 21:48:04 +03:00

test_graph.py

hcq: cpu can be graphed (#11474)

2025-08-02 21:01:19 +03:00

test_image_dtype.py

ops_gpu -> ops_cl (#12103)

2025-09-10 15:15:48 -04:00

test_interop.py

hotfix: interop example (#9237)

2025-02-25 10:32:00 +03:00

test_jit_cases.py

enumerate cases of Tensors in the JIT (#10548)

2025-05-28 11:51:27 -07:00

test_jit.py

unify cpu and llvm (#11982)

2025-09-09 13:54:44 +03:00

test_kernel_cache.py

make method_cache account for compiler (#12156)

2025-09-13 17:00:11 +03:00

test_linearizer_dumb.py

early reduce simplify (#12046)

2025-09-10 21:02:46 +08:00

test_linearizer.py

do not use getenv('PTX') in tests (#12095)

2025-09-10 14:04:07 +03:00

test_memory_planner.py

memplan: fix large buffers planning (#11878)

2025-08-27 23:54:27 +03:00

test_method_cache.py

make method_cache account for compiler (#12156)

2025-09-13 17:00:11 +03:00

test_multitensor.py

smaller test_simple_reduce (#12124)

2025-09-11 15:45:38 -04:00

test_nn.py

assert embedding input is integer dtype (#11963)

2025-09-03 01:44:26 +02:00

test_ops.py

use tags instead of graph_rewrite_map in rangeify (#12110)

2025-09-14 11:39:01 +08:00

test_opt_gemm.py

delete more tests (#12043)

2025-09-05 15:31:30 -07:00

test_optim.py

[bounty] Muon optim (#11414)

2025-08-13 14:27:55 -04:00

test_opts.py

ops_gpu -> ops_cl (#12103)

2025-09-10 15:15:48 -04:00

test_outerworld_range.py

remove dtype from range, it will be dtypes.index soon [pr] (#11914)

2025-08-29 09:52:07 -07:00

test_pickle.py

move into codegen late [pr] (#11823)

2025-08-24 10:23:25 -07:00

test_profiler.py

unify cpu and llvm (#11982)

2025-09-09 13:54:44 +03:00

test_quantize_onnx.py

delete more tests (#12043)

2025-09-05 15:31:30 -07:00

test_randomness.py

do not use getenv('PTX') in tests (#12095)

2025-09-10 14:04:07 +03:00

test_rangeify.py

use tags instead of graph_rewrite_map in rangeify (#12110)

2025-09-14 11:39:01 +08:00

test_remote.py

Remote scheduler changes (#11177)

2025-07-21 09:29:44 -07:00

test_renderer_failures.py

Move Ops.SPECIAL arg to src (#11918)

2025-09-04 09:31:44 +02:00

test_sample.py

enable WEBGPU tests with buffer limit (#11489)

2025-08-03 13:02:44 -07:00

test_schedule.py

use tags instead of graph_rewrite_map in rangeify (#12110)

2025-09-14 11:39:01 +08:00

test_setitem.py

Add Test for Setitem (#10559)

2025-07-30 22:03:41 -04:00

test_softmax_fusion.py

a*(1/b) -> a/b on LLVM, CPU (#11743)

2025-08-20 09:35:10 -04:00

test_stunning.py

move bind to big graph (#11539)

2025-08-06 13:27:51 -07:00

test_subbuffer.py

redundant code (#11014)

2025-06-29 09:06:10 -07:00

test_symbolic_jit.py

less iterations for symbolic double for loops (#12006)

2025-09-04 15:09:17 -04:00

test_symbolic_ops.py

less iterations for symbolic double for loops (#12006)

2025-09-04 15:09:17 -04:00

test_tensor_uop.py

rename lazydata to uop (#10698)

2025-06-08 08:42:22 -07:00

test_tensor_variable.py

[bounty] Remove using reshape to register symbolic shape (#11771)

2025-08-28 12:30:49 -04:00

test_tensor.py

Arange overflow raises error and one_hot upcast (#11975)

2025-09-13 00:18:25 +02:00

test_tiny.py

isolate the const hacks in the old kernelize (#12126)

2025-09-12 08:35:35 +08:00

test_to_numpy.py

Apply ruff linting rules to tests (#2473)

2023-11-27 21:24:06 -08:00

test_transcendental.py

Fix DSP transcendentals (#9542)

2025-03-22 11:08:18 +08:00

test_uop_graph.py

update store load noop pattern to use Invalid (#12141)

2025-09-12 22:25:53 +02:00

test_uops_stats.py

do not use getenv('PTX') in tests (#12095)

2025-09-10 14:04:07 +03:00

test_uops.py

ops_gpu -> ops_cl (#12103)

2025-09-10 15:15:48 -04:00

test_zero_copy.py

unify cpu and llvm (#11982)

2025-09-09 13:54:44 +03:00