Unverified Commit b05fc269 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Fix discrepancy in regenerate.py (#1583)

parent e4ba602e
...@@ -62,9 +62,9 @@ def generate_base_workflow(base_workflow_name, python_version, cu_version, ...@@ -62,9 +62,9 @@ def generate_base_workflow(base_workflow_name, python_version, cu_version,
d["unicode_abi"] = '1' d["unicode_abi"] = '1'
if cu_version == "cu92": if cu_version == "cu92":
d["wheel_docker_image"] = "soumith/manylinux-cuda92" d["wheel_docker_image"] = "pytorch/manylinux-cuda92"
elif cu_version == "cu100": elif cu_version == "cu100":
d["wheel_docker_image"] = "soumith/manylinux-cuda100" d["wheel_docker_image"] = "pytorch/manylinux-cuda100"
if filter_branch is not None: if filter_branch is not None:
d["filters"] = {"branches": {"only": filter_branch}} d["filters"] = {"branches": {"only": filter_branch}}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment