OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Fix GPU perf tests on CircleCI (#12491)

Browse Source 
Summary:
`COMMIT_SOURCE` is missing in the current CircleCI config, which is used in perf tests to decide whether to store the new numbers as baseline.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12491

Differential Revision: D10274426

Pulled By: yf225

fbshipit-source-id: 047ef6cc61a12738062f9940d1bfd4c3bf152909
This commit is contained in:
Will Feng 2018-10-09 13:44:38 -07:00 committed by Facebook Github Bot
parent b572e27502
commit 2b22c60980
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.circleci/config.yml
View File

@ -63,6 +63,7 @@ setup_ci_environment: &setup_ci_environment
if [[ "${JOB_BASE_NAME}" == *-build ]]; then
echo "declare -x IN_CIRCLECI=1" > /home/circleci/project/env
echo "declare -x COMMIT_SOURCE=${CIRCLE_BRANCH}" >> /home/circleci/project/env
echo "declare -x PYTHON_VERSION=${PYTHON_VERSION}" >> /home/circleci/project/env
echo "declare -x SCCACHE_BUCKET=ossci-compiler-cache-circleci-v2" >> /home/circleci/project/env
if [ -n "${CUDA_VERSION}" ]; then
@ -95,6 +96,7 @@ pytorch_linux_cpu_build_test_defaults: &pytorch_linux_cpu_build_test_defaults
no_output_timeout: "1h"
command: |
export IN_CIRCLECI=1
export COMMIT_SOURCE=${CIRCLE_BRANCH}
export SCCACHE_BUCKET=ossci-compiler-cache-circleci-v2
export SCCACHE_MAX_JOBS=`expr $(nproc) - 1`
export MEMORY_LIMIT_MAX_JOBS=8 # the "large" resource class on CircleCI has 32 CPU cores, if we use all of them we'll OOM