Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Support ROCm 6.2 #396

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
Open

Support ROCm 6.2 #396

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
9eff8f3
add rocm-6.2 build variant
kmaehashi Sep 25, 2024
fdd1887
refresh list of packages on PyPI and GitHub Releases
kmaehashi Sep 25, 2024
8ac658e
fix CI to build rocm-6.2
kmaehashi Sep 25, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .pfnci/config.pbtxt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ configs {
time_limit {
seconds: 10800
}
command: ".pfnci/wheel-linux/main.sh rocm-5.0 3.9"
command: ".pfnci/wheel-linux/main.sh rocm-6.2 3.9"
environment_variables { key: "CUPY_RELEASE_SKIP_VERIFY" value: "1" }
}
}
Expand Down
60 changes: 16 additions & 44 deletions check_release_assets.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,6 @@
import distlib.locators


CP36 = 'cp36-cp36m'
CP37 = 'cp37-cp37m'
CP38 = 'cp38-cp38'
CP39 = 'cp39-cp39'
CP310 = 'cp310-cp310'
CP311 = 'cp311-cp311'
Expand All @@ -24,61 +21,36 @@

sdist_project = 'cupy'

_v13_cuda_matrix = list(itertools.product(
_main_cuda_x86_matrix = list(itertools.product(
(CP39, CP310, CP311, CP312), (LINUX, WINDOWS)))
_v13_aarch64_matrix = list(itertools.product(
_main_cuda_aarch64_matrix = list(itertools.product(
(CP39, CP310, CP311, CP312), (LINUX_AARCH64,)))
_v13_rocm_matrix = list(itertools.product(
_main_rocm_matrix = list(itertools.product(
(CP39, CP310, CP311, CP312), (LINUX,)))
Comment on lines +24 to 29
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Support for Python 3.9 was dropped in cupy/cupy#8491


_v12_cuda_matrix = list(itertools.product(
(CP38, CP39, CP310, CP311, CP312), (LINUX, WINDOWS)))
_v12_aarch64_matrix = list(itertools.product(
(CP38, CP39, CP310, CP311, CP312), (LINUX_AARCH64,)))
_v12_rocm_matrix = list(itertools.product(
(CP38, CP39, CP310, CP311, CP312), (LINUX,)))
_v13_cuda_x86_matrix = _main_cuda_x86_matrix
_v13_cuda_aarch64_matrix = _main_cuda_aarch64_matrix
_v13_rocm_matrix = _main_rocm_matrix

pypi_wheel_projects = {
# v13.x
'13': [
('cupy-cuda11x', _v13_cuda_matrix + _v13_aarch64_matrix),
('cupy-cuda12x', _v13_cuda_matrix + _v13_aarch64_matrix),
('cupy-rocm-4-3', _v13_rocm_matrix),
('cupy-rocm-5-0', _v13_rocm_matrix),
# v14.x
'14': [
('cupy-cuda11x', _main_cuda_x86_matrix + _main_cuda_aarch64_matrix),
('cupy-cuda12x', _main_cuda_x86_matrix + _main_cuda_aarch64_matrix),
('cupy-rocm-6-2', _main_rocm_matrix),
],

# v12.x
'12': [
('cupy-cuda102', _v12_cuda_matrix + _v12_aarch64_matrix),
('cupy-cuda110', _v12_cuda_matrix),
('cupy-cuda111', _v12_cuda_matrix),
('cupy-cuda11x', _v12_cuda_matrix + _v12_aarch64_matrix),
('cupy-cuda12x', _v12_cuda_matrix + _v12_aarch64_matrix),
('cupy-rocm-4-3', _v12_rocm_matrix),
('cupy-rocm-5-0', _v12_rocm_matrix),
],
}

github_wheel_projects = {
# v13.x
'13': [
('cupy-cuda11x', _v13_cuda_matrix + _v13_aarch64_matrix),
('cupy-cuda12x', _v13_cuda_matrix + _v13_aarch64_matrix),
('cupy-cuda11x', _v13_cuda_x86_matrix + _v13_cuda_aarch64_matrix),
('cupy-cuda12x', _v13_cuda_x86_matrix + _v13_cuda_aarch64_matrix),
('cupy-rocm-4-3', _v13_rocm_matrix),
('cupy-rocm-5-0', _v13_rocm_matrix),
],
# v12.x
'12': [
('cupy-cuda102', _v12_cuda_matrix + _v12_aarch64_matrix),
('cupy-cuda110', _v12_cuda_matrix),
('cupy-cuda111', _v12_cuda_matrix),
('cupy-cuda11x', _v12_cuda_matrix + _v12_aarch64_matrix),
('cupy-cuda12x', _v12_cuda_matrix + _v12_aarch64_matrix),
('cupy-rocm-4-3', _v12_rocm_matrix),
('cupy-rocm-5-0', _v12_rocm_matrix),
('cupy-rocm-6-2', _v13_rocm_matrix),
],
}

github_wheel_projects = pypi_wheel_projects


def get_basenames(project, version):
# List all wheels including unsupported ones by the current Python
Expand Down
15 changes: 13 additions & 2 deletions dist_config.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -155,6 +155,18 @@
'verify_systems': ['default'],
'system_packages': 'rocm-hip-sdk hip-runtime-amd roctracer-dev' # NOQA
},
'rocm-6.2': {
'name': 'cupy-rocm-6-2',
'kind': 'rocm',
'platform_version': '6.2',
'image': 'rocm/dev-centos-7:6.2',
'libs': [],
'includes': [],
'preloads': [],
'verify_image': 'rocm/rocm-terminal:6.2',
'verify_systems': ['default'],
'system_packages': 'rocm-hip-sdk hip-runtime-amd roctracer-dev' # NOQA
},
}

# Key-value of CUDA version and its corresponding build settings for Windows.
Expand Down Expand Up @@ -205,8 +217,7 @@
- `cupy-cuda12x <https://pypi.org/project/cupy-cuda12x/>`_ (for CUDA 12.x)
- `cupy-cuda11x <https://pypi.org/project/cupy-cuda11x/>`_ (for CUDA 11.2 ~ 11.x)

- `cupy-rocm-5-0 <https://pypi.org/project/cupy-rocm-5-0/>`_ (for ROCm 5.0)
- `cupy-rocm-4-3 <https://pypi.org/project/cupy-rocm-4-3/>`_ (for ROCm 4.3)
- `cupy-rocm-6-2 <https://pypi.org/project/cupy-rocm-6-2/>`_ (for ROCm 6.2)

Please see `Installation Guide <https://docs.cupy.dev/en/latest/install.html>`_ for the detailed instructions.
''' # NOQA
Expand Down
Loading