Skip to content

GitLab

Unverified Commit 87c5e572 authored by KyleCZH's avatar KyleCZH Committed by GitHub
Browse files

add rocm4.3.1 to nightly wheel builds (#4516)

parent b81d189d
Hide whitespace changes
Inline Side-by-side
Showing with 85 additions and 85 deletions
.circleci/config.yml
View file @ 87c5e572
......@@ -1008,16 +1008,16 @@ workflows:
name: binary_linux_wheel_py3.6_cu113
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel:
cu_version: rocm4.1
name: binary_linux_wheel_py3.6_rocm4.1
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-rocm:4.1
- binary_linux_wheel:
cu_version: rocm4.2
name: binary_linux_wheel_py3.6_rocm4.2
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_linux_wheel:
cu_version: rocm4.3.1
name: binary_linux_wheel_py3.6_rocm4.3.1
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1047,16 +1047,16 @@ workflows:
name: binary_linux_wheel_py3.7_cu113
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel:
cu_version: rocm4.1
name: binary_linux_wheel_py3.7_rocm4.1
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.1
- binary_linux_wheel:
cu_version: rocm4.2
name: binary_linux_wheel_py3.7_rocm4.2
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_linux_wheel:
cu_version: rocm4.3.1
name: binary_linux_wheel_py3.7_rocm4.3.1
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1081,16 +1081,16 @@ workflows:
name: binary_linux_wheel_py3.8_cu113
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel:
cu_version: rocm4.1
name: binary_linux_wheel_py3.8_rocm4.1
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.1
- binary_linux_wheel:
cu_version: rocm4.2
name: binary_linux_wheel_py3.8_rocm4.2
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_linux_wheel:
cu_version: rocm4.3.1
name: binary_linux_wheel_py3.8_rocm4.3.1
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1115,16 +1115,16 @@ workflows:
name: binary_linux_wheel_py3.9_cu113
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel:
cu_version: rocm4.1
name: binary_linux_wheel_py3.9_rocm4.1
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.1
- binary_linux_wheel:
cu_version: rocm4.2
name: binary_linux_wheel_py3.9_rocm4.2
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_linux_wheel:
cu_version: rocm4.3.1
name: binary_linux_wheel_py3.9_rocm4.3.1
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1887,15 +1887,15 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.6_cu113_upload
- binary_linux_wheel:
cu_version: rocm4.1
cu_version: rocm4.2
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_rocm4.1
name: nightly_binary_linux_wheel_py3.6_rocm4.2
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-rocm:4.1
wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -1903,29 +1903,29 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_rocm4.1_upload
name: nightly_binary_linux_wheel_py3.6_rocm4.2_upload
requires:
- nightly_binary_linux_wheel_py3.6_rocm4.1
subfolder: rocm4.1/
- nightly_binary_linux_wheel_py3.6_rocm4.2
subfolder: rocm4.2/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.6_rocm4.1_smoke_test_pip
name: nightly_binary_linux_wheel_py3.6_rocm4.2_smoke_test_pip
python_version: '3.6'
requires:
- nightly_binary_linux_wheel_py3.6_rocm4.1_upload
- nightly_binary_linux_wheel_py3.6_rocm4.2_upload
- binary_linux_wheel:
cu_version: rocm4.2
cu_version: rocm4.3.1
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_rocm4.2
name: nightly_binary_linux_wheel_py3.6_rocm4.3.1
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-rocm:4.2
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -1933,19 +1933,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_rocm4.2_upload
name: nightly_binary_linux_wheel_py3.6_rocm4.3.1_upload
requires:
- nightly_binary_linux_wheel_py3.6_rocm4.2
subfolder: rocm4.2/
- nightly_binary_linux_wheel_py3.6_rocm4.3.1
subfolder: rocm4.3.1/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.6_rocm4.2_smoke_test_pip
name: nightly_binary_linux_wheel_py3.6_rocm4.3.1_smoke_test_pip
python_version: '3.6'
requires:
- nightly_binary_linux_wheel_py3.6_rocm4.2_upload
- nightly_binary_linux_wheel_py3.6_rocm4.3.1_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2071,15 +2071,15 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.7_cu113_upload
- binary_linux_wheel:
cu_version: rocm4.1
cu_version: rocm4.2
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.1
name: nightly_binary_linux_wheel_py3.7_rocm4.2
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.1
wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2087,29 +2087,29 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.1_upload
name: nightly_binary_linux_wheel_py3.7_rocm4.2_upload
requires:
- nightly_binary_linux_wheel_py3.7_rocm4.1
subfolder: rocm4.1/
- nightly_binary_linux_wheel_py3.7_rocm4.2
subfolder: rocm4.2/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.7_rocm4.1_smoke_test_pip
name: nightly_binary_linux_wheel_py3.7_rocm4.2_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_rocm4.1_upload
- nightly_binary_linux_wheel_py3.7_rocm4.2_upload
- binary_linux_wheel:
cu_version: rocm4.2
cu_version: rocm4.3.1
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.2
name: nightly_binary_linux_wheel_py3.7_rocm4.3.1
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.2
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2117,19 +2117,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.2_upload
name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
requires:
- nightly_binary_linux_wheel_py3.7_rocm4.2
subfolder: rocm4.2/
- nightly_binary_linux_wheel_py3.7_rocm4.3.1
subfolder: rocm4.3.1/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.7_rocm4.2_smoke_test_pip
name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_rocm4.2_upload
- nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2255,15 +2255,15 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.8_cu113_upload
- binary_linux_wheel:
cu_version: rocm4.1
cu_version: rocm4.2
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.1
name: nightly_binary_linux_wheel_py3.8_rocm4.2
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.1
wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2271,29 +2271,29 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.1_upload
name: nightly_binary_linux_wheel_py3.8_rocm4.2_upload
requires:
- nightly_binary_linux_wheel_py3.8_rocm4.1
subfolder: rocm4.1/
- nightly_binary_linux_wheel_py3.8_rocm4.2
subfolder: rocm4.2/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.8_rocm4.1_smoke_test_pip
name: nightly_binary_linux_wheel_py3.8_rocm4.2_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_linux_wheel_py3.8_rocm4.1_upload
- nightly_binary_linux_wheel_py3.8_rocm4.2_upload
- binary_linux_wheel:
cu_version: rocm4.2
cu_version: rocm4.3.1
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.2
name: nightly_binary_linux_wheel_py3.8_rocm4.3.1
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.2
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2301,19 +2301,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.2_upload
name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
requires:
- nightly_binary_linux_wheel_py3.8_rocm4.2
subfolder: rocm4.2/
- nightly_binary_linux_wheel_py3.8_rocm4.3.1
subfolder: rocm4.3.1/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.8_rocm4.2_smoke_test_pip
name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_linux_wheel_py3.8_rocm4.2_upload
- nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2439,15 +2439,15 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.9_cu113_upload
- binary_linux_wheel:
cu_version: rocm4.1
cu_version: rocm4.2
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.1
name: nightly_binary_linux_wheel_py3.9_rocm4.2
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.1
wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2455,29 +2455,29 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.1_upload
name: nightly_binary_linux_wheel_py3.9_rocm4.2_upload
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.1
subfolder: rocm4.1/
- nightly_binary_linux_wheel_py3.9_rocm4.2
subfolder: rocm4.2/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.9_rocm4.1_smoke_test_pip
name: nightly_binary_linux_wheel_py3.9_rocm4.2_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.1_upload
- nightly_binary_linux_wheel_py3.9_rocm4.2_upload
- binary_linux_wheel:
cu_version: rocm4.2
cu_version: rocm4.3.1
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.2
name: nightly_binary_linux_wheel_py3.9_rocm4.3.1
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.2
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2485,19 +2485,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.2_upload
name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.2
subfolder: rocm4.2/
- nightly_binary_linux_wheel_py3.9_rocm4.3.1
subfolder: rocm4.3.1/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.9_rocm4.2_smoke_test_pip
name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.2_upload
- nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
- binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......
.circleci/regenerate.py
View file @ 87c5e572
......@@ -32,7 +32,7 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,
for os_type in ["linux", "macos", "win"]:
python_versions = PYTHON_VERSIONS
cu_versions_dict = {
"linux": ["cpu", "cu102", "cu111", "cu113", "rocm4.1", "rocm4.2"],
"linux": ["cpu", "cu102", "cu111", "cu113", "rocm4.2", "rocm4.3.1"],
"win": ["cpu", "cu102", "cu111", "cu113"],
"macos": ["cpu"],
}
......
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