George Hotz
82be8abfd2
move opt under codegen ( #11569 )
2025-08-07 14:19:17 -07:00
George Hotz
92678e59ee
move kernel to opt ( #10899 )
2025-06-20 15:22:28 -07:00
chenyu
5e7ad70aae
don't run linearize().uop tests in get_action_space test ( #10766 )
...
* don't run linearize().uop tests in get_action_space test
this part takes 2 minutes in CI and has nothing to do with action space. also not sure if the "for some reason" comment is still relevant
* -n=auto test/models
2025-06-10 17:23:53 -04:00
chenyu
6c30948df6
hand_coded_optimizations returns list[Opt] [pr] ( #9938 )
...
new api looks like `k.apply_opts(hand_coded_optimizations(k))`
2025-04-19 20:26:59 -04:00
chenyu
8c6299bced
move hand_coded_optimizations to heuristic.py [pr] ( #9844 )
...
* move hand_coded_optimizations to heuristic.py [pr]
also folded all long lines
* make a copy and rename self -> k
* fix test
2025-04-10 23:40:16 -04:00
George Hotz
74ee9febec
remove iter from uopgraph ( #6110 )
...
* remove iter from uopgraph
* linearize returns uops
* fix tests
* linearize in linearize
* tests fix
* touchup
* test failures
2024-08-16 15:58:29 -07:00
George Hotz
03c2dc8bd7
lowerer is kernel [run_process_replay] ( #5437 )
2024-07-12 18:50:55 -07:00
George Hotz
870dc8c350
s/Linearizer/Lowerer [run_process_replay] ( #5428 )
2024-07-12 15:54:07 -07:00
George Hotz
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
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
chenyu
8080298739
s/tinytqdm/tqdm ( #5103 )
...
except in unit test where tqdm is imported
2024-06-22 14:18:26 -04:00
chenyu
a3ed4176c8
use tinytqdm in active tests and examples ( #5038 )
...
* use tinytqdm in active tests and examples
stress test this before 0.9.1
* no set_description
2024-06-18 16:01:19 -04:00
George Hotz
ff64bcab69
move graph/search to engine ( #4596 )
2024-05-14 23:12:59 -07:00
qazal
e3e89c244b
multioutput uoping infra ( #3706 )
...
* linearize multioutput
* add vars to copy
2024-03-15 21:56:59 -07:00
George Hotz
7103b716c4
merge kernel and optimizer ( #2200 )
...
* merge kernel and optimizer
* linearize is reentrant
* move global/local size
* clean up linearizer copy
* remove unneeded lin copies
* stop linearizing twice
* oops, that should be None
2023-11-01 15:20:01 -07:00
George Hotz
abeba8f1fc
optimization: get actions in CI ( #2125 )
...
* get actions in CI
* actually run the test
* pythonpath
2023-10-20 12:22:01 -07:00
George Hotz
c36d306606
KOPT is over, BEAM is upstream ( #2071 )
...
* create cache for q learning
* make linter happy
* global beam
* where it belongs
* bugfix
* ditch the kopt, use the beam
* faster lin and DEBUG=2 okay
* remove kopt, move search to features
2023-10-16 09:46:03 -07:00
George Hotz
49bcfec383
0s in the action space ( #2070 )
...
* 0s in the action space
* simpler
* skip duplicate actions
2023-10-14 11:22:48 -07:00
George Hotz
4124cf1df5
cleanup tensor cores, expose exclude local upcast ( #2064 )
...
* expose exclude_local_upcast
* convert apply tensor cores to ops
* update comment
* put LOCAL back to what it was, BEAM is better than way
2023-10-14 09:21:03 -07:00
George Hotz
90c777d815
remove apply_auto_opt ( #2063 )
2023-10-13 07:44:14 -07:00
George Hotz
6f1810af2d
with unroll, the action space goes from 161 -> 127 ( #2060 )
...
* with unroll, the action space goes from 161 -> 127
* more reliable instrumentation
* beam search is so op
* beam bugfix
2023-10-12 20:52:23 -07:00
