Draft: feat(HPC): build image on JUSUF

packages/arbor/package.py

@@ -17,6 +17,7 @@ class Arbor(CMakePackage, CudaPackage):
 
     version("master", branch="master", submodules=True)
     version("develop", branch="master", submodules=True)
+    version("jusuf-benchmark", commit="c4ff08ceef2760796538c4117a259fd2104442ed", submodules=True)
     version(
         "0.10.0",
         sha256="72966b7a2f45ce259b8ba167ca3e4f5ab9f212136a300267aaac0c04ed3fe3fc",
@@ -107,6 +108,11 @@ class Arbor(CMakePackage, CudaPackage):
     # end EBRAINS (modified)
     depends_on("py-mpi4py", when="+mpi+python", type=("build", "run"))
 
+    # begin EBRAINS (modified: added build dep)
+    depends_on("py-pybind11-stubgen", when="@0.11:", type="build")
+    depends_on("py-pybind11-stubgen", when="@jusuf-benchmark", type="build")
+    # end EBRAINS (modified)
+
     # python (bindings)
     with when("+python"):
         extends("python")
@@ -140,6 +146,10 @@ class Arbor(CMakePackage, CudaPackage):
         if "+cuda" in self.spec:
             args.append("-DARB_GPU=cuda")
             args.append(self.define_from_variant("ARB_USE_GPU_RNG", "gpu_rng"))
+            # begin EBRAINS (modified: set cuda arch)
+            cuda_arch = self.spec.variants["cuda_arch"].value
+            args.append("-DCMAKE_CUDA_ARCHITECTURES=%s" % cuda_arch)
+            # end EBRAINS
 
         # query spack for the architecture-specific compiler flags set by its wrapper
         args.append("-DARB_ARCH=none")
@@ -154,6 +164,10 @@ class Arbor(CMakePackage, CudaPackage):
 
         return args
 
+    @on_package_attributes(when="+cuda")
+    def check(self):
+        pass
+
     @run_after("install", when="+python")
     @on_package_attributes(run_tests=True)
     def install_test(self):