Skip to content

Memory Management

Sherlock edited this page Mar 13, 2021 · 2 revisions

BFCArena

  • Why we need an arena? What problem does it solve?

Memory Planning

Memory Pattern

  • How does ORT come up with a peak memory consumption?

External Torch's CUDACachingAllocator

  • How does ORTModule uses pytorch allocator? (ortmoduel.py)
  • Advance: What's the difference between BFCArena and CUDACachingAllocator?
Clone this wiki locally