diff --git a/.pfnci/config.pbtxt b/.pfnci/config.pbtxt
index dc5c730..adfd0cb 100644
--- a/.pfnci/config.pbtxt
+++ b/.pfnci/config.pbtxt
@@ -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" }
}
}
diff --git a/check_release_assets.py b/check_release_assets.py
index 6551296..2eb735c 100755
--- a/check_release_assets.py
+++ b/check_release_assets.py
@@ -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'
@@ -24,61 +21,39 @@
sdist_project = 'cupy'
-_v13_cuda_matrix = list(itertools.product(
+_main_cuda_x86_matrix = list(itertools.product(
+ (CP310, CP311, CP312), (LINUX, WINDOWS)))
+_main_cuda_aarch64_matrix = list(itertools.product(
+ (CP310, CP311, CP312), (LINUX_AARCH64,)))
+_main_rocm_matrix = list(itertools.product(
+ (CP310, CP311, CP312), (LINUX,)))
+_v13_cuda_x86_matrix = list(itertools.product(
(CP39, CP310, CP311, CP312), (LINUX, WINDOWS)))
-_v13_aarch64_matrix = list(itertools.product(
+_v13_cuda_aarch64_matrix = list(itertools.product(
(CP39, CP310, CP311, CP312), (LINUX_AARCH64,)))
_v13_rocm_matrix = list(itertools.product(
(CP39, CP310, CP311, CP312), (LINUX,)))
-_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,)))
-
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
@@ -149,7 +124,7 @@ def get_expected_wheels(wheel_projects, version):
def parse_args(argv) -> argparse.Namespace:
parser = argparse.ArgumentParser()
- parser.add_argument('--version')
+ parser.add_argument('--version', required=True)
parser.add_argument('--github', action='store_true', default=False)
parser.add_argument('--pypi-sdist', action='store_true', default=False)
parser.add_argument('--pypi-wheel', action='store_true', default=False)
diff --git a/dist_config.py b/dist_config.py
index 95f72a0..3bf0096 100644
--- a/dist_config.py
+++ b/dist_config.py
@@ -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.
@@ -205,8 +217,7 @@
- `cupy-cuda12x `_ (for CUDA 12.x)
- `cupy-cuda11x `_ (for CUDA 11.2 ~ 11.x)
-- `cupy-rocm-5-0 `_ (for ROCm 5.0)
-- `cupy-rocm-4-3 `_ (for ROCm 4.3)
+- `cupy-rocm-6-2 `_ (for ROCm 6.2)
Please see `Installation Guide `_ for the detailed instructions.
''' # NOQA