summaryrefslogtreecommitdiff
path: root/var
diff options
context:
space:
mode:
authorWileam Y. Phan <50928756+wyphan@users.noreply.github.com>2023-12-28 01:25:36 -0800
committerGitHub <noreply@github.com>2023-12-28 10:25:36 +0100
commit858b7387751fc476907818b2225c54ef9e8b5752 (patch)
tree9937febabcf9e960f50ed785e9ea0a2c7e5ee9e0 /var
parentaf49cb9724c5be81d8775344a076851138999a5a (diff)
downloadspack-858b7387751fc476907818b2225c54ef9e8b5752.tar.gz
spack-858b7387751fc476907818b2225c54ef9e8b5752.tar.bz2
spack-858b7387751fc476907818b2225c54ef9e8b5752.tar.xz
spack-858b7387751fc476907818b2225c54ef9e8b5752.zip
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
Diffstat (limited to 'var')
-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"))