tlcpack

Tensor learning compiler binary distribution package.

Github Actions

We use github action to build wheel and conda packages nightly.

Checkout .github/workflows

Build Process

1. Build docker images

./docker/build_image.sh <CONTAINER_TYPE>

CONTAINER_NAME: Type of the docker container used to build wheels, e.g., (cpu|cpu_aarch64|cu100|cu101|cu102)

2. 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

3. 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.

4. Get the wheels

The wheels are now available in

./tvm/python/repaired_wheels