fix problems with missing rocm dependencies (#33381)
This commit is contained in:
parent
8e78a91ebd
commit
8aa09fd1c0
1 changed files with 37 additions and 1 deletions
|
@ -192,7 +192,43 @@ def test(self):
|
|||
# using the tests copied from <prefix>/share/Tasmanian/testing
|
||||
cmake_dir = self.test_suite.current_test_cache_dir.testing
|
||||
|
||||
if not self.run_test(cmake_bin, options=[cmake_dir], purpose="Generate the Makefile"):
|
||||
options = [
|
||||
cmake_dir,
|
||||
]
|
||||
if "+rocm" in self.spec:
|
||||
options.append(
|
||||
"-DAMDDeviceLibs_DIR="
|
||||
+ join_path(self.spec["llvm-amdgpu"].prefix, "lib", "cmake", "AMDDeviceLibs")
|
||||
)
|
||||
options.append(
|
||||
"-Damd_comgr_DIR="
|
||||
+ join_path(self.spec["comgr"].prefix, "lib", "cmake", "amd_comgr")
|
||||
)
|
||||
options.append(
|
||||
"-Dhsa-runtime64_DIR="
|
||||
+ join_path(self.spec["hsa-rocr-dev"].prefix, "lib", "cmake", "hsa-runtime64")
|
||||
)
|
||||
options.append(
|
||||
"-DHSA_HEADER=" + join_path(self.spec["hsa-rocr-dev"].prefix, "include")
|
||||
)
|
||||
options.append(
|
||||
"-DCMAKE_INCLUDE_PATH="
|
||||
+ join_path(self.spec["hsa-rocr-dev"].prefix, "include", "hsa")
|
||||
)
|
||||
options.append(
|
||||
"-Drocblas_DIR="
|
||||
+ join_path(self.spec["rocblas"].prefix, "lib", "cmake", "rocblas")
|
||||
)
|
||||
options.append(
|
||||
"-Drocsparse_DIR="
|
||||
+ join_path(self.spec["rocsparse"].prefix, "lib", "cmake", "rocsparse")
|
||||
)
|
||||
options.append(
|
||||
"-Drocsolver_DIR="
|
||||
+ join_path(self.spec["rocsolver"].prefix, "lib", "cmake", "rocsolver")
|
||||
)
|
||||
|
||||
if not self.run_test(cmake_bin, options=options, purpose="Generate the Makefile"):
|
||||
tty.msg("Skipping tasmanian test: failed to generate Makefile")
|
||||
return
|
||||
|
||||
|
|
Loading…
Reference in a new issue