File tree 1 file changed +6
-2
lines changed
.github/actions/test-pytorch-binary
1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change 34
34
-w / \
35
35
"${DOCKER_IMAGE}"
36
36
)
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
+
39
43
docker exec -t -w "${PYTORCH_ROOT}" "${container_name}" bash -c "bash .circleci/scripts/binary_populate_env.sh"
40
44
# Generate test script
41
45
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