build_all_wheel.sh 463 Bytes
Newer Older
Chen Xin's avatar
Chen Xin committed
1
2
3
4
5
6
#!/usr/bin/env bash

set -eou pipefail

TOPDIR=$(git rev-parse --show-toplevel)/builder

7
8
CUDA_VER=${CUDA_VER:-11.8}

Chen Xin's avatar
Chen Xin committed
9
PLAT_NAME=manylinux2014_x86_64
10
for cuver in ${CUDA_VER}; do
Chen Xin's avatar
Chen Xin committed
11
12
13
14
15
16
17
    DOCKER_TAG=cuda${cuver}
    OUTPUT_FOLDER=cuda${cuver}_dist
    for pyver in py38 py39 py310 py311; do
        bash ${TOPDIR}/manywheel/build_wheel.sh ${pyver} ${PLAT_NAME} ${DOCKER_TAG} ${OUTPUT_FOLDER} \
            |& tee ${PLAT_NAME}.${pyver}.cuda${cuver}.log.txt
    done
done