upcxx: new package

Adding a UPC++ spack package pointing to the default 2019.3.2 UPC++
version, and allowing to build UPC++ with the experimental CUDA variant.

1 files changed, 89 insertions, 0 deletions

diff --git a/var/spack/repos/builtin/packages/upcxx/package.py b/var/spack/repos/builtin/packages/upcxx/package.py
new file mode 100644
index 0000000000..9a02e3274f
--- /dev/null
+++ b/var/spack/repos/builtin/packages/upcxx/package.py
@@ -0,0 +1,89 @@
+# Copyright 2013-2019 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 llnl.util.tty as tty
+import os
+
+
+def detect_scheduler():
+    if (os.environ.get('CROSS') is None):
+        srunbin = which('srun')
+        if srunbin is None:
+            aprunbin = which('aprun')
+            if aprunbin is None:
+                tty.warn("CROSS has not been set, however "
+                         "cannot detect scheduler.")
+                return 'none'
+            else:
+                tty.warn("CROSS has not been set, however "
+                         "aprun has been found, assuming alps scheduler.")
+                return 'alps'
+        else:
+            tty.warn("CROSS has not been set, however "
+                     "srun has been found, assuming slurm scheduler.")
+            return 'slurm'
+    else:
+        tty.warn("CROSS has been set to %s by the user."
+                 % os.environ.get('CROSS'))
+        return 'user'
+
+
+class Upcxx(Package):
+    """UPC++ is a C++ library that supports Partitioned Global Address Space
+    (PGAS) programming, and is designed to interoperate smoothly and
+    efficiently with MPI, OpenMP, CUDA and AMTs. It leverages GASNet-EX to
+    deliver low-overhead, fine-grained communication, including Remote Memory
+    Access (RMA) and Remote Procedure Call (RPC)."""
+
+    homepage = "https://upcxx.lbl.gov"
+
+    version('2019.3.2', '844722cb0e8c0bc649017fce86469457')
+
+    variant('cuda', default=False, 
+            description='Builds a CUDA-enabled version of UPC++')
+
+    variant('scheduler', values=('slurm', 'alps', 'user', 'none'), 
+            default=detect_scheduler(),
+            description="Resource manager to use")
+    conflicts('scheduler=none', when='platform=cray', 
+              msg='None is unacceptable on Cray.')
+
+    depends_on('cuda', when='+cuda')
+
+    def url_for_version(self, version):
+        if version > Version('2019.3.2'):
+            url = "https://bitbucket.org/berkeleylab/upcxx/downloads/upcxx-{0}.tar.gz"
+        else:
+            url = "https://bitbucket.org/berkeleylab/upcxx/downloads/upcxx-{0}-offline.tar.gz"
+        return url.format(version)
+
+    def setup_environment(self, spack_env, run_env):
+        if 'platform=cray' in self.spec:
+            spack_env.set('GASNET_CONFIGURE_ARGS', '--enable-mpi=probe')
+            if "scheduler=slurm" in self.spec:
+                tty.warn("CROSS has been automatically set to %s."
+                         % 'cray-aries-slurm')
+                spack_env.set('CROSS', 'cray-aries-slurm')
+            elif "scheduler=alps" in self.spec:
+                tty.warn("CROSS has been automatically set to %s."
+                         % 'cray-aries-alps')
+                spack_env.set('CROSS', 'cray-aries-alps')
+        if '+cuda' in self.spec:
+            spack_env.set('UPCXX_CUDA', '1')
+            spack_env.set('UPCXX_CUDA_NVCC', self.spec['cuda'].prefix.bin.nvcc)
+
+    def setup_dependent_package(self, module, dep_spec):
+        dep_spec.upcxx = self.prefix.bin.upcxx
+
+    def setup_dependent_environment(self, spack_env, run_env, dependent_spec):
+        spack_env.set('UPCXX_INSTALL', self.prefix)
+        spack_env.set('UPCXX', self.prefix.bin.upcxx)
+        if 'platform=cray' in self.spec:
+            spack_env.set('UPCXX_GASNET_CONDUIT', 'aries')
+
+    def install(self, spec, prefix):
+        installsh = Executable('./install')
+        installsh(prefix)