new package: op2-dsl (#28396)

1 files changed, 58 insertions, 0 deletions

diff --git a/var/spack/repos/builtin/packages/op2-dsl/package.py b/var/spack/repos/builtin/packages/op2-dsl/package.py
new file mode 100644
index 0000000000..9679404d5c
--- /dev/null
+++ b/var/spack/repos/builtin/packages/op2-dsl/package.py
@@ -0,0 +1,58 @@
+# 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 *
+
+
+class Op2Dsl(MakefilePackage, CudaPackage):
+    """OP2 is a high-level embedded domain specific language for writing
+    unstructured mesh algorithms with automatic parallelisation on multi-core
+    and many-core architectures."""
+
+    homepage = "https://op-dsl.github.io/"
+    git      = "https://github.com/OP-DSL/OP2-Common.git"
+
+    maintainers = ['gihanmudalige', 'reguly', 'bozbez']
+
+    version('master', branch='master')
+    version('1.1.0', tag='v1.1.0')
+
+    build_directory = 'op2'
+
+    variant('mpi', default=False, description='Enable MPI support')
+
+    variant('parmetis', default=True, when='+mpi',
+            description='Enable ParMETIS partitioning support')
+
+    variant('scotch', default=True, when='+mpi',
+            description='Enable PT-Scotch partitioning support')
+
+    depends_on('mpi', when='+mpi')
+
+    depends_on('parmetis', when='+parmetis')
+    depends_on('scotch', when='+scotch')
+
+    depends_on('hdf5+fortran+mpi', when='+mpi')
+    depends_on('hdf5+fortran~mpi', when='~mpi')
+
+    def edit(self, spec, prefix):
+        compiler_map = {
+            'gcc': 'gnu',
+            'cce': 'cray',
+            'intel': 'intel',
+            'nvhpc': 'nvhpc',
+            'xl': 'xl',
+        }
+
+        if self.spec.compiler.name in compiler_map:
+            env['OP2_COMPILER'] = compiler_map[self.spec.compiler.name]
+
+        if '+cuda' in self.spec and spec.variants['cuda_arch'].value[0] != 'none':
+            env['CUDA_GEN'] = ','.join(spec.variants['cuda_arch'].value)
+
+    def install(self, spec, prefix):
+        install_tree('op2/lib', prefix.lib)
+        install_tree('op2/include', prefix.include)
+        install_tree('op2/mod', prefix.mod)