Tensor learning compiler binary distribution package.
We use github action to build wheel and conda packages nightly.
Checkout .github/workflows
- Build docker images
./docker/build_image.sh <CONTAINER_TYPE>
CONTAINER_NAME: Type of the docker container used to build wheels, e.g., (cpu|cu100|cu101|cu102)
- Checkout tvm and sync version
git clone https://github.com/apache/tvm --recursive
# synchronize the package version
python common/sync_package.py [tlcpack|tlcpack-nightly]
The nightly will point to the latest main, tlcpack will point to a stable build hashtag defined in common/sync_package.py
- Build tlcpack manylinux wheels.
./docker/bash.sh [docker-image] ./wheel/build_wheel_manylinux.sh --cuda none
To build wheels for a specific CUDA version, for example, CUDA 11.1, run
./docker/bash.sh [docker-image] ./wheel/build_wheel_manylinux.sh --cuda 11.1
The docker image is built in step 1 and needs to match the cuda version.
- Get the wheels
The wheels are now available in
./tvm/python/repaired_wheels