build-wheels-dev.sh 1.43 KB
Newer Older
yan.yan's avatar
yan.yan committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#!/bin/bash
# Copyright 2021 Yan Yan
# 
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# 
#     http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

set -e -u -x

function repair_wheel {
    wheel="$1"
    outpath="$2"
    if ! auditwheel show "$wheel"; then
        echo "Skipping non-platform wheel $wheel"
    else
        auditwheel repair "$wheel" --plat "$PLAT" -w "$outpath"
    fi
}
gcc -v
export SPCONV_DISABLE_JIT="1"
yan.yan's avatar
yan.yan committed
29
export CUMM_CUDA_ARCH_LIST="8.6"
yan.yan's avatar
yan.yan committed
30
export SPCONV_PYTHON_LIST="3.13"
yan.yan's avatar
yan.yan committed
31
32
33
34
35
36
# Compile wheels, we only support 3.6-3.10.
# "/opt/python/cp36-cp36m/bin/pip" wheel /io/ --no-deps -w /io/wheelhouse_tmp

for PYVER in ${SPCONV_PYTHON_LIST//;/ }
do
    PYVER2=`echo "$PYVER" | sed 's/\.//'`
yan.yan's avatar
yan.yan committed
37
    PYVER_T=`echo "$PYVER2" | sed 's/t//'`
yan.yan's avatar
yan.yan committed
38

yan.yan's avatar
yan.yan committed
39
    PYVER_CP="cp$PYVER_T-cp$PYVER2"
yan.yan's avatar
yan.yan committed
40
41
42
43
44
45
46
47
48
    "/opt/python/$PYVER_CP/bin/pip" wheel /io/  -v --no-deps -w /io/wheelhouse_tmp
done

# Bundle external shared libraries into the wheels
for whl in /io/wheelhouse_tmp/*.whl; do
    repair_wheel "$whl" /io/dist
done

rm -rf /io/wheelhouse_tmp