mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-02-16 09:37:11 -05:00
48061e8400
* Core change to gate stores in IFs * Updates to cstyle renderer to handle IFs around STOREs * Make uops asserts happy * Add tests and fix newly broken tests * make ruff happy * make mypy happy * Simplify renderer to have all gated stores use IF * Revert some changes * Make test_where_fold happy * Revert unnecessary handling of ifs rendering. Was included before when changes weren't fully built out * Rewrite graph to have IFs be dependent on RANGEs if STORE is already dependent on RANGE * Re-change broken test * Make ifs be grouped together * get non-merged IFs working. ALl tests pass except grouping related ifs together * Fix tests by making the IF UOp dependent on the correct node of the STORE UOp * Changes to uopgraph * Simplify graph rewrite logic * Changes to get test_padto_where_multireduce working * Simplify uops.store renderer * Make test_padto_where_multireduce pass but now other tests fail * Clean up uopgraph from scrach work * Ignore sudo IF srcs when rendering * Attempt to fix llvm tests * rm comment * reduce lines * Add line to make mypy happy :( * llvmir fix pt 1 * Mods after rebasing to master * Fix llvmir * Fix ptx tests * Fix other ptx tests * Move changes from uops.py to ops.py * rm uops.py * Fix TestGateStoreRewrite tests * Get multireduce tests working * reset to remote branch * Fix linearizer tests * uop_graph test patch * Add comment to create_gate * hotfix: uncomment those tests * Attempt to fix ptx tests by including whitespace inside if block * Patch from remote tinybox. Tests passing here * Min changes to get some ptx tests passsing * Changes after rebase * Exclude ifs and endifs from ptx * IF conditional branching within ptx * Save lines on delete_redundant_gates * Simplify merge_gates * rm noqa * Remove unnecessary checks when merging gates * Fix ops error msg * Smarter check for if/endif in llvmir * simplify delete redundant gates to only have 2 returns * spacing * Smarter check at beginning of merge_gates * patches from comments * Remove need for merge_gates * include proper srcs in IF from the get-go * test expand ifs dumb will result in 4 ifs, not 1 now * Make tests happy * Fix uops stats * rm merge_gates method. Will add back in separate PR * Spacing * cleaner error msg * Fix uops rendering when expanding. test_failure_43 * patch tests * undo changes in delete_redundant_gates * process replay attempt * re-intro deletion of redundant gates * fix addition of gates when they get nested in stores and loads * patch tests * smarter init of IF srcs when adding gate to STORE * make ruff happy * Resp to comment * include all src[2]'s srcs in IF for gated store * add reference of the storing value to the gate's src * minor patch after rebasing * change ptx renderer --------- Co-authored-by: qazal <qazal.software@gmail.com>
124 KiB
124 KiB