George HotzandGitHub 8a6dffc87e Tensor.callify will be the JIT (#14983) ...

* close

* simple callify, support linear in the scheduler

* all tests pass

* everyone is happy

* dumb test

* Remove unnecessary blank line in rangeify.py

2026-02-24 18:42:24 +08:00

..

apps

correct llm.py mem bw benchmark for moe (#14626)

2026-02-20 16:11:22 -05:00

codegen

consider indexing math cost for IMAGE=1 (#14973)

2026-02-23 18:57:45 -05:00

engine

Tensor.callify will be the JIT (#14983)

2026-02-24 18:42:24 +08:00

mixin

move more to mixins (#14780)

2026-02-16 11:35:00 +08:00

nn

fix: optim device is never none here (#14963)

2026-02-22 23:34:57 -08:00

renderer

Bool cast to cmpne (#14544)

2026-02-23 10:31:36 -05:00

runtime

am: tiny err handling cleanups (#14981)

2026-02-24 12:43:45 +03:00

schedule

Tensor.callify will be the JIT (#14983)

2026-02-24 18:42:24 +08:00

uop

Tensor.callify will be the JIT (#14983)

2026-02-24 18:42:24 +08:00

viz

rename rewrites + sink filter + bump to dagre 2.0.0 (#14966)

2026-02-23 22:47:22 +08:00

__init__.py

move files into uop dir (#10399)

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

device.py

remove handle_allreduce_multirank and group_id [pr] (#14869)

2026-02-18 16:13:54 -05:00

dtype.py

cleanup fp8 conversion helpers and fp8 edge-case tests (#14953)

2026-02-22 09:16:42 -05:00

gradient.py

remove all the outerworld stuff, it was too complex (#14852)

2026-02-18 17:44:11 +08:00

helpers.py

print schedule caller with DEBUG=1 (#14933)

2026-02-21 16:22:45 +08:00

py.typed

add a single py.typed (#6083)

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

tensor.py

Tensor.callify will be the JIT (#14983)

2026-02-24 18:42:24 +08:00