George Hotz c13da83f12 tests from lowerer branch (#5339) ...

* tests from lowerer branch

* Update test_image_dtype.py

* Update test_image_dtype.py

* Update test_image_dtype.py

2024-07-08 21:23:19 -07:00

..

extract_policynet.py

move graph/search to engine (#4596)

2024-05-14 23:12:59 -07:00

extract_sa_pairs.py

move graph/search to engine (#4596)

2024-05-14 23:12:59 -07:00

generate_dataset.sh

Convert BinaryOps.DIV to UnaryOps.RECIP and BinaryOps.IDIV (#4887)

2024-06-14 02:43:46 -07:00

get_action_space.py

tests from lowerer branch (#5339)

2024-07-08 21:23:19 -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

move graph/search to engine (#4596)

2024-05-14 23:12:59 -07:00

run_qnet.py

move graph/search to engine (#4596)

2024-05-14 23:12:59 -07:00

search.py

move graph/search to engine (#4596)

2024-05-14 23:12:59 -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

move graph/search to engine (#4596)

2024-05-14 23:12:59 -07:00

test_time_linearizer.py

move graph/search to engine (#4596)

2024-05-14 23:12:59 -07:00