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..01f7899 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,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,))) - -_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 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