George Hotz 347a3acb37 add renderer class (#4524) ...

* add renderer class

* tests pass

* fix pylint

* fix tensor cores

2024-05-10 21:40:02 -07:00

..

extract_policynet.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

extract_sa_pairs.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

generate_dataset.sh

update sops.gz, fix tests and add new linearizer test (#4437)

2024-05-05 17:31:25 -04:00

get_action_space.py

multioutput uoping infra (#3706)

2024-03-15 21:56:59 -07:00

helpers.py

fuzz_linearizer: add --ast and --file params to read kernels (#3877)

2024-03-22 14:27:40 -04:00

pretrain_valuenet.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

rl.py

fixed imports (#2185)

2023-10-30 22:07:17 -07:00

run_qnet.py

Q network for pruning BEAM / uops deduping / BEAM_ESTIMATE (#2142)

2023-10-27 10:53:06 -10:00

search.py

add renderer class (#4524)

2024-05-10 21:40:02 -07:00

test_beam_search.py

use float32 for rand buffer in test_beam_search and test in metal (#3831)

2024-03-19 23:22:58 -04:00

test_net.py

fixed imports (#2185)

2023-10-30 22:07:17 -07:00

test_time_linearizer.py

KOPT is over, BEAM is upstream (#2071)

2023-10-16 09:46:03 -07:00