Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions .circleci/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -36,9 +36,10 @@ pip install lark-parser
# Install Pytorch
xla/scripts/apply_patches.sh
python setup.py build develop
sccache --show-stats

# Bazel doesn't work with sccache gcc. https://github.com/bazelbuild/bazel/issues/3642
sudo add-apt-repository "deb http://apt.llvm.org/trusty/ llvm-toolchain-trusty-7 main"
sudo add-apt-repository "deb http://apt.llvm.org/xenial/ llvm-toolchain-xenial-7 main"
wget -O - https://apt.llvm.org/llvm-snapshot.gpg.key|sudo apt-key add -
sudo apt-get -qq update

Expand Down Expand Up @@ -81,7 +82,7 @@ pushd vision
pip install -q --user .
popd

bazels3cache --bucket=${XLA_CACHE_S3_BUCKET_NAME} --maxEntrySizeBytes=0 --logging.level=verbose
bazels3cache --bucket=${XLA_CLANG_CACHE_S3_BUCKET_NAME} --maxEntrySizeBytes=0 --logging.level=verbose

# install XLA
pushd "$XLA_DIR"
Expand Down
8 changes: 4 additions & 4 deletions .circleci/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ linux_default: &linux_default
echo "declare -x SCCACHE_BUCKET=${SCCACHE_BUCKET}" >> /home/circleci/project/env
echo "declare -x AWS_ACCESS_KEY_ID=${CIRCLECI_AWS_ACCESS_KEY_FOR_SCCACHE_AND_XLA_BAZEL_S3_BUCKET_V1}" >> /home/circleci/project/env
echo "declare -x AWS_SECRET_ACCESS_KEY=${CIRCLECI_AWS_SECRET_KEY_FOR_SCCACHE_AND_XLA_BAZEL_S3_BUCKET_V1}" >> /home/circleci/project/env
echo "declare -x XLA_CACHE_S3_BUCKET_NAME=${XLA_CACHE_S3_BUCKET_NAME}" >> /home/circleci/project/env
echo "declare -x XLA_CLANG_CACHE_S3_BUCKET_NAME=${XLA_CLANG_CACHE_S3_BUCKET_NAME}" >> /home/circleci/project/env
echo "declare -x CIRCLE_JOB=${CIRCLE_JOB}" >> /home/circleci/project/env
echo "declare -x MAX_JOBS=8" >> /home/circleci/project/env
echo "declare -x CC=clang-7 CXX=clang++-7" >> /home/circleci/project/xla_env
Expand All @@ -53,13 +53,13 @@ linux_default: &linux_default
docker exec -u jenkins $(cat .docker_pid) .circleci/test.sh

jobs:
pytorch_xla_linux_trusty_py3_6_gcc5_4:
pytorch_xla_linux_xenial_py3_6_clang_7:
environment:
DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-trusty-py3.6-gcc5.4:278"
DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-py3.6-clang7:336"
<<: *linux_default

workflows:
version: 2
build:
jobs:
- pytorch_xla_linux_trusty_py3_6_gcc5_4
- pytorch_xla_linux_xenial_py3_6_clang_7