Skip to content

Commit

Permalink
Revert "Disable ABI on the master branch. (#5641)" (#5649)
Browse files Browse the repository at this point in the history
This reverts commit a0a3aba.
  • Loading branch information
vanbasten23 authored Sep 26, 2023
1 parent 2c6e4a7 commit a1e76b4
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 1 deletion.
1 change: 0 additions & 1 deletion infra/ansible/config/env.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,6 @@ build_env:
BAZEL_REMOTE_CACHE: 1
SILO_NAME: "cache-silo-{{ arch }}-{{ accelerator }}-{{ clang_version }}"
DISABLE_XRT: "{{ disable_xrt }}"
CXX_ABI: 0

amd64:
ARCH: amd64
Expand Down
12 changes: 12 additions & 0 deletions infra/ansible/roles/build_srcs/tasks/main.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,18 @@
name: "{{ pytorch_wheels.files | map(attribute='path') }}"
state: "forcereinstall"

- name: Check if build_torch_xla_libs.sh script exists
stat:
path: "{{ (src_root, 'pytorch/xla/build_torch_xla_libs.sh') | path_join }}"
register: build_torch_xla_libs_result

- name: Build XLA computation client library
ansible.builtin.command:
cmd: bash build_torch_xla_libs.sh -O -D_GLIBCXX_USE_CXX11_ABI=1
chdir: "{{ (src_root, 'pytorch/xla') | path_join }}"
environment: "{{ env_vars }}"
when: build_torch_xla_libs_result.stat.exists

- name: Build PyTorch/XLA
ansible.builtin.command:
cmd: python setup.py bdist_wheel
Expand Down

0 comments on commit a1e76b4

Please sign in to comment.