From e6f0fdd6539eab7d42b76a826ca9090272d4f0ef Mon Sep 17 00:00:00 2001 From: Suffian Khan Date: Thu, 11 Nov 2021 09:32:09 -0800 Subject: [PATCH] Strip AMD libraries bundled with Python package due to libonnxruntime_providers_rocm.so change (#9679) * remove AMD library depedence from libonnxruntime_providers_rocm.so * fix flake error * remove rocm dependency from original library as well --- setup.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 41b775b4f2..c6ff349f07 100644 --- a/setup.py +++ b/setup.py @@ -163,9 +163,11 @@ try: check=True, stdout=subprocess.PIPE, universal_newlines=True) cuda_dependencies = ['libcublas.so', 'libcublasLt.so', 'libcudnn.so', 'libcudart.so', 'libcurand.so', 'libcufft.so', 'libnvToolsExt.so'] + rocm_dependencies = ['librccl.so', 'libamdhip64.so', 'librocblas.so', 'libMIOpen.so', + 'libhsa-runtime64.so', 'libhsakmt.so'] args = ['patchelf', '--debug'] for line in result.stdout.split('\n'): - for dependency in cuda_dependencies: + for dependency in (cuda_dependencies + rocm_dependencies): if dependency in line: if dependency not in to_preload: to_preload_cuda.append(line)