Torch-PIM: Automated Profile-Guided PIM Offloading for PyTorch

Heeeon Lee, Hyunwoo Nam, Junyong Heo +6

Modern deep learning (DL) workloads are limited by data movement, and processing-in-memory (PIM) targets this bottleneck by placing compute units near the memory. However, PyTorch…