En esta guía, se muestra cómo comenzar a desarrollar el proyecto de XLA.
Antes de comenzar, completa los siguientes requisitos previos:
- Ve a CONTRIBUTING.md y revisa el proceso de contribución.
- Si aún no lo hiciste, firma el Contrato de Licencia para Colaboradores.
- Instala o configura las siguientes dependencias:
Luego, sigue los pasos a continuación para obtener el código fuente, configurar un entorno, compilar el repositorio y crear una solicitud de extracción.
Cómo obtener el código
- Crea una bifurcación del repositorio de XLA.
Clona la bifurcación del repositorio y reemplaza
<USER>
por tu nombre de usuario de GitHub:git clone <a href="https://github.com/">https://github.com/</a><USER>/xla.git
Cambia al directorio
xla
:cd xla
Configura el repositorio upstream remoto:
git remote add upstream <a href="https://github.com/openxla/xla.git">https://github.com/openxla/xla.git</a>
Configura un entorno
Instala Bazel.
Para compilar XLA, debes tener instalado Bazel. La forma recomendada de instalar Bazel es usar Bazelisk, que descarga automáticamente la versión correcta de Bazel para XLA. Si Bazelisk no está disponible, puedes instalar Bazel de forma manual.
Crea y ejecuta un contenedor de Docker de TensorFlow.
A fin de obtener la imagen de Docker de TensorFlow para CPU, ejecuta el siguiente comando:
docker run --name xla -w /xla -it -d --rm -v $PWD:/xla tensorflow/build:latest-python3.9 bash
Como alternativa, a fin de obtener la imagen de Docker de TensorFlow para GPU, ejecuta el siguiente comando:
docker run --name xla_gpu -w /xla -it -d --rm -v $PWD:/xla tensorflow/tensorflow:devel-gpu bash
Compila
Compilación para CPU:
docker exec xla ./configure
docker exec xla bazel build --test_output=all --spawn_strategy=sandboxed //xla/...
Compilación para GPU:
docker exec -e TF_NEED_CUDA=1 xla_gpu ./configure
docker exec xla_gpu bazel build --test_output=all --spawn_strategy=sandboxed //xla/...
Tu primera compilación tardará bastante tiempo porque debe compilar toda la pila, incluidos XLA, MLIR y StableHLO.
Si deseas obtener más información para compilar XLA, consulta Cómo compilar desde el código fuente.
Crea una solicitud de extracción
Cuando estés listo para enviar los cambios a revisión, crea una solicitud de extracción.
Para obtener más información sobre la filosofía de revisión de código de XLA, consulta Revisiones de código.