saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-28 22:56:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[ROCm] update rocm package exclude libs (#18130)

Browse source 
update rocm package exclude libs.

- change librocblas.so.0 to librocblas.so.3 which is used on ROCm5.6 and
ROCm5.7
- add librocfft.so.0, libhipfft.so.0, libhiprtc.so.5 and sort the list.
This commit is contained in:
PeixuanZuo 2023-10-31 08:41:01 +08:00 committed by GitHub
parent 785e2b1eae
commit efef6407bc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
setup.py
View file

@ -203,19 +203,22 @@ try:
"libcurand.so.10",
]
rocm_dependencies = [
"librccl.so.1",
"libnuma.so.1",
"libamd_comgr.so.2",
"libdrm.so.2",
"librocblas.so.0",
"libdrm_amdgpu.so.1",
"libamdhip64.so.5",
"libroctracer64.so.4",
"libMIOpen.so.1",
"libtinfo.so.6",
"libdrm.so.2",
"libdrm_amdgpu.so.1",
"libelf.so.1",
"librocm_smi64.so.5",
"libhipfft.so.0",
"libhiprtc.so.5",
"libhsa-runtime64.so.1",
"libMIOpen.so.1",
"libnuma.so.1",
"librccl.so.1",
"librocblas.so.3",
"librocfft.so.0",
"librocm_smi64.so.5",
"libroctracer64.so.4",
"libtinfo.so.6",
]
tensorrt_dependencies = ["libnvinfer.so.8", "libnvinfer_plugin.so.8", "libnvonnxparser.so.8"]