[Kokkos-kernels] [package list]

Kokkos-kernels/4.6.02-cpeAMD-25.03-rocm (Kokkos-kernels-4.6.02-cpeAMD-25.03-rocm.eb)

Install with the EasyBuild-user module:

eb Kokkos-kernels-4.6.02-cpeAMD-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 Kokkos-kernels/4.6.02-cpeAMD-25.03-rocm.

EasyConfig:

easyblock = 'CMakeMake'

name =          'Kokkos-kernels'
version =       '4.6.02'
versionsuffix = '-rocm'

homepage = 'https://github.com/kokkos/kokkos-kernels'

whatis = [
    "Kokkos C++ Performance Portability Programming Ecosystem: Math Kernels - Provides BLAS, Sparse BLAS and Graph Kernels"
]

description = """
Kokkos C++ Performance Portability Programming Ecosystem: Math Kernels - 
Provides BLAS, Sparse BLAS and Graph Kernels
"""

toolchain = {'name': 'cpeAMD', 'version': '25.03'}
toolchainopts = {'verbose': False, 'openmp': True}

sources = [{
    'filename': '%(namelower)s-%(version)s.tar.gz',
    'git_config': {
        'url': 'https://github.com/kokkos',
        'repo_name': '%(namelower)s',
        'commit': '%(version)s',
    }
}]

builddependencies = [
    ('buildtools', '%(toolchain_version)s', '', True),
]

dependencies = [
    ('rocm',   EXTERNAL_MODULE),
    ('Kokkos', version,         '-rocm'),
]

preconfigopts = 'module unload craype-accel-amd-gfx90a && '

configopts = ' '.join([
    '-D Kokkos_ROOT=${EBROOTKOKKOS}',
    '-D CMAKE_CXX_COMPILER=CC', # Isn't really needed as EasyBuild does that already, but stresses the difference with the cpeGNU version.
    '-D BUILD_SHARED_LIBS=ON',
    '-D KokkosKernels_ENABLE_TPL_BLAS=ON',
    '-D KokkosKernels_ENABLE_TPL_LAPACK=ON',
])

prebuildopts = 'module unload craype-accel-amd-gfx90a && '

sanity_check_paths = {
    'files': ['lib/libkokkoskernels.%s' % SHLIB_EXT,
              'include/KokkosKernels_config.h'],
    'dirs':  ['lib', 'include'],
}

moduleclass = 'lib'

[Kokkos-kernels] [package list]