[CuPy] [package list]

CuPy/13.5.1-cpeGNU-25.03-rocm (CuPy-13.5.1-cpeGNU-25.03-rocm.eb)

Install with the EasyBuild-user module:

eb CuPy-13.5.1-cpeGNU-25.03-rocm.eb -r

To access module help after installation and get reminded for which stacks and partitions the module is installed, use module spider CuPy/13.5.1-cpeGNU-25.03-rocm.

EasyConfig:

#DOC CuPy to be used with Cray Python.
easyblock = 'PythonBundle'

local_CuPy_version      = '13.5.1'
local_cython_version    = '3.1.2'
local_fastrlock_version = '0.8.3'

name = 'CuPy'
version = local_CuPy_version
versionsuffix = '-rocm'

homepage = 'https://cupy.dev'

whatis = [
    'Description: CuPy is an open-source array library for GPU-accelerated computing with Python'
]

description = """"
CuPy is an open-source array library for GPU-accelerated computing with Python.
CuPy acts as a drop-in replacement to run existing NumPy/SciPy code on NVIDIA CUDA
or AMD ROCm platforms.
"""

toolchain = {'name': 'cpeGNU', 'version': '25.03'}

builddependencies = [
    # Provides wheel
    ('buildtools-python', '%(toolchain_version)s', '-cray-python%(pyshortver)s', True),
    # CuPy needs Cython>3
    ('Cython', local_cython_version),
]

dependencies = [
    ('cray-python', EXTERNAL_MODULE),
    ('rocm',        EXTERNAL_MODULE),
]

use_pip = True

exts_default_options = {
    'source_urls': [PYPI_LOWER_SOURCE],
    'use_pip': True
}

exts_list = [
    ('fastrlock', local_fastrlock_version),
    ('cupy', version, {
        'preinstallopts': 'ROCM_HOME=$ROCM_PATH CUPY_INSTALL_USE_HIP=1 HCC_AMDGPU_TARGET=gfx90a HIPCC="hipcc" CFLAGS="-w $CFLAGS" ',
        'patches': ['cupy_lic_def.patch'],
    }),
]

sanity_check_paths = {
    'files': [
        'lib/python%(pyshortver)s/site-packages/cupyx/cusolver.cpython-311-x86_64-linux-gnu.so',
    ],
    'dirs': ['lib/python%(pyshortver)s/site-packages'],
}

moduleclass = 'devel'

[CuPy] [package list]