George HotzandGitHub 94599c0637 fixup ast in kernel to be MetaOps.SINK [run_process_replay] (#5424) ...

* fixup ast in kernel to be MetaOps.SINK [run_process_replay]

* fix tests

* fix more tests

2024-07-12 14:01:03 -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

fixup ast in kernel to be MetaOps.SINK [run_process_replay] (#5424)

2024-07-12 14:01:03 -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