diff options
| author | Wileam Y. Phan <50928756+wyphan@users.noreply.github.com> | 2023-12-28 01:25:36 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-28 10:25:36 +0100 |
| commit | 858b7387751fc476907818b2225c54ef9e8b5752 (patch) | |
| tree | 9937febabcf9e960f50ed785e9ea0a2c7e5ee9e0 | |
| parent | af49cb9724c5be81d8775344a076851138999a5a (diff) | |
| download | spack-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
| -rw-r--r-- | var/spack/repos/builtin/packages/rocm-tensile/package.py | 10 |
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")) |