File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -168,6 +168,34 @@ def repair_linux(wheels):
168168 "libnvshmem*" ,
169169 "libnvfatbin*" ,
170170 "libnvcuvid*" ,
171+ # ROCm/HIP runtime and its system deps: provided by the torch-ROCm wheel
172+ # (torch/lib/) at runtime, exactly like the CUDA libs above. Never bundle
173+ # them — they'd duplicate torch's copies and bloat the wheel.
174+ "libamdhip64*" ,
175+ "libamd_comgr*" ,
176+ "libhsa-runtime64*" ,
177+ "libhiprtc*" ,
178+ "librocm-core*" ,
179+ "librocprofiler-register*" ,
180+ "libroctx64*" ,
181+ "librocroller*" ,
182+ "libMIOpen*" ,
183+ "libhipblas*" ,
184+ "libhipfft*" ,
185+ "libhiprand*" ,
186+ "libhipsolver*" ,
187+ "libhipsparse*" ,
188+ "librocblas*" ,
189+ "librocfft*" ,
190+ "librocrand*" ,
191+ "librocsolver*" ,
192+ "librocsparse*" ,
193+ "librccl*" ,
194+ "libnuma*" ,
195+ "libdrm*" ,
196+ "libelf*" ,
197+ "libbz2*" ,
198+ "liblzma*" ,
171199 ):
172200 excludes += ["--exclude" , pattern ]
173201 for wheel in wheels :
You can’t perform that action at this time.
0 commit comments