Fix device placement for memory-planned buffers in Runtime.load_program - #22058

Merged

shoumikhin merged 18 commits into

mainfrom

fix-pybindings-planned-buffer-device

Aug 25, 2026

Commits on Aug 24, 2026

Fix device placement for memory-planned buffers in Runtime.load_program
shoumikhin
committed
25a9fce
Document how planned memory is shared between methods
shoumikhin
committed
d401727
Allocate device buffers before host arenas
shoumikhin
committed
6017c7e
Raise instead of aborting on a buffer and device count mismatch
shoumikhin
committed
638890f
Read a method's planned buffer metadata in one pass
shoumikhin
committed
43f274e
Declare the direct dependency on device_memory_buffer
shoumikhin
committed
247caed
Correct three comments in the pybindings memory code
shoumikhin
committed
c51de12
Check the planned buffer size before reading it
shoumikhin
committed
2694176
Correct the load_method docstring on planned memory
shoumikhin
committed
4c938f0
Declare the exir schema dependency in the pybindings tests
shoumikhin
committed
72e317e
Test that a device planned method allocates on the device
shoumikhin
committed
f30dd9d
Run the pybindings device test in the CUDA unit test job
shoumikhin
committed
704e8f7
Use prefix increment in the loops this change adds
shoumikhin
committed
e06b993
Name the right owner of device arenas in the load_method docstring
shoumikhin
committed
2452cd6
Fail the CUDA job if the pybindings device test skips, and document the arena split

r
authored and
shoumikhin
committed
a3f584c
Merge branch 'main' into fix-pybindings-planned-buffer-device
shoumikhin
authored
be35a3d

Commits on Aug 25, 2026