From 45de1ab36acd101e1adf3b0863a4e231dab304ec Mon Sep 17 00:00:00 2001
From: boeschf <48126478+boeschf@users.noreply.github.com>
Date: Tue, 22 Nov 2022 14:14:24 +0100
Subject: [PATCH] spack gpu option (#2043)
added conditional variant for cuda builds to enable GPU-based random number generation
---
spack/package.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/spack/package.py b/spack/package.py
index 14fe8ea1..f39f7c10 100644
--- a/spack/package.py
+++ b/spack/package.py
@@ -56,6 +56,12 @@ class Arbor(CMakePackage, CudaPackage):
default=False,
description="Enable vectorization of computational kernels",
)
+ variant(
+ "gpu_rng",
+ default=False,
+ description="Use GPU generated random numbers -- not bitwise equal to CPU version",
+ when="+cuda",
+ )
# https://docs.arbor-sim.org/en/latest/install/build_install.html#compilers
conflicts("%gcc@:8")
@@ -109,6 +115,7 @@ 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"))
# query spack for the architecture-specific compiler flags set by its wrapper
args.append("-DARB_ARCH=none")
--
GitLab