fix #36801 (#41607)

Summary: unittest actually did stdout testname like (test_accumulate_grad (__main__.TestAutograd) ... ) first befor test start running. export PYTHONUNBUFFERED=1 or python -u could record this msg. ezyang Pull Request resolved: pytorch/pytorch#41607 Reviewed By: izdeby Differential Revision: D22673930 Pulled By: ezyang fbshipit-source-id: 18512b6f5f80485c2b0d812f2ebdecc1fdc4b4ec
TonyTangYu · Jul 23, 2020 · c0e3839 · c0e3839
1 parent 272fb36
commit c0e3839
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -543,6 +543,7 @@ jobs:
         no_output_timeout: "90m"
         command: |
           set -e
+          export PYTHONUNBUFFERED=1
           # See Note [Special build images]
           output_image=${DOCKER_IMAGE}-${CIRCLE_SHA1}
           if [[ ${BUILD_ENVIRONMENT} == *"xla"* ]]; then

diff --git a/.circleci/verbatim-sources/job-specs/pytorch-job-specs.yml b/.circleci/verbatim-sources/job-specs/pytorch-job-specs.yml
@@ -81,6 +81,7 @@ jobs:
         no_output_timeout: "90m"
         command: |
           set -e
+          export PYTHONUNBUFFERED=1
           # See Note [Special build images]
           output_image=${DOCKER_IMAGE}-${CIRCLE_SHA1}
           if [[ ${BUILD_ENVIRONMENT} == *"xla"* ]]; then