Add XLA_SANDBOX_BUILD=1 to experimental build (#4455)

* Add `XLA_SANDBOX_BUILD=1` to experimental build

* Update build_torch_xla_libs.sh

* Increase shm size in docker build

Author: will-cromar

build_torch_xla_libs.sh

@@ -36,7 +36,7 @@ fi
 
 BUILD_STRATEGY="standalone"
 if [[ "$XLA_SANDBOX_BUILD" == "1" ]]; then
-  BUILD_STRATEGY="sandboxed --sandbox_tmpfs_path=/tmp"
+  BUILD_STRATEGY="sandboxed --sandbox_base=/dev/shm"
 else
   # We can remove this after https://github.com/bazelbuild/bazel/issues/15359 is resolved
   unset CC

docker/experimental/Dockerfile

@@ -75,6 +75,9 @@ RUN cd third_party/tensorflow && find ../../tf_patches -name '*.diff' | xargs -t
 
 COPY build_torch_xla_libs.sh .
 
+# TODO: Remove this when it's not required anymore
+ENV XLA_SANDBOX_BUILD=1
+
 ARG tpuvm
 ARG tf_cuda_compute_capabilities
 ARG bazel_jobs

docker/experimental/cloudbuild.yaml

@@ -18,6 +18,7 @@ steps:
 
     args=(${_BUILD_ARGS//,/ })
     flags=(
+      --shm-size=${_SHM_SIZE}
       --build-arg=bazel_jobs=${_BAZEL_JOBS}
       --build-arg=python_version=${_PYTHON_VERSION}
       # TODO: use current version in setup.py
@@ -125,6 +126,7 @@ substitutions:
   _CACHE: 'true'
   _CACHE_TTL: '18h'
   _BUILD_ARGS: tpuvm=1,cuda=0
+  _SHM_SIZE: '16gb'
 options:
   # To run in a test project, either point to your pool or replace this with
   # `machineType`. You may have to reduce _BAZEL_JOBS.