Skip to content

Commit 2596686

Browse files
nightly build changes
1 parent 1e75325 commit 2596686

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/actions/test-pytorch-binary/action.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ runs:
3535
)
3636
3737
echo "CONTAINER_NAME=${container_name}" >> "$GITHUB_ENV"
38-
if [[ "${GPU_ARCH_TYPE}" != "rocm" && "${BUILD_ENVIRONMENT}" != "linux-aarch64-binary-manywheel" && "${BUILD_ENVIRONMENT}" != "linux-s390x-binary-manywheel" && "${GPU_ARCH_TYPE}" != "xpu" ]]; then
38+
if [[ "${GPU_ARCH_TYPE}" != "rocm" && "${BUILD_ENVIRONMENT}" != "linux-aarch64-binary-manywheel" && "${BUILD_ENVIRONMENT}" != "linux-s390x-binary-manywheel" && "${BUILD_ENVIRONMENT}" != "linux-ppc64le-binary-manywheel" && "${GPU_ARCH_TYPE}" != "xpu" ]]; then
3939
# Propagate download.pytorch.org IP to container. This is only needed on Linux non aarch64 runner
4040
grep download.pytorch.org /etc/hosts | docker exec -i "${container_name}" bash -c "/bin/cat >> /etc/hosts"
4141
fi
@@ -46,7 +46,7 @@ runs:
4646
docker exec -t "${container_name}" bash -c "source ${BINARY_ENV_FILE} && bash -x /run.sh"
4747
4848
- name: Cleanup docker
49-
if: always() && (env.BUILD_ENVIRONMENT == 'linux-s390x-binary-manywheel' || env.GPU_ARCH_TYPE == 'xpu')
49+
if: always() && (env.BUILD_ENVIRONMENT == 'linux-s390x-binary-manywheel' || "${BUILD_ENVIRONMENT}" != "linux-ppc64le-binary-manywheel" || env.GPU_ARCH_TYPE == 'xpu')
5050
shell: bash
5151
run: |
5252
# on s390x or xpu stop the container for clean worker stop

0 commit comments

Comments
 (0)