George Hotz be8958e26b use CONTRACT before REDUCE (#5903) ...

* use CONTRACT before REDUCE [run_process_replay]

* support half expand

* EXPAND GEP

2024-08-04 16:17:33 -07:00

..

codegen

use CONTRACT before REDUCE (#5903)

2024-08-04 16:17:33 -07:00

engine

setitem with arange fusion 1 (#5898)

2024-08-04 16:09:21 +03:00

nn

pow(2) -> square in RMSNorm [run_process_replay] (#5901)

2024-08-04 14:21:31 -04:00

renderer

get program fields in __post_init__ [run_process_replay] (#5878)

2024-08-02 09:57:12 -07:00

runtime

hcq graph refactor (#5887)

2024-08-03 23:35:33 +03:00

shape

unify negative shape creation to raise ValueError (#5817)

2024-07-30 13:42:59 -04:00

__init__.py

add mnist tutorial

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

device.py

remove useless _ensure_shared_time_base (#5899)

2024-08-04 17:01:54 +03:00

dtype.py

support using str to specify dtype (#5897)

2024-08-04 12:56:28 -04:00

function.py

threefry as pattern matcher (#5371)

2024-07-13 11:59:03 -07:00

helpers.py

indexing fusion 2 (#5888)

2024-08-03 13:13:39 +03:00

lazy.py

support using str to specify dtype (#5897)

2024-08-04 12:56:28 -04:00

multi.py

MLB support reshape for uneven shards (#5804)

2024-08-01 02:36:03 -07:00

ops.py

add test to validate lazyops dims (#5845)

2024-07-31 12:59:38 -07:00

tensor.py

tracemeta fixups (#5904)

2024-08-04 16:15:06 -07:00