From 858b7387751fc476907818b2225c54ef9e8b5752 Mon Sep 17 00:00:00 2001 From: "Wileam Y. Phan" <50928756+wyphan@users.noreply.github.com> Date: Thu, 28 Dec 2023 01:25:36 -0800 Subject: rocm-tensile: fix tensile_architecture variant (#35726) * rocm-tensile: Fix TENSILE_USE_OPENMP CMake flag to track openmp variant * rocm-tensile: Add CMAKE_HIP_ARCHITECTURES CMake flag --- var/spack/repos/builtin/packages/rocm-tensile/package.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/rocm-tensile/package.py b/var/spack/repos/builtin/packages/rocm-tensile/package.py index 7f707969b4..a56de4ff8e 100644 --- a/var/spack/repos/builtin/packages/rocm-tensile/package.py +++ b/var/spack/repos/builtin/packages/rocm-tensile/package.py @@ -252,6 +252,7 @@ class RocmTensile(CMakePackage): self.define("Tensile_LOGIC", "asm_full"), self.define("Tensile_CODE_OBJECT_VERSION", "V3"), self.define("Boost_USE_STATIC_LIBS", "OFF"), + self.define_from_variant("TENSILE_USE_OPENMP", "openmp"), self.define("BUILD_WITH_TENSILE_HOST", "ON" if "@3.7.0:" in self.spec else "OFF"), ] @@ -265,7 +266,14 @@ class RocmTensile(CMakePackage): else: args.append(self.define("TENSILE_USE_OPENMP", "OFF")), - args.append(self.define("Tensile_ARCHITECTURE", self.get_gpulist_for_tensile_support())) + if self.spec.satisfies("^cmake@3.21.0:"): + args.append( + self.define("CMAKE_HIP_ARCHITECTURES", self.get_gpulist_for_tensile_support()) + ) + else: + args.append( + self.define("Tensile_ARCHITECTURE", self.get_gpulist_for_tensile_support()) + ) if self.spec.satisfies("^cmake@3.21.0:3.21.2"): args.append(self.define("__skip_rocmclang", "ON")) -- cgit v1.2.3-60-g2f50