zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 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_LOGIC", "asm_full"),
self.define("Tensile_CODE_OBJECT_VERSION", "V3"), self.define("Tensile_CODE_OBJECT_VERSION", "V3"),
self.define("Boost_USE_STATIC_LIBS", "OFF"), 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"), 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: else:
args.append(self.define("TENSILE_USE_OPENMP", "OFF")), 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"): if self.spec.satisfies("^cmake@3.21.0:3.21.2"):
args.append(self.define("__skip_rocmclang", "ON")) args.append(self.define("__skip_rocmclang", "ON"))