# IT4Innovations
# LK 2023

easyblock = 'Tarball'

name = 'cuTENSOR'
version = '2.0.0.7'
versionsuffix = '-CUDA-12.2.0'

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-x86_64/']
sources = ['libcutensor-linux-x86_64-%(version)s-archive.tar.xz']
checksums = ['df6338aed18e5b6ae0c49521053d1606d2a41e187d7f5eccf9180af6bbc4ef68']

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

sanity_check_paths = {
    'files': ['include/cutensor.h', 'include/cutensor/types.h',
              'lib/12/libcutensor.so',
              'lib/12/libcutensor_static.a'],
    'dirs': [],
}

modextrapaths = {
    'LD_LIBRARY_PATH': ['lib/12'],
    'LIBRARY_PATH': ['lib/12'],
}

moduleclass = 'lib'