add ansible context

Author: tengyifei

infra/ansible/build_for_torchprime.Dockerfile

@@ -1,3 +1,4 @@
+# syntax=docker/dockerfile:1.4
 ARG python_version=3.10
 ARG debian_version=bullseye
 
@@ -12,7 +13,7 @@ RUN pip install *.whl
 # Install the dependencies including libtpu.
 WORKDIR /ansible
 RUN pip install ansible
-COPY . /ansible
+COPY --from=ansible . /ansible
 
 ARG ansible_vars
 RUN ansible-playbook -vvv playbook.yaml -e "stage=release" -e "${ansible_vars}" --tags "install_deps"

infra/ansible/build_for_torchprime.sh

@@ -4,7 +4,13 @@ set -ex
 
 export IMAGE_NAME="gcr.io/tpu-pytorch/for-torchprime-ci:test"
 export DOCKERFILE_PATH="infra/ansible/build_for_torchprime.Dockerfile"
-export CONTEXT_PATH="/tmp/wheels"
+export DEFAULT_CONTEXT_PATH="/tmp/wheels"
 
-docker build -t "${IMAGE_NAME}" "${CONTEXT_PATH}" -f "${DOCKERFILE_PATH}" --build-arg ansible_vars='{"arch":"amd64","accelerator":"tpu","pytorch_git_rev":"main","xla_git_rev":"foobar","bundle_libtpu":"0","package_version":"2.8","nightly_release":true}' --build-arg python_version=3.10 --build-arg debian_version=bullseye
+docker build -t "${IMAGE_NAME}" \
+    --build-context ansible=infra/ansible \
+    "${DEFAULT_CONTEXT_PATH}" \
+    -f "${DOCKERFILE_PATH}" \
+    --build-arg ansible_vars='{"arch":"amd64","accelerator":"tpu","pytorch_git_rev":"main","xla_git_rev":"foobar","bundle_libtpu":"0","package_version":"2.8","nightly_release":true}' \
+    --build-arg python_version=3.10 \
+    --build-arg debian_version=bullseye
 docker push "${IMAGE_NAME}"