fix rocm_3.8.0 build error for rocprim,rocrand,rocfft packages (#20446)
* fix rocm_3.8.0 build error for rocprim,rocrand,rocfft packages * fix flake8 errors
This commit is contained in:
parent
4c8b773a42
commit
ba62dc3dce
@ -82,7 +82,10 @@ def get_rocm_prefix_info(self):
|
|||||||
for name in ('llvm-amdgpu', 'hsa-rocr-dev',
|
for name in ('llvm-amdgpu', 'hsa-rocr-dev',
|
||||||
'rocminfo', 'rocm-device-libs'))
|
'rocminfo', 'rocm-device-libs'))
|
||||||
mydict['rocm-path'] = self.spec.prefix
|
mydict['rocm-path'] = self.spec.prefix
|
||||||
device_lib_path = mydict['rocm-device-libs'].amdgcn.bitcode
|
if '@:3.8.0' in self.spec:
|
||||||
|
device_lib_path = mydict['rocm-device-libs'].lib
|
||||||
|
else:
|
||||||
|
device_lib_path = mydict['rocm-device-libs'].amdgcn.bitcode
|
||||||
mydict['device_lib_path'] = device_lib_path
|
mydict['device_lib_path'] = device_lib_path
|
||||||
return mydict
|
return mydict
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user