# IT4Innovations
# LK 2024

easyblock = 'MakeCp'

name = 'VASP'
version = '6.4.3'
local_cudaversion = '12.3.0'
versionsuffix = '-CUDA-%s' % local_cudaversion

homepage = 'https://www.vasp.at'
docurls = 'https://www.vasp.at/wiki/index.php/The_VASP_Manual'
description = """
The Vienna Ab initio Simulation Package (VASP) is a local computer program for atomic scale
materials modelling, e.g. electronic structure calculations and quantum-mechanical molecular dynamics,
from first principles.

This is a GPU-enabled build.

To use VASP, you need an academic license from University of Vienna.
Follow the instructions at https://www.vasp.at/index.php/faqs.

Please send us the ID of your VASP license, list of authorized users for whom you require access,
and their email which is associated with your license (use only https://support.it4i.cz/rt).
We are responsible for verifying your licenses.
"""

toolchain = {'name': 'NVHPC', 'version': '24.3-CUDA-12.3.0'}
toolchainopts = {'pic': True}

# Vasp is proprietary software, see http://www.vasp.at/index.php/faqs on
# how to get access to the code
sources = ['%(namelower)s.%(version)s.tgz']
patches = ['VASP-%(version)s-NVHPC-24.3-CUDA-12.3.0-adjust-makefile.patch']
checksums = [
    {'vasp.%(version)s.tgz': 'fe30e773f2a3e909b5e0baa9654032dfbdeff7ec157bc348cee7681a7b6c24f4'},
    {'VASP-%(version)s-NVHPC-24.3-CUDA-12.3.0-adjust-makefile.patch': '2991eb38d138f4066164d185968f37ad9ab88ec00ce80e7cfe9f095dffa115b5'},
]

dependencies = [
    ('OpenMPI', '4.1.6'),
    ('FFTW.MPI', '3.3.10'),
    ('FFTW', '3.3.10'),
    ('imkl', '2022.2.1', '', True),
    ('ScaLAPACK', '3.0'),
    ('HDF5', '1.14.3'),
    ('QD', '2.3.17'),
]

prebuildopts = 'cp arch/makefile.include.nvhpc_acc ./makefile.include && '

# VASP uses LIBS as a list of folders
prebuildopts += 'unset LIBS && '

# AMD/Intel CPU switch - We set xHost by default; change it to -march=core-avx2 when necessary
import os
if os.environ.get("CLUSTERNAME") in ["KAROLINA"]:
    prebuildopts += 'sed -i "s|-xHOST|-march=core-avx2|" makefile.include && '
    prebuildopts += 'sed -i "s|-march=xHost|-march=core-avx2|" makefile.include && '

prebuildopts += 'sed -i "s|cuda12.1|cuda12.3|g" makefile.include && '

buildopts = 'std gam ncl '

parallel = 1

files_to_copy = [(['bin/vasp_std', 'bin/vasp_gam', 'bin/vasp_ncl'], 'bin')]

sanity_check_paths = {
    'files': ['bin/vasp_std', 'bin/vasp_gam', 'bin/vasp_ncl'],
    'dirs': []
}

modluafooter = """
add_property('state','license')
add_property('arch', 'gpu')
"""


moduleclass = 'chem'