George HotzandGitHub 7a9dee4e50 add call/param UOps (#14433) ...

* add call/param UOps

* resolve call

* skip that for now

* grad on call

* fix tests

2026-01-30 14:51:45 +08:00

..

apps

return types for all math.py function (#14413)

2026-01-28 20:10:11 -05:00

codegen

add few more types [pr] (#14425)

2026-01-29 14:04:09 -05:00

engine

cleanup around disk buffer [pr] (#14428)

2026-01-29 16:18:44 -05:00

mixin

return types for all math.py function (#14413)

2026-01-28 20:10:11 -05:00

nn

simplify nn.Embedding, support AFTER in CUSTOM_KERNEL (#14419)

2026-01-29 17:22:13 +08:00

renderer

return types for all math.py function (#14413)

2026-01-28 20:10:11 -05:00

runtime

nv: pma for 5090 (#14420)

2026-01-29 20:06:01 +03:00

schedule

add call/param UOps (#14433)

2026-01-30 14:51:45 +08:00

uop

add call/param UOps (#14433)

2026-01-30 14:51:45 +08:00

viz

add call/param UOps (#14433)

2026-01-30 14:51:45 +08:00

__init__.py

move files into uop dir (#10399)

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

device.py

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

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

dtype.py

add InvalidType to ConstType [pr] (#14373)

2026-01-27 14:09:34 -05:00

gradient.py

add call/param UOps (#14433)

2026-01-30 14:51:45 +08:00

helpers.py

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

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

py.typed

…

tensor.py

add call/param UOps (#14433)

2026-01-30 14:51:45 +08:00