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',
|
||||
'rocminfo', 'rocm-device-libs'))
|
||||
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
|
||||
return mydict
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user