Dieses Handbuch enthält die ersten Schritte für die Entwicklung des XLA-Projekts.
Bevor du beginnst, müssen die folgenden Voraussetzungen erfüllt sein:
- Rufen Sie CONTRIBUTING.md auf und überprüfen Sie den Beitrag. .
- Unterschreiben Sie die Contributor License Agreement, falls noch nicht geschehen.
- Installieren oder konfigurieren Sie die folgenden Abhängigkeiten:
Führen Sie dann die folgenden Schritte aus, um den Quellcode abzurufen, eine Umgebung einzurichten, das Repository zu erstellen und einen Pull-Request zu erstellen.
Code abrufen
- Erstellen Sie einen Fork des XLA-Repositorys.
Klonen Sie die Verzweigung des Repositorys und ersetzen Sie dabei
<USER>
durch Ihren GitHub-Nutzernamen:git clone <a href="https://github.com/">https://github.com/</a><USER>/xla.git
Wechseln Sie in das Verzeichnis
xla
:cd xla
Konfigurieren Sie das Remote-Upstream-Repository:
git remote add upstream <a href="https://github.com/openxla/xla.git">https://github.com/openxla/xla.git</a>
Umgebung einrichten
Installieren Sie Bazel.
Um XLA zu erstellen, muss Bazel installiert sein. Wir empfehlen, Bazel mit Bazelisk zu installieren. Dabei wird automatisch die richtige Bazel-Version für XLA heruntergeladen. Wenn Bazelisk nicht verfügbar ist, können Sie Bazel manuell installieren.
Erstellen Sie ein TensorFlow Docker-Container
Führen Sie den folgenden Befehl aus, um das TensorFlow-Docker-Image für die CPU- und GPU-Erstellung abzurufen: folgenden Befehl:
docker run --name xla -w /xla -it -d --rm -v $PWD:/xla tensorflow/build:latest-python3.9 bash
Build
Für CPU erstellen:
docker exec xla ./configure.py --backend=CPU
docker exec xla bazel build --test_output=all --spawn_strategy=sandboxed //xla/...
Für GPU erstellen:
docker exec xla ./configure.py --backend=CUDA
docker exec xla bazel build --test_output=all --spawn_strategy=sandboxed //xla/...
./configure.py --backend=CUDA
bazel build --test_output=all --spawn_strategy=sandboxed //xla/...
Der erste Build dauert ziemlich lange, da der gesamte Stack erstellt werden muss, einschließlich XLA, MLIR und StableHLO.
Weitere Informationen zum Erstellen von XLA finden Sie unter Aus Quelle erstellen.
Pull-Anfrage erstellen
Wenn Sie bereit sind, die Änderungen zur Überprüfung zu senden, erstellen Sie ein Pull-Anfrage.
Weitere Informationen zur Philosophie der XLA-Codeüberprüfung finden Sie unter Überprüfungsprozess.