File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
.github/actions/test-pytorch-binary Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change 3434 -w / \
3535 "${DOCKER_IMAGE}"
3636 )
37- # Propagate download.pytorch.org IP to container
38- grep download.pytorch.org /etc/hosts | docker exec -i "${container_name}" bash -c "/bin/cat >> /etc/hosts"
37+
38+ if [[ "${GPU_ARCH_TYPE}" != "rocm" ]]; then
39+ # Propagate download.pytorch.org IP to container. This is only needed on Linux runner
40+ grep download.pytorch.org /etc/hosts | docker exec -i "${container_name}" bash -c "/bin/cat >> /etc/hosts"
41+ fi
42+
3943 docker exec -t -w "${PYTORCH_ROOT}" "${container_name}" bash -c "bash .circleci/scripts/binary_populate_env.sh"
4044 # Generate test script
4145 docker exec -t -w "${PYTORCH_ROOT}" -e OUTPUT_SCRIPT="/run.sh" "${container_name}" bash -c "bash .circleci/scripts/binary_linux_test.sh"
You can’t perform that action at this time.
0 commit comments