tree: d2d9d7c2b74a851b576a1ea78f80926d6352be3e
  1. build-environment.yaml
  2. manylinux_build_libtvm_runtime_cuda.sh
  3. README.md
  4. windows_build_libtvm_runtime_cuda.bat
ci/scripts/package/README.md

TVM wheel packaging

The wheels are built by a standard cibuildwheel flow, configured in .github/workflows/publish_wheel.yml and pyproject.toml ([tool.cibuildwheel] and [tool.scikit-build]). This directory holds the few helper scripts that flow invokes:

  • manylinux_build_libtvm_runtime_cuda.sh — run by the build_cuda_runtime CI stage; builds the libtvm_runtime_cuda.so sidecar inside the prebuilt quay.io/manylinux_cuda image (CUDA toolkit preinstalled).
  • windows_build_libtvm_runtime_cuda.bat — the Windows equivalent (run with shell: cmd); installs the CUDA toolkit via conda and builds tvm_runtime_cuda.dll.
  • build-environment.yaml — conda environment for building the wheel.