Do our own call into Python for DTensor dispatch#166370

Closed

swolchok wants to merge 17 commits intogh/swolchok/862/basefrom

gh/swolchok/862/head

Commits on Oct 28, 2025

Do our own call into Python for DTensor dispatch
swolchok
committed
fix bottom-of-stack PR on "Do our own call into Python for DTensor dispatch"
swolchok
committed
Update on "Do our own call into Python for DTensor dispatch"
swolchok
committed

Commits on Oct 31, 2025

rebase, significant fixes to #166370 and following PR on "Do our own call into Python for DTensor dispatch"
swolchok
committed
fix dtensor dispatch key PRs on "Do our own call into Python for DTensor dispatch"
swolchok
committed
Update on "Do our own call into Python for DTensor dispatch"
swolchok
committed
reproduce dispatch key setup from python fallback in dtensor fallback. fix our "upcall" from Python to DTensor dispatch key to be an explicit redispatch so that the preceding fix doesn't break it. …
swolchok
committed
fixed dependency on "Do our own call into Python for DTensor dispatch"
swolchok
committed

Commits on Nov 1, 2025

Commits on Nov 3, 2025

fix several bugs in #166371 that tests found on "Do our own call into Python for DTensor dispatch"
swolchok
committed

Commits on Nov 4, 2025

remove hardcoded custom op handling. sending for posterity as I plan to experiment with a different approach that doesn't use a DTensor dispatch key on "Do our own call into Python for DTensor disp…
swolchok
committed