# Copyright 2013-2021 Lawrence Livermore National Security, LLC and other # Spack Project Developers. See the top-level COPYRIGHT file for details. # # SPDX-License-Identifier: (Apache-2.0 OR MIT) from spack import * import datetime as dt class Lammps(CMakePackage, CudaPackage): """LAMMPS stands for Large-scale Atomic/Molecular Massively Parallel Simulator. This package uses patch releases, not stable release. See https://github.com/spack/spack/pull/5342 for a detailed discussion. """ homepage = "http://lammps.sandia.gov/" url = "https://github.com/lammps/lammps/archive/patch_1Sep2017.tar.gz" git = "https://github.com/lammps/lammps.git" tags = ['ecp', 'ecp-apps'] version('master', branch='master') version('20200721', sha256='845bfeddb7b667799a1a5dbc166b397d714c3d2720316604a979d3465b4190a9') version('20200630', sha256='413cbfabcc1541a339c7a4ab5693fbeb768f46bb1250640ba94686c6e90922fc') version('20200505', sha256='c49d77fd602d28ebd8cf10f7359b9fc4d14668c72039028ed7792453d416de73') version('20200303', sha256='9aa56dfb8673a06e6c88588505ec1dfc01dd94f9d60e719ed0c605e48cc06c58') version('20200227', sha256='1aabcf38bc72285797c710b648e906151a912c36b634a9c88ac383aacf85516e') version('20200218', sha256='73bcf146660804ced954f6a0a8dce937482677778d46018ca5a688127bf97211') version('20200204', sha256='3bf3de546ede34ffcd89f1fca5fd66aa78c662e7c8a76e30ce593e44a00d23ce') version('20200124', sha256='443829560d760690e1ae21ad54922f56f34f640a81e817f5cc65d2a4af3a6a5d') version('20200109', sha256='f2fd24f6c10837801f490913d73f672ec7c6becda08465d7e834a2bfbe3d7cd6') version('20191120', sha256='fd146bf517a6c2fb8a69ecb3749dc352eef94414739cd7855c668c690af85d27') version('20191030', sha256='5279567f731386ffdb87800b448903a63de2591064e13b4d5216acae25b7e541') version('20190919', sha256='0f693203afe86bc70c084c55f29330bdeea3e3ad6791f81c727f7a34a7f6caf3') version('20190807', sha256='895d71914057e070fdf0ae5ccf9d6552b932355056690bdb8e86d96549218cc0') version('20190605', sha256='c7b35090aef7b114d2b47a7298c1e8237dd811da87995c997bf7639cca743152') version('20181212', sha256='ccc5d2c21c4b62ce4afe7b3a0fe2f37b83e5a5e43819b7c2e2e255cce2ce0f24') version('20181207', sha256='d92104d008a7f1d0b6071011decc5c6dc8b936a3418b20bd34b055371302557f') version('20181127', sha256='c076b633eda5506f895de4c73103df8b995d9fec01be82c67c7608efcc345179') version('20181115', sha256='3bc9c166e465cac625c4a8e4060e597003f4619dadd57d3bc8d25bcd930f286e') version('20181109', sha256='dd30fe492fa147fb6f39bfcc79d8c786b9689f7fbe86d56de58cace53b6198c9') version('20181024', sha256='a171dff5aff7aaa2c9606ab2abc9260f2b685a5c7f6d650e7f2b59cf4aa149d6') version('20181010', sha256='bda762ee2d2dcefe0b4e36fb689c6b9f7ede49324444ccde6c59cba727b4b02d') version('20180918', sha256='02f143d518d8647b77137adc527faa9725c7afbc538d670253169e2a9b3fa0e6') version('20180905', sha256='ee0df649e33a9bf4fe62e062452978731548a56b7487e8e1ce9403676217958d') version('20180831', sha256='6c604b3ebd0cef1a5b18730d2c2eb1e659b2db65c5b1ae6240b8a0b150e4dff3') version('20180822', sha256='9f8942ca3f8e81377ae88ccfd075da4e27d0dd677526085e1a807777c8324074') version('20180629', sha256='1acf7d9b37b99f17563cd4c8bb00ec57bb2e29eb77c0603fd6871898de74763b') version('20180316', sha256='a81f88c93e417ecb87cd5f5464c9a2570384a48ff13764051c5e846c3d1258c1') version('20180222', sha256='374254d5131b7118b9ab0f0e27d20c3d13d96b03ed2b5224057f0c1065828694') version('20170922', sha256='f0bf6eb530d528f4d261d0a261e5616cbb6e990156808b721e73234e463849d3') version('20170901', sha256='5d88d4e92f4e0bb57c8ab30e0d20de556830af820223778b9967bec2184efd46') def url_for_version(self, version): vdate = dt.datetime.strptime(str(version), "%Y%m%d") return "https://github.com/lammps/lammps/archive/patch_{0}.tar.gz".format( vdate.strftime("%d%b%Y").lstrip('0')) supported_packages = ['asphere', 'body', 'class2', 'colloid', 'compress', 'coreshell', 'dipole', 'granular', 'kspace', 'kokkos', 'latte', 'manybody', 'mc', 'meam', 'misc', 'mliap', 'molecule', 'mpiio', 'opt', 'peri', 'poems', 'python', 'qeq', 'replica', 'rigid', 'shock', 'snap', 'spin', 'srd', 'user-atc', 'user-adios', 'user-awpmd', 'user-bocs', 'user-cgsdk', 'user-colvars', 'user-diffraction', 'user-dpd', 'user-drude', 'user-eff', 'user-fep', 'user-h5md', 'user-lb', 'user-manifold', 'user-meamc', 'user-mesodpd', 'user-mesont', 'user-mgpt', 'user-misc', 'user-mofff', 'user-netcdf', 'user-omp', 'user-phonon', 'user-plumed', 'user-ptm', 'user-qtb', 'user-reaction', 'user-reaxc', 'user-sdpd', 'user-smd', 'user-smtbq', 'user-sph', 'user-tally', 'user-uef', 'user-yaff', 'voronoi'] for pkg in supported_packages: variant(pkg, default=False, description='Activate the {0} package'.format(pkg)) variant('lib', default=True, description='Build the liblammps in addition to the executable') variant('mpi', default=True, description='Build with mpi') variant('jpeg', default=True, description='Build with jpeg support') variant('png', default=True, description='Build with png support') variant('ffmpeg', default=True, description='Build with ffmpeg support') variant('openmp', default=True, description='Build with OpenMP') variant('opencl', default=False, description='Build with OpenCL') variant('exceptions', default=False, description='Build with lammps exceptions') variant('cuda_mps', default=False, description='(CUDA only) Enable tweaks for running ' + 'with Nvidia CUDA Multi-process services daemon') depends_on('mpi', when='+mpi') depends_on('mpi', when='+mpiio') depends_on('fftw-api@3', when='+kspace') depends_on('voropp+pic', when='+voronoi') depends_on('netcdf-c+mpi', when='+user-netcdf') depends_on('blas', when='+user-atc') depends_on('lapack', when='+user-atc') depends_on('opencl', when='+opencl') depends_on('latte@1.0.1', when='@:20180222+latte') depends_on('latte@1.1.1:', when='@20180316:20180628+latte') depends_on('latte@1.2.1:', when='@20180629:20200505+latte') depends_on('latte@1.2.2:', when='@20200602:+latte') depends_on('blas', when='+latte') depends_on('lapack', when='+latte') depends_on('python', when='+python') depends_on('mpi', when='+user-lb') depends_on('mpi', when='+user-h5md') depends_on('hdf5', when='+user-h5md') depends_on('jpeg', when='+jpeg') depends_on('libpng', when='+png') depends_on('ffmpeg', when='+ffmpeg') depends_on('kokkos+deprecated_code+shared@3.0', when='@20200303+kokkos') depends_on('kokkos+shared@3.1:', when='@20200505:+kokkos') depends_on('adios2', when='+user-adios') depends_on('plumed', when='+user-plumed') depends_on('eigen@3:', when='+user-smd') conflicts('+cuda', when='+opencl') conflicts('+body', when='+poems@:20180628') conflicts('+latte', when='@:20170921') conflicts('+python', when='~lib') conflicts('+qeq', when='~manybody') conflicts('+user-atc', when='~manybody') conflicts('+user-misc', when='~manybody') conflicts('+user-phonon', when='~kspace') conflicts('+user-misc', when='~manybody') conflicts('%gcc@9:', when='@:20200303+openmp') conflicts('+kokkos', when='@:20200227') conflicts( '+meam', when='@20181212:', msg='+meam was removed after @20181212, use +user-meamc instead') conflicts( '+user-meamc', when='@:20181212', msg='+user-meamc only added @20181212, use +meam instead') conflicts( '+user-reaction', when='@:20200303', msg='+user-reaction only supported for version 20200505 and later') conflicts('+mliap', when='~snap') conflicts( '+adios +mpi', when='^adios2~mpi', msg='With +adios, mpi setting for adios2 and lammps must be the same') conflicts( '+adios ~mpi', when='^adios2+mpi', msg='With +adios, mpi setting for adios2 and lammps must be the same') patch("lib.patch", when="@20170901") patch("660.patch", when="@20170922") root_cmakelists_dir = 'cmake' def cmake_args(self): spec = self.spec mpi_prefix = 'ENABLE' pkg_prefix = 'ENABLE' if spec.satisfies('@20180629:'): mpi_prefix = 'BUILD' pkg_prefix = 'PKG' args = [ '-DBUILD_SHARED_LIBS={0}'.format( 'ON' if '+lib' in spec else 'OFF'), '-DLAMMPS_EXCEPTIONS={0}'.format( 'ON' if '+exceptions' in spec else 'OFF'), '-D{0}_MPI={1}'.format( mpi_prefix, 'ON' if '+mpi' in spec else 'OFF'), '-DBUILD_OMP={0}'.format( 'ON' if '+openmp' in spec else 'OFF'), ] if spec.satisfies('+cuda'): args.append('-DPKG_GPU=ON') args.append('-DGPU_API=cuda') cuda_arch = spec.variants['cuda_arch'].value if cuda_arch != 'none': args.append('-DGPU_ARCH=sm_{0}'.format(cuda_arch[0])) args.append('-DCUDA_MPS_SUPPORT={0}'.format( 'ON' if '+cuda_mps' in spec else 'OFF')) elif spec.satisfies('+opencl'): args.append('-DPKG_GPU=ON') args.append('-DGPU_API=opencl') else: args.append('-DPKG_GPU=OFF') if spec.satisfies('@20180629:+lib'): args.append('-DBUILD_LIB=ON') args.append('-DWITH_JPEG={0}'.format( 'ON' if '+jpeg' in spec else 'OFF')) args.append('-DWITH_PNG={0}'.format( 'ON' if '+png' in spec else 'OFF')) args.append('-DWITH_FFMPEG={0}'.format( 'ON' if '+ffmpeg' in spec else 'OFF')) for pkg in self.supported_packages: opt = '-D{0}_{1}'.format(pkg_prefix, pkg.upper()) if '+{0}'.format(pkg) in spec: args.append('{0}=ON'.format(opt)) else: args.append('{0}=OFF'.format(opt)) if '+kspace' in spec: if '^fftw' in spec: args.append('-DFFT=FFTW3') if '^mkl' in spec: args.append('-DFFT=MKL') if '^amdfftw' in spec: fftw_prefix = spec['amdfftw'].prefix args.append('-DFFTW_HOME={0}'.format(fftw_prefix)) args.append('-DFFTW_INCLUDE_DIRS={0}' .format(fftw_prefix.include)) args.append('-DFFTW_LIBRARY_DIRS={0}'.format(fftw_prefix.lib)) if '+kokkos' in spec: args.append('-DEXTERNAL_KOKKOS=ON') if '+user-adios' in spec: args.append('-DADIOS2_DIR={0}'.format(self.spec['adios2'].prefix)) if '+user-plumed' in spec: args.append('-DDOWNLOAD_PLUMED=no') if '+shared' in self.spec['plumed']: args.append('-DPLUMED_MODE=shared') else: args.append('-DPLUMED_MODE=static') if '+user-smd' in spec: args.append('-DDOWNLOAD_EIGEN3=no') args.append('-DEIGEN3_INCLUDE_DIR={0}'.format( self.spec['eigen'].prefix.include)) return args def setup_run_environment(self, env): env.set('LAMMPS_POTENTIALS', self.prefix.share.lammps.potentials)