From 21bffd4c460811b4664c284d88ac9835a7526c3b Mon Sep 17 00:00:00 2001 From: Will Cromar Date: Mon, 22 Jul 2024 14:00:09 -0700 Subject: [PATCH 1/2] Build CUDA plugin before `torch` and `torch_xla` in nightly build --- infra/ansible/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/infra/ansible/Dockerfile b/infra/ansible/Dockerfile index 369548413725..2a9f6f5f4e2a 100644 --- a/infra/ansible/Dockerfile +++ b/infra/ansible/Dockerfile @@ -8,8 +8,8 @@ RUN pip install ansible COPY . /ansible ARG ansible_vars -RUN ansible-playbook -vvv playbook.yaml -e "stage=build" -e "${ansible_vars}" -RUN ansible-playbook -vvv playbook.yaml -e "stage=build_plugin" -e "${ansible_vars}" --skip-tags=fetch_srcs,install_deps +RUN ansible-playbook -vvv playbook.yaml -e "stage=build_plugin" -e "${ansible_vars}" +RUN ansible-playbook -vvv playbook.yaml -e "stage=build" -e "${ansible_vars}" --skip-tags=fetch_srcs,install_deps FROM python:${python_version}-${debian_version} AS release From c5cd27dba77aa27284251e3c0ecf0b79acadf59f Mon Sep 17 00:00:00 2001 From: Will Cromar Date: Mon, 22 Jul 2024 14:10:24 -0700 Subject: [PATCH 2/2] install deps? --- infra/ansible/Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/infra/ansible/Dockerfile b/infra/ansible/Dockerfile index 2a9f6f5f4e2a..9762406f9e72 100644 --- a/infra/ansible/Dockerfile +++ b/infra/ansible/Dockerfile @@ -8,6 +8,8 @@ RUN pip install ansible COPY . /ansible ARG ansible_vars +# HACK: install build dependencies only, but skip build step +RUN ansible-playbook -vvv playbook.yaml -e "stage=build" -e "${ansible_vars}" --tags "bazel,configure_env,install_deps" RUN ansible-playbook -vvv playbook.yaml -e "stage=build_plugin" -e "${ansible_vars}" RUN ansible-playbook -vvv playbook.yaml -e "stage=build" -e "${ansible_vars}" --skip-tags=fetch_srcs,install_deps