summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/rocm-tensile/package.py10
1 files changed, 9 insertions, 1 deletions
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"))