Skip to content

Commit 68ed43c

Browse files
committed
properly bundle wheel
1 parent b353d2d commit 68ed43c

1 file changed

Lines changed: 28 additions & 0 deletions

File tree

packaging/repair_wheel.py

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff 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:

0 commit comments

Comments
 (0)