Preprint
Sep 2026
Torch-PIM: Automated Profile-Guided PIM Offloading for PyTorch
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 and other DL frameworks lack compiler support for making this decision on the code they lower: existing offloading frameworks target hand-wr...
Heeeon Lee, H. Nam, Junyong Heo et al.
· 0 citations