Az Ultra Accelerator Link, röviden „UALink” egy nyílt szabványú, nagy sebességű hálózati protokoll, amelyet GPU-k és más AI-gyorsítók összekapcsolására terveztek adatközponti, skálázható podokon belül. A UALink akár 1024 gyorsító összefűzését is lehetővé teszi egyetlen domainen belül.

Most küldték be az első javítócsomag-sorozatot, amelynek terjedelme 95 javítócsomag, nagyjából 7 ezer sornyi kód. Ez tartalmazza a UALink alapinfrastruktúráját, valamint az AMDGPU kernel grafikus driver integrációját.

„Egy podon belül az összekapcsolt GPU-k közvetlenül elérik egy másik GPU [távoli] memóriáját UALink-en keresztül. RDMA-val ellentétben itt nincs másolás; közvetlen betöltések és tárolások történnek a fabricen keresztül. A megosztott memóriát csak akkor érheti el egy távoli GPU, ha a memóriát exportálták, és az importálót engedélyezték. A memória megosztásához egységes fizikai címtérre van szükség, amelyet a node-ok megosztanak. Ezt a címtartományt NPA (Network Physical Address) térnek nevezzük. Az NPA címtartományt felosztjuk a GPU-k között, így mindegyik GPU saját szegmenssel rendelkezik, amelyben a memóriáját exportálhatja. Minden GPU dedikált laptáblakészletet tart fenn az NPA teréhez, hasonlóan a GPUVM-hez.

Az exportált memória nincs „pinnelve”. Az exportáló félnek szükség esetén mozgatnia kell a memóriát, vagy elérhetetlenné kell tennie az importálók számára. Ezt távoli TLB-ürítésekkel kezeljük az exportálón, TTM kilakoltatások vagy MMU értesítők esetén, illetve távoli [jelenlét]ellenőrzésekkel és újrapróbálkozással az importálón.

A távoli memória eléréséhez a driver az NPA címeket a folyamatonkénti GPUVM laptáblákba tudja leképezni, ugyanúgy, mint a lokális memóriát. Az alkalmazások átlátszó (opaque) handle-öket használnak a távoli memória reprezentálására. A podon belüli GPU-k közvetlenül kommunikálnak egymással, és így cserélik ki az NPA címeket az importálók és exportálók között. Ha egy node leáll vagy újraindul, a partnerei eltakarítják azokat a [hivatkozásokat], amelyek ilyenkor elvesznek.”

A második javítócsomag-sorozat mindössze három javítócsomagból áll, és az új IOCTL interfészt, valamint a memória importálására és exportálására szolgáló UALink-integráció dokumentációját teszi elérhetővé. Az olvasók számára várhatóan a legérdekesebb rész a dokumentációs javítócsomag.

Örvendetes, hogy az AMD végre nyilvánosan is megmutatta a UALink Linux kernel-integrációján végzett munkát az AMDGPU driverhez.