diff options
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/osu-micro-benchmarks/package.py | 8 |
1 files changed, 7 insertions, 1 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 ad0fc1d794..6840dfe076 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.package import * -class OsuMicroBenchmarks(AutotoolsPackage, CudaPackage): +class OsuMicroBenchmarks(AutotoolsPackage, CudaPackage, ROCmPackage): """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 @@ -48,6 +48,12 @@ class OsuMicroBenchmarks(AutotoolsPackage, CudaPackage): if "none" not in cuda_arch: config_args.append("NVCCFLAGS=" + " ".join(self.cuda_flags(cuda_arch))) + if "+rocm" in spec: + config_args.extend(["--enable-rocm", "--with-rocm=%s" % spec["hip"].prefix]) + rocm_arch = spec.variants["amdgpu_target"].value + if "none" not in rocm_arch: + config_args.append("HCC_AMDGPU_TARGET=" + " ".join(self.hip_flags(rocm_arch))) + # librt not available on darwin (and not required) if not sys.platform == "darwin": config_args.append("LDFLAGS=-lrt") |