find rocm fix (#40388)

* find rocm fix
* format fix
* style fix
* formatting is broken
This commit is contained in:
Miroslav Stoyanov 2023-10-10 13:42:26 -04:00 committed by GitHub
parent 6c7b2e1056
commit 4d51810888
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -189,12 +189,14 @@ def test_make_test(self):
options = [cmake_dir]
if "+rocm" in self.spec:
options.append(f"-Dhip_DIR={self.spec['hip'].prefix.lib.cmake.hip}")
options.append(
f"-DAMDDeviceLibs_DIR={self.spec['llvm-amdgpu'].prefix.lib.cmake.AMDDeviceLibs}"
)
options.append(f"-Damd_comgr_DIR={self.spec['comgr'].prefix.lib.cmake.amd_comgr}")
options.append(
f"-Dhsa-runtime64_DIR={self.spec['hsa-rocr-dev'].prefix.lib.cmake.hsa-runtime64}"
"-Dhsa-runtime64_DIR="
+ join_path(self.spec["hsa-rocr-dev"].prefix.lib.cmake, "hsa-runtime64")
)
options.append(f"-DHSA_HEADER={self.spec['hsa-rocr-dev'].prefix.include}")
options.append(f"-DCMAKE_INCLUDE_PATH={self.spec['hsa-rocr-dev'].prefix.include.hsa}")