UPDF AI

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.