George HotzandGitHub c780cd9abb new linearizer with early endrange (#12823) ...

* new linearizer with early endrange

* cleanups

* second stage removal

* not store

* do that later

* end cleanup

* fix globals

* end

* multi end

* fix ends earlier

* work

* do_merge_ends

* mini change

* range_gate

* fix cpu

* test fixups

* ranges on index

* not for ptx

2025-10-21 17:37:48 +08:00

..

apps

clean up the LLM tokenizer (#12653)

2025-10-14 14:22:01 +08:00

codegen

new linearizer with early endrange (#12823)

2025-10-21 17:37:48 +08:00

engine

after op (#12801)

2025-10-20 12:27:56 +08:00

nn

num_batches_tracked has shape () (#12820)

2025-10-21 09:22:39 +08:00

renderer

new linearizer with early endrange (#12823)

2025-10-21 17:37:48 +08:00

runtime

rename Ops.ENDRANGE -> Ops.END (#12824)

2025-10-21 11:32:18 +08:00

schedule

new linearizer with early endrange (#12823)

2025-10-21 17:37:48 +08:00

uop

new linearizer with early endrange (#12823)

2025-10-21 17:37:48 +08:00

viz

new linearizer with early endrange (#12823)

2025-10-21 17:37:48 +08:00

__init__.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

device.py

hotfix: suppress_finalizing on device __del__

2025-10-21 09:04:36 +08:00

dtype.py

fix ruff 0.14.0 [pr] (#12547)

2025-10-09 01:52:30 -04:00

gradient.py

use apply uop for assign to fix assign metadata (#12732)

2025-10-16 20:34:12 +08:00

helpers.py

remove REAL_SUBSTITUTE=0 and make it fast (#12809)

2025-10-20 12:44:20 +02:00

py.typed

add a single py.typed (#6083)

2024-08-14 17:31:46 -07:00

tensor.py

after op (#12801)

2025-10-20 12:27:56 +08:00