qazalandGitHub be364a1adb viz: add default tracing group (#12009) ...

This enables seeing rewrites in unit tests like `VIZ=1 python3 test/test_uop_graph.py TestUOpGraph.test_in_bounds_access_gated_local` that call graph_rewrite directly.

`@track_rewrites` keeps existing as an optional helper to organize larger traces.

2025-09-04 23:29:56 +03:00

..

apps

rope cache optim for jit prune in llm.py (#11678)

2025-08-28 08:31:29 -07:00

codegen

refactor and split test_linearizer (#12001)

2025-09-04 10:53:07 -07:00

engine

remove some tc hacks from BEAM (#11980)

2025-09-03 09:59:10 -07:00

frontend

Support onnx If OP (#11648)

2025-08-28 10:17:35 -04:00

nn

assert embedding input is integer dtype (#11963)

2025-09-03 01:44:26 +02:00

renderer

Move Ops.SPECIAL arg to src (#11918)

2025-09-04 09:31:44 +02:00

runtime

make EMULATE a context var (#12002)

2025-09-04 11:15:43 -07:00

schedule

refactor and split test_linearizer (#12001)

2025-09-04 10:53:07 -07:00

shape

split_uop is a method (#11984)

2025-09-03 10:46:17 -07:00

uop

viz: add default tracing group (#12009)

2025-09-04 23:29:56 +03:00

viz

viz: remove useless ctx.save and restore calls (#11996)

2025-09-04 14:56:41 +03:00

__init__.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

device.py

tracing: add default clock (#11935)

2025-08-31 18:24:44 +03:00

dtype.py

nbytes should raise an exception when size is unlimited (#11928)

2025-09-03 07:06:20 -07:00

gradient.py

named expression for POW and MAX gradient (#11858)

2025-08-26 16:03:03 -04:00

helpers.py

make EMULATE a context var (#12002)

2025-09-04 11:15:43 -07:00

py.typed

add a single py.typed (#6083)

2024-08-14 17:31:46 -07:00

tensor.py

update many einsum tests (#11981)

2025-09-03 15:40:20 -04:00