XLA 開發人員指南

本指南說明如何開始開發 XLA 專案。

開始之前，請先完成下列必備條件：

  1. 前往「「Ctributing」頁面並查看捐款程序。
  2. 如果您尚未簽署，請簽署貢獻者授權協議。
  3. 安裝或設定下列依附元件：

接著，按照下列步驟取得原始碼、設定環境、建構存放區，以及建立提取要求。

取得程式碼

  1. 建立 XLA 存放區的分支。

  2. 複製分支的存放區，將 <USER> 替換為您的 GitHub 使用者名稱：

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

  3. 變更為 xla 目錄：cd xla

  4. 設定遠端上游存放區：

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

設定環境

  1. 安裝 Bazel。

    如要建構 XLA，您必須安裝 Bazel。安裝 Bazel 是建議的做法是使用 Bazelisk，這樣就能自動下載 XLA 適用的 Bazel 版本。如果無法取得 Bazelisk，您可以手動安裝 Bazel。

  2. 建立並執行 TensorFlow Docker 容器。

    如要取得用於建構 CPU 和 GPU 的 TensorFlow Docker 映像檔，請執行下列指令：

    docker run --name xla -w /xla -it -d --rm -v $PWD:/xla tensorflow/build:latest-python3.9 bash

建構

針對 CPU 建構：

docker exec xla ./configure.py --backend=CPU
docker exec xla bazel build --test_output=all --spawn_strategy=sandboxed //xla/...

為 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/...

由於第一次建構作業必須建構整個堆疊 (包括 XLA、MLR 和 StableHLO)，因此會耗時一段時間。

如要進一步瞭解如何建構 XLA，請參閱「從來源建構」。

建立提取要求

準備好要將變更送審時，請建立提取要求。

如要瞭解 XLA 程式碼審查原則，請參閱「審查程序」。