Skip to content

Commit 3873c1c

Browse files
committed
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]
2 parents 4fa461f + 50d6d05 commit 3873c1c

0 file changed

File tree

    0 commit comments

    Comments
     (0)