Skip to content

Commit

Permalink
Merge pull request easybuilders#21603 from ankekreuzer/core-packages-…
Browse files Browse the repository at this point in the history
…JSC-part10

[lib, numlib]{SYSTEM} cuDNN-9.5.0.5, cuTENSOR-2.0.2.5
  • Loading branch information
SebastianAchilles authored Nov 13, 2024
2 parents df6a3af + 3976ba0 commit c2ee284
Show file tree
Hide file tree
Showing 2 changed files with 77 additions and 0 deletions.
37 changes: 37 additions & 0 deletions easybuild/easyconfigs/c/cuDNN/cuDNN-9.5.0.50-CUDA-12.6.0.eb
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
name = 'cuDNN'
version = '9.5.0.50'
versionsuffix = '-CUDA-%(cudaver)s'
homepage = 'https://developer.nvidia.com/cudnn'
description = """The NVIDIA CUDA Deep Neural Network library (cuDNN) is
a GPU-accelerated library of primitives for deep neural networks."""

toolchain = SYSTEM

# note: cuDNN is tied to specific to CUDA versions,
# see also https://docs.nvidia.com/deeplearning/cudnn/support-matrix/index.html#cudnn-cuda-hardware-versions
local_short_ver = '.'.join(version.split('.')[:3])
local_cuda_major = '12'

source_urls = [
'https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-%(cudnnarch)s/'
]
sources = ['%%(namelower)s-linux-%%(cudnnarch)s-%%(version)s_cuda%s-archive.tar.xz' % local_cuda_major]
checksums = [{
'%%(namelower)s-linux-sbsa-%%(version)s_cuda%s-archive.tar.xz' % local_cuda_major:
'494b640a69feb40ce806a726aa63a1de6b2ec459acbe6a116ef6fe3e6b27877d',
'%%(namelower)s-linux-x86_64-%%(version)s_cuda%s-archive.tar.xz' % local_cuda_major:
'86e4e4f4c09b31d3850b402d94ea52741a2f94c2f717ddc8899a14aca96e032d',
}]

dependencies = [('CUDA', '12.6.0')]

sanity_check_paths = {
'files': [
'include/cudnn.h', 'lib64/libcudnn_adv_static.a', 'lib64/libcudnn_cnn_static.a',
'lib64/libcudnn_engines_precompiled_static.a', 'lib64/libcudnn_engines_runtime_compiled_static.a',
'lib64/libcudnn_graph_static.a', 'lib64/libcudnn_heuristic_static.a', 'lib64/libcudnn_ops_static.a',
],
'dirs': ['include', 'lib64'],
}

moduleclass = 'numlib'
40 changes: 40 additions & 0 deletions easybuild/easyconfigs/c/cuTENSOR/cuTENSOR-2.0.2.5-CUDA-12.6.0.eb
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
easyblock = 'Tarball'

name = 'cuTENSOR'
version = '2.0.2.5'
versionsuffix = '-CUDA-%(cudaver)s'

homepage = 'https://developer.nvidia.com/cutensor'
description = """The cuTENSOR Library is a GPU-accelerated tensor linear algebra library providing tensor contraction,
reduction and elementwise operations."""

toolchain = SYSTEM

source_urls = [
'https://developer.download.nvidia.com/compute/cutensor/redist/libcutensor/linux-%(arch)s/'
]
sources = ['libcutensor-linux-%(arch)s-%(version)s-archive.tar.xz']

checksums = [{
'libcutensor-linux-sbsa-%(version)s-archive.tar.xz':
'5163dd40f11f328e469a6d9b0056c8346f5d59ed538c18d6b954e4ae657c69cc',
'libcutensor-linux-x86_64-%(version)s-archive.tar.xz':
'0e957ae7b352f599de34b6fa1ba999b0617887f885d7436ac5737d71a6b83baa',
}]

local_cudamajver = '12'
dependencies = [('CUDA', '12.6.0')]

sanity_check_paths = {
'files': ['include/cutensor.h', 'include/cutensor/types.h',
'lib/%s/libcutensor.%s' % (local_cudamajver, SHLIB_EXT),
'lib/%s/libcutensor_static.a' % local_cudamajver],
'dirs': [],
}

modextrapaths = {
'LD_LIBRARY_PATH': ['lib/%s' % local_cudamajver],
'LIBRARY_PATH': ['lib/%s' % local_cudamajver],
}

moduleclass = 'lib'

0 comments on commit c2ee284

Please sign in to comment.