Tài liệu này mô tả cách tạo thành phần XLA.
Nếu bạn không sao chép kho lưu trữ XLA hoặc cài đặt Bazel, vui lòng xem "Bắt đầu" của tài liệu README.
Linux
Định cấu hình
Các bản dựng XLA được định cấu hình bằng tệp .bazelrc trong thư mục gốc của kho lưu trữ
thư mục. Bạn có thể dùng tập lệnh ./configure.py để điều chỉnh các chế độ cài đặt thường dùng.
Nếu bạn cần thay đổi cấu hình, hãy chạy tập lệnh ./configure.py từ
thư mục gốc của kho lưu trữ. Tập lệnh này chứa các cờ cho vị trí của XLA
các phần phụ thuộc và các tuỳ chọn cấu hình bản dựng bổ sung (cờ trình biên dịch, đối với
ví dụ). Tham khảo phần Phiên mẫu để biết thông tin chi tiết.
Hỗ trợ CPU
Bạn nên sử dụng vùng chứa Docker phù hợp để tạo/kiểm thử XLA, chẳng hạn như Vùng chứa Docker của TensorFlow:
docker run --name xla -w /xla -it -d --rm -v $PWD:/xla tensorflow/build:latest-python3.9 bash
Bằng cách sử dụng vùng chứa Docker, bạn có thể xây dựng XLA có hỗ trợ CPU bằng cách sử dụng như sau các lệnh:
docker exec xla ./configure.py --backend=CPU
docker exec xla bazel build //xla/... --spawn_strategy=sandboxed --test_output=all
Nếu muốn xây dựng các mục tiêu XLA có hỗ trợ CPU mà không cần Docker, bạn cần cài đặt clang. XLA hiện dựa trên CI với clang-17, nhưng các phiên bản cũ hơn cũng sẽ hoạt động:
apt install clang
Sau đó, hãy định cấu hình và tạo mục tiêu bằng các lệnh sau:
./configure.py --backend=CPU
bazel build --test_output=all --spawn_strategy=sandboxed //xla/...
Hỗ trợ GPU
Bạn nên sử dụng cùng một vùng chứa Docker như trên để tạo XLA với GPU hỗ trợ:
docker run --name xla_gpu -w /xla -it -d --rm -v $PWD:/xla tensorflow/build:latest-python3.9 bash
Để xây dựng XLA có hỗ trợ GPU, hãy sử dụng lệnh sau:
docker exec xla_gpu ./configure.py --backend=CUDA
docker exec xla_gpu bazel build --test_output=all --spawn_strategy=sandboxed //xla/...
Để biết thêm chi tiết về Bạn có thể xem tài liệu này để xem hình ảnh Docker GPU của TensorFlow.
Bạn cũng có thể tạo các mục tiêu XLA có hỗ trợ GPU mà không cần Docker. Định cấu hình và tạo mục tiêu bằng cách sử dụng các lệnh sau:
./configure.py --backend=CUDA
bazel build --test_output=all --spawn_strategy=sandboxed //xla/...
Để biết thêm chi tiết về để tham khảo tài liệu này, hãy tham khảo tài liệu này.