hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

rocm-tensile: fix tensile_architecture variant (#35726)

Browse source 
* rocm-tensile: Fix TENSILE_USE_OPENMP CMake flag to track openmp variant

* rocm-tensile: Add CMAKE_HIP_ARCHITECTURES CMake flag
This commit is contained in:
Wileam Y. Phan 2023-12-28 01:25:36 -08:00 committed by GitHub
parent af49cb9724
commit 858b738775
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
var/spack/repos/builtin/packages/rocm-tensile/package.py
View file

@ -252,6 +252,7 @@ def cmake_args(self):
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 @@ def cmake_args(self):
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"))