Improving Oversubscribed GPU Memory Performance in the PyTorch Framework
Improving Oversubscribed GPU Memory Performance in the PyTorch Framework
Jake Choi,H. Yeom,Yoonhee Kim
2022 · DOI: 10.1007/s10586-022-03805-x
Cluster Computing · 1 Citations
TLDR
This paper builds upon the UM implementation and creates and utilize a minimal overhead CUPTI dynamic profiler to trace unified memory page fault and memory transfer statistics in PyTorch applications and implements CUDA memory prefetch and advise API.
