KI-Hardware und Software-Stack
Warum benötigen elementweise Operationen wie ReLU oder Layer-Normalisierung in KI-Berechnungen im Verhältnis zur Zahl ihrer Operationen leicht viel Zeit?
- AWeil die ReLU-Berechnung mehr Operationen erfordert als ein Matrixprodukt
- BWeil im Verhältnis zur gelesenen und geschriebenen Datenmenge je Element wenig gerechnet wird und die Geschwindigkeit daher von der Speicherbandbreite bestimmt wird
- CWeil sich elementweise Operationen auf der GPU nicht ausführen lassen
Antwort und Erläuterung anzeigen
Antwort: B. Weil im Verhältnis zur gelesenen und geschriebenen Datenmenge je Element wenig gerechnet wird und die Geschwindigkeit daher von der Speicherbandbreite bestimmt wird
ReLU liest jedes Element, vergleicht es mit 0 und schreibt es zurück; der Rechenaufwand je Element ist gering. Die arithmetische Intensität ist sehr niedrig, sodass die Zeit nicht von der Leistung der Recheneinheiten, sondern von der Speicherbandbreite bestimmt wird. Auch wenn solche Operationen nur einen kleinen Teil der Operationszahl ausmachen, können sie einen nicht vernachlässigbaren Anteil der Zeit beanspruchen. Sie werden direkt an ein Matrixprodukt fusioniert oder die Normalisierung wird in einem Kernel zusammengefasst, um Speicherzugriffe zu verringern.