tinygrad/extra/optimization
Sieds Lykles cc038b31b6
Shrink instead of reshape to unregister symbolic (#12241)
* Slice to unbind symbolic

* use vmax for now

* assert shape in reshape is valid

* update test_symbolic_ops to use shrink instead of reshape

* remove infer_with_bound_values for npw

* symbolic output doesnt have symbolic strides

* symbolic jit tests use shrink to unregister symbolic

* update test

* update more tests

* wrap vmax in int()

* only create a new st if the store is not an assigne

* unwrap st

* comments
2025-09-19 06:04:35 +02:00
..
extract_dataset.py fix extract_dataset + add tests to CI (#10995) 2025-06-27 01:51:36 +03:00
extract_policynet.py move opt under codegen (#11569) 2025-08-07 14:19:17 -07:00
extract_sa_pairs.py move opt under codegen (#11569) 2025-08-07 14:19:17 -07:00
generate_dataset.sh ops_gpu -> ops_cl (#12103) 2025-09-10 15:15:48 -04:00
get_action_space.py move opt under codegen (#11569) 2025-08-07 14:19:17 -07:00
helpers.py var_vals uses str for var (#12011) 2025-09-06 04:16:12 +02:00
pretrain_valuenet.py move opt under codegen (#11569) 2025-08-07 14:19:17 -07:00
rl.py move opt under codegen (#11569) 2025-08-07 14:19:17 -07:00
run_qnet.py move opt under codegen (#11569) 2025-08-07 14:19:17 -07:00
search.py move opt under codegen (#11569) 2025-08-07 14:19:17 -07:00
test_beam_search.py Shrink instead of reshape to unregister symbolic (#12241) 2025-09-19 06:04:35 +02:00
test_helpers.py fix load_worlds filter_novariable (#7564) 2024-11-05 16:06:39 -05:00
test_net.py move opt under codegen (#11569) 2025-08-07 14:19:17 -07:00
test_time_linearizer.py move opt under codegen (#11569) 2025-08-07 14:19:17 -07:00