qazal
bcb2f1caa3
init REDUCE_AXIS with BinaryOps ( #6256 )
...
* REDUCE_AXIS arg with BinaryOps
* more work in kernel.py
fixup sops.gz
* fix TestGraphRewriteEfficiency
2024-08-24 11:28:41 +03:00
qazal
0d4887e9df
use UOps.WMMA everywhere ( #6255 )
...
* add UOps.WMMA_AXIS
* delete ReduceOps.WMMA from ops
2024-08-23 15:03:26 -04:00
George Hotz
9bc81c6db4
UOps.SHAPETRACKER ( #6129 )
...
* UOps.SHAPETRACKER [run_process_replay]
* no process replay
2024-08-16 23:26:34 -07:00
George Hotz
89c7989659
no shapetracker in ops [run_process_replay] ( #6117 )
2024-08-16 17:23:27 -07:00
qazal
28c75bf2a6
merge uops with ops ( #6111 )
...
Co-authored-by: chenyu <chenyu@fastmail.com >
2024-08-16 18:17:57 -04:00
qazal
c23d44c779
AST is UOp ( #6030 )
...
* most of the work from the uops2 branch
* schedule
* realize
* kernel
* lowerer
* search
* green
* merge uops with ops
* Revert "merge uops with ops"
This reverts commit 1408a59f12
2024-08-16 22:09:00 +03:00
