Commit Graph

34 Commits

Author SHA1 Message Date
qazal 5517a07a09
viz late to_program and benchmarks [pr] (#6851)
* viz late to_program [pr]

* benchmark resnet

* delete all of checkStatus

* revert that

* fixup

* get from kernel
2024-10-03 18:29:04 +08:00
qazal 1735f8ef1c
viz rewrite part 1 [pr] (#6842)
* core viz spec

* leaaan

* refine docs

* .

* add rewrite_count back in ui
2024-10-02 11:56:25 +08:00
qazal 0cb82f308c
viz don't include graph_rewrites that return a non-UOp (#6832)
* viz don't include graph_rewrites that return a non-UOp

* delete bad things
2024-10-01 18:13:53 +08:00
George Hotz d726eb6f48
uop resolve [run_process_replay] (#6826)
* uop bool and int and stuff [run_process_replay]

* add ne support

* can't even be None anymore

* BinaryOps.AND support

* less compare
2024-10-01 13:11:42 +08:00
George Hotz 9dd9f71011
no global kernel stuff [run_process_replay] (#6808)
* use traceback instead of global metadata crap [run_process_replay]

* save the kernel

* correct, imports clean, no device

* UNPARENTED

* speed

* proudly unparented

* Update ops.py

* update tests for unparented

---------

Co-authored-by: qazal <qazal.software@gmail.com>
2024-09-30 13:52:33 +08:00
qazal 7099af4450
VIZ show rendering errors (#6807)
* VIZ show rendering errors

* show the entire traceback
2024-09-30 09:35:36 +08:00
qazal 3c15e64273
VIZ prep for the new kernel render (#6800)
* refactor to list

* remove prints in test_viz

* more cleanup
2024-09-29 20:06:31 +08:00
qazal 01c9653614
add UOps.BUFFER, delete Buffer in UOps.DEFINE_GLOBAL (#6798)
* delete DEFINE_GLOBAL buffer arg

* add UOps.BUFFER
2024-09-29 18:56:07 +08:00
qazal fb3fe6f39b
better VIZ (#6781)
* ui changes

* make kernels global

* dont save buffers when running VIZ=1

* remove flex in layout

* use os.execv

* del server thread

* server close

* cleanup

* logs cleanup

* rm getenv

* cleanups

* remove global
2024-09-27 18:38:31 +08:00
qazal 900b21ef0c
viz delete const after fold (#6778)
* viz delete const after fold

* add base to tests
2024-09-27 11:58:01 +08:00
qazal 98a81b36e1
viz table view (#6743)
* fix matcher with ctx

* current_kernel fix

* add table

* make the right things clickable

* some more init work

* add kernel resizer

* Revert "add kernel resizer"

This reverts commit 035eef37039aa1e848a766a29e3c4e81bbff2bab.

* allow scroll
2024-09-27 10:26:46 +08:00
qazal 197f8fd986
early uop globals with Buffer (#6753) 2024-09-26 15:34:21 +08:00
qazal 6c69fec1ef
viz more info for rewrite location (#6729) 2024-09-25 14:49:40 +08:00
qazal 048483ee0b
viz fold const nodes and UOp/float4 syntax highlight (#6695)
* fold const nodes

* show rewrite count

* hotfix: cpp

* more syntax highlight

* custom language definitions

* only cpp

* small fixups for UPat

* extend python

* cleanups

* rewrites helper

* better message
2024-09-24 14:36:59 +08:00
qazal e9248b9e27
viz highlight new nodes (#6665)
* p2

* ret adds and dels

* maybe that way

* add additions

* simpler test_viz
2024-09-23 13:46:18 +08:00
qazal d24e4b1042
viz more kernel view work (#6659) 2024-09-23 10:48:35 +08:00
qazal d1bae42d35
viz lowerer and graph_rewrite dedup try 2 (#6652) 2024-09-22 21:09:46 +08:00
qazal 4751159139
second iteration on viz/serve.py (#6643)
* small detail in checkStatus

* better abstractions for the api

* update test_viz

* ui updates
2024-09-22 08:49:44 +08:00
qazal 72c7087420
viz add kernel code (#6620)
* viz add kernel code

* no defaultdict

* ctxs
2024-09-20 16:31:47 +08:00
qazal 74f8f86631
viz kernel tree view (#6614)
* viz kernel tree view

* use get_kernel

* remove current_kernel

cleanup current_kernel

* unset kernel name
2024-09-20 15:52:12 +08:00
qazal 31748c72c4
refactor viz to parse_qs (#6608) 2024-09-19 19:51:41 +08:00
qazal d06b36e527
viz open UPat links in editor (#6601)
* move the reloader

* open links in editor

* less things in ui
2024-09-19 16:48:09 +08:00
qazal 94effe2a71
simple VIZ=1 and get_location changes (#6599)
* simpler replace

* this get_location is fine?

* python things

* ctx location
2024-09-19 15:58:33 +08:00
qazal 309ea63c03
include cached replaces in VIZ=1 (#6596)
* pick some work from vizmore branch

* fix the ctx location

* fix that loc
2024-09-19 14:48:31 +08:00
qazal 0c9b7c9167
more detailed UPat view in VIZ (#6594) 2024-09-19 13:18:11 +08:00
qazal d01e011a8c
start multi graph VIZ=1 (#6587)
* add all rewrites

* add a picker

* drop this here

* more work

* reset that

* start multigraph
2024-09-19 08:31:56 +08:00
qazal 9295bc0189
viz more work [run_process_replay] (#6568)
* infra

* found it

* real work

* bring those back

* cleanup test_viz

* comment that out
2024-09-17 19:27:09 +08:00
qazal 455a27dd43
start viz unittests (#6550)
* test_viz

* more tests
2024-09-17 18:58:23 +08:00
qazal 5a30a32af8
small viz fixups from the swizzle pads branch [run_process_replay] (#6557)
* small viz fixups from the swizzle pads branch [run_process_replay]

* handle indexed ones
2024-09-17 14:37:53 +08:00
qazal dae3615008
replace viz graph when it's sink (#6541) 2024-09-16 16:00:27 +08:00
qazal 9b9b83b8b0
viz tests (#6532)
* vizz fuzz tests

* caching

* print timings

* hotfix: update currentRewrite onClick

* import from typing

* indent into __main__
2024-09-16 13:08:42 +08:00
qazal 89b950c6b3
viz more work (#6517)
* infra

* actually replace the UOp

* extra per rewrite

* dont allow pyint
2024-09-15 16:42:17 +08:00
qazal 2d53e47b14
refactor viz saved context (prereq for tree view) (#6516)
* more styling

* warns

* refactor viz ctx to dataclass

* meh, fine for now

* name ctx

* allow smaller zooms

* more work

* fixup ctx.diffs
2024-09-15 14:08:55 +08:00
George Hotz 9188245677
Viz (#6502)
* start viz tool

* start work

* more readme

* graceful shutdown that reloader

* add VIZ=1

* aesthetics

* typings

* more work

* work left

* more work on rewrites saving

* maybe try zoom

* add some metadata

* generic extra, show code and ast

* more tooling

* add rewritten graphs

* show graph_rewrites

* small details

* more diff cleanups

* differ as the cherry on top

* no useless styles

---------

Co-authored-by: qazal <qazal.software@gmail.com>
2024-09-14 16:15:29 +08:00