diff --git a/var/spack/repos/builtin/packages/nekrs/package.py b/var/spack/repos/builtin/packages/nekrs/package.py new file mode 100644 index 0000000000..731c34e21f --- /dev/null +++ b/var/spack/repos/builtin/packages/nekrs/package.py @@ -0,0 +1,141 @@ +# 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 os + + +class Nekrs(Package, CudaPackage, ROCmPackage): + """nekRS is an open-source Navier Stokes solver based on the spectral + element method targeting classical processors and hardware accelerators + like GPUs""" + + homepage = "https://github.com/Nek5000/nekRS" + git = "https://github.com/Nek5000/nekRS.git" + + tags = ['cfd', 'flow', 'hpc', 'solver', 'navier-stokes', + 'spectral-elements', 'fluid', 'ecp', 'ecp-apps'] + + maintainers = ['thilinarmtb', 'stgeke'] + + version('21.0', tag='v21.0') + + variant('cuda', + default=False, + description='Activates support for CUDA') + variant('opencl', + default=False, + description='Activates support for OpenCL') + variant('hip', + default=False, + description='Activates support for HIP') + + # Conflicts: + # nekrs includes following packages, but in order to build as part of + # CEED we can't leave them in as conflicts. They should be enabled + # sometime in future. + # for pkg in ['occa', 'hyper', 'nek5000', 'blas', 'lapack', 'gslib']: + # conflicts('^' + pkg, msg=(pkg + " is built into nekRS")) + + # Dependencies + depends_on('mpi') + depends_on('git') + depends_on('cmake') + + depends_on('cuda', when='+cuda') + depends_on('hip', when='+rocm') + + @run_before('install') + def fortran_check(self): + if not self.compiler.f77: + msg = 'Cannot build NekRS without a Fortran 77 compiler.' + raise RuntimeError(msg) + + # Following 4 methods are stolen from OCCA since we are using OCCA + # shipped with nekRS. + def _setup_runtime_flags(self, s_env): + spec = self.spec + s_env.set('OCCA_CXX', self.compiler.cxx) + + cxxflags = spec.compiler_flags['cxxflags'] + if cxxflags: + # Run-time compiler flags: + s_env.set('OCCA_CXXFLAGS', ' '.join(cxxflags)) + + if '+cuda' in spec: + cuda_dir = spec['cuda'].prefix + # Run-time CUDA compiler: + s_env.set('OCCA_CUDA_COMPILER', + join_path(cuda_dir, 'bin', 'nvcc')) + + def setup_build_environment(self, env): + spec = self.spec + # The environment variable CXX is automatically set to the Spack + # compiler wrapper. + + # The cxxflags, if specified, will be set by the Spack compiler wrapper + # while the environment variable CXXFLAGS will remain undefined. + # We define CXXFLAGS in the environment to tell OCCA to use the user + # specified flags instead of its defaults. This way the compiler will + # get the cxxflags twice - once from the Spack compiler wrapper and + # second time from OCCA - however, only the second one will be seen in + # the verbose output, so we keep both. + cxxflags = spec.compiler_flags['cxxflags'] + if cxxflags: + env.set('CXXFLAGS', ' '.join(cxxflags)) + + # For the cuda, openmp, and opencl variants, set the environment + # variable OCCA_{CUDA,OPENMP,OPENCL}_ENABLED only if the variant is + # disabled. Otherwise, let OCCA autodetect what is available. + + if '+cuda' in spec: + cuda_dir = spec['cuda'].prefix + cuda_libs_list = ['libcuda', 'libcudart', 'libOpenCL'] + cuda_libs = find_libraries(cuda_libs_list, + cuda_dir, + shared=True, + recursive=True) + env.set('OCCA_INCLUDE_PATH', cuda_dir.include) + env.set('OCCA_LIBRARY_PATH', ':'.join(cuda_libs.directories)) + env.set('OCCA_CUDA_ENABLED', '1') + else: + env.set('OCCA_CUDA_ENABLED', '0') + + env.set('OCCA_OPENCL_ENABLED', '1' if '+opencl' in spec else '0') + env.set('OCCA_HIP_ENABLED', '1' if '+rocm' in spec else '0') + + # Setup run-time environment for testing. + env.set('OCCA_VERBOSE', '1') + self._setup_runtime_flags(env) + + def setup_run_environment(self, env): + # The 'env' is included in the Spack generated module files. + self._setup_runtime_flags(env) + + def setup_dependent_build_environment(self, env, dependent_spec): + # Export OCCA_* variables for everyone using this package from within + # Spack. + self._setup_runtime_flags(env) + + def install(self, spec, prefix): + script_dir = 'scripts' + + with working_dir(script_dir): + # Make sure nekmpi wrapper uses srun when we know OpenMPI + # is not built with mpiexec + if '^openmpi~legacylaunchers' in spec: + filter_file(r'mpirun -np', 'srun -n', 'nrsmpi') + filter_file(r'mpirun -np', 'srun -n', 'nrspre') + filter_file(r'mpirun -np', 'srun -n', 'nrsbmpi') + + makenrs = Executable(os.path.join(os.getcwd(), "makenrs")) + + makenrs.add_default_env("NEKRS_INSTALL_DIR", prefix) + makenrs.add_default_env("NEKRS_CC", spec['mpi'].mpicc) + makenrs.add_default_env("NEKRS_CXX", spec['mpi'].mpicxx) + makenrs.add_default_env("NEKRS_FC", spec['mpi'].mpifc) + makenrs.add_default_env("TRAVIS", "true") + + makenrs(output=str, error=str, fail_on_error=True)