George HotzandGitHub 8564e28a1b new memory scheduler with explicit refcounts (#4198) ...

* new memory scheduler with explict refcounts

* move central memory planner

* typo + use central memory planner in openpilot

* cleanups

* include lb_refcount in pickle

* replace PlaceHolder with memory planner

* cleaner

2024-04-17 08:46:47 +04:00

..

codegen

Multireduce Kernels - prereq refactor (#4173)

2024-04-14 20:16:54 -04:00

engine

new memory scheduler with explicit refcounts (#4198)

2024-04-17 08:46:47 +04:00

features

search: BEAM_UOPS_MAX to prune candidates with too many uops (#4088)

2024-04-15 18:56:22 -04:00

nn

Resnet fp16 training with fp32 master weight copy (#4144)

2024-04-14 11:25:08 -04:00

renderer

Update ssa input order and annotate types in cstyle and assembly (#4117)

2024-04-09 13:10:29 -04:00

runtime

fix kernargs check in kfd (#4194)

2024-04-17 00:44:50 +03:00

shape

assert if expr_idxs return might be outside of int32 (#4157)

2024-04-12 14:18:35 -04:00

__init__.py

spend 5 lines to bring mnist into the repo (#4122)

2024-04-09 19:24:57 -07:00

buffer.py

new memory scheduler with explicit refcounts (#4198)

2024-04-17 08:46:47 +04:00

device.py

logging: change LOGKERN to LOGKERNS to match LOGOPS (#4193)

2024-04-16 16:08:32 -04:00

dtype.py

rename Scalar to ConstType and cast_scalar to as_const (#3946)

2024-03-26 22:39:58 -04:00

function.py

New docs are in mkdocs (#4178)

2024-04-16 10:59:51 +04:00

helpers.py

PADTO SUM if parents of sum are all zero-preserving (#4140)

2024-04-10 22:16:12 -04:00

lazy.py

new memory scheduler with explicit refcounts (#4198)

2024-04-17 08:46:47 +04:00

ops.py

optionally use a copy kernel instead of SDMA (#4116)

2024-04-12 23:10:41 -07:00

tensor.py

new memory scheduler with explicit refcounts (#4198)

2024-04-17 08:46:47 +04:00