diff options
| author | Ailing Zhang <ailzhang@fb.com> | 2019-01-25 08:35:55 -0800 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-01-25 09:24:45 -0800 |
| commit | f087c65a56f3600af54f4642059897f4f4c6d35d (patch) | |
| tree | 7528d6e3eaf6bf6d0d25b8378d54bab20602f139 /.circleci | |
| parent | 45602ce9a2b3e06cf7356a04f2670b9e623632dd (diff) | |
| download | pytorch-f087c65a56f3600af54f4642059897f4f4c6d35d.tar.gz pytorch-f087c65a56f3600af54f4642059897f4f4c6d35d.tar.bz2 pytorch-f087c65a56f3600af54f4642059897f4f4c6d35d.zip | |
Add xla test in CI (#15978)
Summary:
Adding xla CPU tests in our CI.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15978
Differential Revision: D13816344
Pulled By: ailzhang
fbshipit-source-id: f74c52e846976ea4ac439313847908a0e99d05eb
Diffstat (limited to '.circleci')
| -rw-r--r-- | .circleci/config.yml | 37 |
1 files changed, 34 insertions, 3 deletions
diff --git a/.circleci/config.yml b/.circleci/config.yml index 87bd7ada1c..f00c05a652 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -169,9 +169,15 @@ setup_ci_environment: &setup_ci_environment export MAX_JOBS=$(( ${SCCACHE_MAX_JOBS} > ${MEMORY_LIMIT_MAX_JOBS} ? ${MEMORY_LIMIT_MAX_JOBS} : ${SCCACHE_MAX_JOBS} )) echo "declare -x MAX_JOBS=${MAX_JOBS}" >> /home/circleci/project/env - # This IAM user allows write access to S3 bucket for sccache - echo "declare -x AWS_ACCESS_KEY_ID=${CIRCLECI_AWS_ACCESS_KEY_FOR_SCCACHE_S3_BUCKET_V3}" >> /home/circleci/project/env - echo "declare -x AWS_SECRET_ACCESS_KEY=${CIRCLECI_AWS_SECRET_KEY_FOR_SCCACHE_S3_BUCKET_V3}" >> /home/circleci/project/env + if [[ "${JOB_BASE_NAME}" == *xla* ]]; then + # This IAM user allows write access to S3 bucket for sccache & bazels3cache + 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 + else + # This IAM user allows write access to S3 bucket for sccache + echo "declare -x AWS_ACCESS_KEY_ID=${CIRCLECI_AWS_ACCESS_KEY_FOR_SCCACHE_S3_BUCKET_V3}" >> /home/circleci/project/env + echo "declare -x AWS_SECRET_ACCESS_KEY=${CIRCLECI_AWS_SECRET_KEY_FOR_SCCACHE_S3_BUCKET_V3}" >> /home/circleci/project/env + fi fi # This IAM user only allows read-write access to ECR @@ -994,6 +1000,19 @@ jobs: resource_class: large <<: *pytorch_linux_test_defaults + pytorch_xla_linux_trusty_py3_6_gcc5_4_build: + environment: + JOB_BASE_NAME: pytorch-xla-linux-trusty-py3.6-gcc5.4-build + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-trusty-py3.6-gcc5.4:278" + <<: *pytorch_linux_build_defaults + + pytorch_xla_linux_trusty_py3_6_gcc5_4_test: + environment: + JOB_BASE_NAME: pytorch-xla-linux-trusty-py3.6-gcc5.4-test + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-trusty-py3.6-gcc5.4:278" + resource_class: large + <<: *pytorch_linux_test_defaults + pytorch_linux_trusty_py3_6_gcc7_build: environment: JOB_BASE_NAME: pytorch-linux-trusty-py3.6-gcc7-build @@ -3155,6 +3174,18 @@ workflows: - pytorch_linux_trusty_py3_6_gcc7_test: requires: - pytorch_linux_trusty_py3_6_gcc7_build + - pytorch_xla_linux_trusty_py3_6_gcc5_4_build: + filters: + branches: + only: + - master + - pytorch_xla_linux_trusty_py3_6_gcc5_4_test: + filters: + branches: + only: + - master + requires: + - pytorch_xla_linux_trusty_py3_6_gcc5_4_build - pytorch_linux_trusty_pynightly_build - pytorch_linux_trusty_pynightly_test: requires: |