Skip to content

Commit 50d6d05

Browse files
committed
Update base for Update on "[ET Device Support] Device-aware memory planning: separate buffers per device type"
Extends memory planning to separate device tensors from CPU tensors into distinct memory buffers. Non-CPU TensorSpecs (e.g., CUDA) are pre-assigned device-specific mem_ids before the greedy/naive algorithm runs, ensuring they get planned into independent memory buffers that never share space with CPU tensors. Differential Revision: [D97447105](https://our.internmc.facebook.com/intern/diff/D97447105/) [ghstack-poisoned]
1 parent 858d660 commit 50d6d05

0 file changed

File tree

    0 commit comments

    Comments
     (0)