George HotzandGitHub 3b674df34b generic changes from define_reg_2 (#11315) ...

* generic changes from define_reg_2

* fix for ptx

* ugh, that one

2025-07-21 15:14:06 -07:00

..

apps

add a test for 1B llm (#11124)

2025-07-07 18:47:25 -07:00

codegen

use AddrSpace instead of local (#11314)

2025-07-21 14:00:06 -07:00

engine

Remote scheduler changes (#11177)

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

frontend

OnnxRunner file as input (#10789)

2025-07-12 14:27:46 -04:00

nn

type remove Union and Optional [pr] (#11283)

2025-07-19 14:05:52 -04:00

opt

use AddrSpace instead of local (#11314)

2025-07-21 14:00:06 -07:00

renderer

generic changes from define_reg_2 (#11315)

2025-07-21 15:14:06 -07:00

runtime

hcq: use devices instead of ids in HCQGraph (#11303)

2025-07-21 20:03:12 +03:00

schedule

rename kernelize to schedule, try 2 (#11305)

2025-07-21 11:18:36 -07:00

shape

type remove Union and Optional [pr] (#11283)

2025-07-19 14:05:52 -04:00

uop

generic changes from define_reg_2 (#11315)

2025-07-21 15:14:06 -07:00

viz

viz: close event streams before profiler render (#11300)

2025-07-21 15:42:31 +03:00

__init__.py

move files into uop dir (#10399)

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

device.py

Remote scheduler changes (#11177)

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

dtype.py

generic changes from define_reg_2 (#11315)

2025-07-21 15:14:06 -07:00

gradient.py

more robust reduce_gradient (#10965)

2025-06-24 14:09:33 -07:00

helpers.py

hcq: move cpu to hcq (#11262)

2025-07-21 15:10:38 +03:00

py.typed

add a single py.typed (#6083)

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

tensor.py

Tensor.roll supports dims=None (#11313)

2025-07-21 17:29:23 -04:00