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:
Sreenivasa Murthy Kolam 2020-12-25 10:24:59 +05:30 committed by GitHub
parent 4c8b773a42
commit ba62dc3dce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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