Questa guida mostra come iniziare a sviluppare il progetto XLA.
Prima di iniziare, completa i seguenti prerequisiti:
- Vai alla pagina Contributi e consulta la procedura di invio.
- Se non l'hai ancora fatto, firma il Contratto di licenza per i collaboratori.
- Installa o configura le seguenti dipendenze:
Poi segui i passaggi riportati di seguito per recuperare il codice sorgente, configurare un ambiente, compilare il repository e creare una richiesta di pull.
Ottieni il codice
- Crea un fork del repository XLA.
Clona il tuo fork del repository sostituendo
<USER>
con il tuo nome utente GitHub:git clone https://github.com/
/xla.git Passa alla directory
xla
:cd xla
Configura il repository upstream remoto:
git remote add upstream https://github.com/openxla/xla.git
Configurare un ambiente
Installa Bazel.
Per compilare XLA, devi avere installato Bazel. Il modo consigliato per installare Bazel è utilizzare Bazelisk, che scarica automaticamente la versione corretta di Bazel per XLA. Se Bazelisk non è disponibile, puoi installare Bazel manualmente.
Crea ed esegui un container Docker TensorFlow.
Per ottenere l'immagine Docker di TensorFlow per la compilazione sia su CPU che su GPU, esegui il seguente comando:
docker run --name xla -w /xla -it -d --rm -v $PWD:/xla tensorflow/build:latest-python3.9 bash
Build
Compila per la CPU:
docker exec xla ./configure.py --backend=CPU
docker exec xla bazel build --test_output=all --spawn_strategy=sandboxed //xla/...
Progettazione per GPU:
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/...
La prima build richiederà un po' di tempo perché deve creare l'intero stack, inclusi XLA, MLIR e StableHLO.
Per scoprire di più sulla compilazione di XLA, consulta Compilare da sorgente.
Creare una richiesta pull
Quando tutto è pronto per inviare le modifiche per la revisione, crea una richiesta di pull.
Per conoscere la filosofia di revisione del codice XLA, consulta la Procedura di revisione.