George Hotz 1242b302fa expand UOps with rewrite rules (#5501) ...

* expand UOps with rewrite rules [run_process_replay]

* progress

* much closer

* close, way less bugs

* bunch of expander tests

* fix contract

* ops tests pass

* fix barrier

* mostly passing

* bitcast in expanded ops

* support more expand merges

* all tests pass maybe

* fix empty EXPAND

* fix LIN fuzzing

* add ALL_SAME assert

* all same

* all same work

* raise CompileError

* pass fuzz linearizer

* revert whitespace

* fix nv tensor core test

* fix mypy

* bug fix

* fuzzer passes

* put tests back

* expand arg to idx

2024-07-17 10:17:50 -07:00

..

codegen

expand UOps with rewrite rules (#5501)

2024-07-17 10:17:50 -07:00

engine

infra for multi reduce asts (#5522)

2024-07-17 17:23:46 +03:00

nn

uniform init to match torch (#5494)

2024-07-15 12:07:44 -07:00

renderer

remove outdated thread local aliases (#5493)

2024-07-15 11:28:11 -07:00

runtime

elf loader (#5508)

2024-07-17 17:09:34 +03:00

shape

minimize view _reshape_mask API [run_process_replay] (#5359)

2024-07-10 17:13:01 -04:00

__init__.py

add mnist tutorial

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

device.py

docs: hcq add types (#5495)

2024-07-15 22:14:48 +03:00

dtype.py

BEAM_COMPARE=2 validates the correctness of BEAM kernels (#5458)

2024-07-13 13:53:43 -07:00

function.py

threefry as pattern matcher (#5371)

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

helpers.py

move colorize_float to helpers.py (#5490)

2024-07-15 11:29:03 -07:00

lazy.py

clean up rest of the loadop [run_process_replay] (#5440)

2024-07-12 23:38:51 -04:00

multi.py

s/loadops/metaops [run_process_replay] (#5421)

2024-07-12 13:26:50 -07:00

ops.py

(re)start multireduce in codegen/* (#5391)

2024-07-16 14:20:48 +03:00

tensor.py

Add Interpolate Function (#5482)

2024-07-16 09:44:01 -07:00