Skip to content

Commit fa3d4cc

Browse files
authored
Merge pull request #6 from alykhantejani/add_fix_wheel_hash
Add code to regenerate RECORD file for the wheel after we use patchelf
2 parents 68e0635 + e7e0eee commit fa3d4cc

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

manywheel/build.sh

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,24 @@ for whl in /$WHEELHOUSE_DIR/torch*manylinux*.whl; do
130130
# libshm
131131
patchelf --set-rpath '$ORIGIN' torch/lib/libshm.so
132132

133+
# regenerate the RECORD file with new hashes
134+
record_file=`echo $(basename whl) | sed -e 's/-cp.*$/.dist-info\/RECORD/g'`
135+
new_record_file="$record_file"_new
136+
while read -r line
137+
do
138+
record_item="$line"
139+
IFS=, read -r filename digestmethod digest size <<< "$record_item"
140+
if [ $filename == $record_file ]
141+
then
142+
echo "$line" >> $new_record_file
143+
else
144+
new_digest=`openssl dgst -sha256 -binary $filename | openssl base64 | sed -e 's/+/-/g' | sed -e 's/\//_/g' | sed -e 's/=//g'`
145+
new_size=`ls -nl $filename | awk '{print $5}'`
146+
echo $filename,$digestmethod,$new_digest,$new_size >> $new_record_file
147+
fi
148+
done < "$record_file"
149+
mv $new_record_file $record_file
150+
133151
# zip up the wheel back
134152
zip -r $(basename $whl) torch*
135153

0 commit comments

Comments
 (0)