summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/osu-micro-benchmarks/package.py11
1 files changed, 5 insertions, 6 deletions
diff --git a/var/spack/repos/builtin/packages/osu-micro-benchmarks/package.py b/var/spack/repos/builtin/packages/osu-micro-benchmarks/package.py
index 8ae5c19631..bf18832bcd 100644
--- a/var/spack/repos/builtin/packages/osu-micro-benchmarks/package.py
+++ b/var/spack/repos/builtin/packages/osu-micro-benchmarks/package.py
@@ -8,7 +8,7 @@ import sys
from spack import *
-class OsuMicroBenchmarks(AutotoolsPackage):
+class OsuMicroBenchmarks(AutotoolsPackage, CudaPackage):
"""The Ohio MicroBenchmark suite is a collection of independent MPI
message passing performance microbenchmarks developed and written at
The Ohio State University. It includes traditional benchmarks and
@@ -28,10 +28,7 @@ class OsuMicroBenchmarks(AutotoolsPackage):
version('5.4', sha256='e1ca762e13a07205a59b59ad85e85ce0f826b70f76fd555ce5568efb1f2a8f33')
version('5.3', sha256='d7b3ad4bee48ac32f5bef39650a88f8f2c23a3050b17130c63966283edced89b')
- variant('cuda', default=False, description="Enable CUDA support")
-
depends_on('mpi')
- depends_on('cuda', when='+cuda')
def configure_args(self):
spec = self.spec
@@ -43,13 +40,15 @@ class OsuMicroBenchmarks(AutotoolsPackage):
if '+cuda' in spec:
config_args.extend([
'--enable-cuda',
- '--with-cuda=%s' % spec['cuda'].prefix,
+ '--with-cuda=%s' % spec['cuda'].prefix
])
+ cuda_arch = spec.variants['cuda_arch'].value
+ if 'none' not in cuda_arch:
+ config_args.append('NVCCFLAGS=' + ' '.join(self.cuda_flags(cuda_arch)))
# librt not available on darwin (and not required)
if not sys.platform == 'darwin':
config_args.append('LDFLAGS=-lrt')
-
return config_args
def setup_run_environment(self, env):