EventReplay: extend beyond aten ops with auto-import and custom initializers - #607

Open

ajassani wants to merge 11 commits into

mainfrom

feature/event-replay-custom-ops

Commits on Sep 11, 2026

Expand EventReplay to support custom ops beyond aten::
ajassani
committed
8ab482c
Add string arg defaults, op aliases, and module resolution
ajassani
committed
ff57fa2
Add custom initializers, auto-import, and updated docs for EventReplay
ajassani
committed
b1e5907
Fix bugs, add tests, and improve EventReplay docs
ajassani
committed
b249269
Port EventReplay custom-op docs into the ROCm how-to page.

ajassani
and
cursoragent
committed
0b27256
Match extract_batch_context to exact PagedAttentionInit op names.

ajassani
and
cursoragent
committed
bf7b5d0
Move EventReplay tests under tests/ and keep GPU smoke as an example.

ajassani
and
cursoragent
committed
ca587e4
Re-apply custom inits after lazy replay rebuilds tensors.

ajassani
and
cursoragent
committed
c6d7a55
Resolve batched_replay ops through EventReplayer's custom-op path.

ajassani
and
cursoragent
committed
9b06ff8
Let last register_custom_init override built-in initializers.

ajassani
and
cursoragent
committed
0aaf095
Format EventReplay Python with Black 26.3.1 to pass CI lint.

ajassani
and
cursoragent
committed
28c20b8