Skip to content

Commit d864b4f

Browse files
authored
Fix auditwheel version issue (#288)
Auditwheel 6.3.0 changed/removed the lddtree function so cap constraint to 6.2.x
1 parent 7a172d7 commit d864b4f

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

build/rocm/tools/build_wheels.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,10 @@ def fix_wheel(path, jax_path):
226226
py_bin = "/opt/python/cp310-cp310/bin"
227227
env["PATH"] = "%s:%s" % (py_bin, env["PATH"])
228228

229-
cmd = ["pip", "install", "auditwheel>=6"]
229+
# NOTE(mrodden): auditwheel 6.0 added lddtree module, but 6.3.0 changed
230+
# the fuction to ldd and also changed its behavior
231+
# constrain range to 6.0 to 6.2.x
232+
cmd = ["pip", "install", "auditwheel>=6,<6.3"]
230233
subprocess.run(cmd, check=True, env=env)
231234

232235
fixwheel_path = os.path.join(jax_path, "build/rocm/tools/fixwheel.py")

0 commit comments

Comments
 (0)