Guida per gli sviluppatori XLA

Questa guida mostra come iniziare a sviluppare il progetto XLA.

Prima di iniziare, completa i seguenti prerequisiti:

  1. Vai alla pagina Contributi e consulta la procedura di invio.
  2. Se non l'hai ancora fatto, firma il Contratto di licenza per i collaboratori.
  3. 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

  1. Crea un fork del repository XLA.

  2. Clona il tuo fork del repository sostituendo <USER> con il tuo nome utente GitHub: 

        
        git clone https://github.com//xla.git

  3. Passa alla directory xla: cd xla

  4. Configura il repository upstream remoto: 

        
        git remote add upstream https://github.com/openxla/xla.git

Configurare un ambiente

  1. 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.

  2. 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.