⚡ AI и ROCm ускорили графовый layout в 80x

Разработчики создали графовый движок с помощью AI и оптимизировали его под AMD GPU.

Движок использует force-directed алгоритм: узлы отталкиваются, рёбра притягиваются как пружины, а границы не дают выйти за рамки. Логика реализована на PyTorch с автоматическим переносом на GPU через ROCm. Ключ — параллельные вычисления: все позиции обновляются одновременно с помощью broadcasting тензоров.

На AMD MI350X достигнуто до 80x ускорения по сравнению с CPU.

#rocm #amd_gpu #ai_codegen #graph_layout #pytorch