KI-Hardware und Software-Stack

Welche Aussage zur Tensorparallelität, bei der das Matrixprodukt einer großen Schicht, die nicht in den Speicher einer GPU passt, auf mehrere GPUs aufgeteilt wird, ist richtig?

  1. AJede GPU hält eine Kopie des gesamten Modells, und die Daten werden aufgeteilt
  2. BDie Gewichtsmatrix wird zeilen- oder spaltenweise aufgeteilt und auf die GPUs verteilt; nach der Berechnung von Teilprodukten werden die Ergebnisse per Kommunikation zusammengeführt
  3. CJede Schicht wird auf einer anderen GPU platziert und der Reihe nach verarbeitet
Antwort und Erläuterung anzeigen

Antwort: B. Die Gewichtsmatrix wird zeilen- oder spaltenweise aufgeteilt und auf die GPUs verteilt; nach der Berechnung von Teilprodukten werden die Ergebnisse per Kommunikation zusammengeführt

Bei der Tensorparallelität wird die Gewichtsmatrix z. B. spaltenweise in vier Teile geteilt und auf vier GPUs verteilt; jede GPU berechnet das Produkt der gleichen Eingabe mit den ihr zugewiesenen Spalten. Da am Ende je Schicht eine Kommunikation zum Zusammenfügen oder Aufsummieren der Ergebnisse nötig ist, wird sie üblicherweise bei GPUs innerhalb desselben Servers eingesetzt, die über sehr schnelle Verbindungen gekoppelt sind. In Kombination mit Daten- und Pipeline-Parallelität werden damit riesige Modelle trainiert.

Dieses Thema üben Zur Themenseite

Weitere Fragen zu KI-Hardware und Software-Stack

  1. Warum wird bei der Pipeline-Parallelität, bei der die Schichten des Modells der Reihe nach auf mehrere GPUs aufgeteilt werden, der Batch in kleine Mikrobatches unterteilt und durchgeleitet?
  2. Welche Funktion hat die beim verteilten Training verwendete Kommunikationsoperation All-Reduce?
  3. Welche Aussage zum beim Training verwendeten Gradient Checkpointing (Neuberechnung von Aktivierungen) ist richtig?
  4. Was ist der Hauptgrund dafür, dass Systolic Arrays in KI-Beschleunigern Matrixprodukte effizient berechnen können?
  5. Was ist der Hauptgrund dafür, dass KI-Beschleuniger mitunter großen SRAM (einige zehn bis einige hundert MB) auf dem Chip tragen?