Scheduling Data to Compute: Hierarchical Data Pipelining for LLM Training on AMD GPUs
Read the original on arXiv AI →The paper presents a hierarchical data pipelining approach for large language model (LLM) training on AMD GPUs, focusing on reducing idle compute time caused by data unavailability. It leverages three key hardware features—Tensor Data Mover (TDM), transpose‑load instructions, and System/Remote DMA (SDMA/RDMA)—to move and rearrange tensor data efficiently between memory and compute units, and between GPUs. Kernel‑level experiments show average speedups of 1.88× for TDM, 1.19× for transpose‑load, and 1.16×–1.33× for overlapped communication, and the implementation is integrated into open‑source AMD frameworks.
Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.