From de9fc038f7e2622f3b55b2fb1d9724058724eeca Mon Sep 17 00:00:00 2001 From: Pak Lui Date: Mon, 26 Sep 2022 14:26:58 -0700 Subject: add ROCmPackage to OSU Micro Benchmarks (#32806) --- var/spack/repos/builtin/packages/osu-micro-benchmarks/package.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'var') 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") -- cgit v1.2.3-60-g2f50