Python packaging scripts.

Note that packages will be placed in bindings/python/packaging/dist with the canonical instructions. However, the setup scripts can be run from anywhere and will create build and dist directories where run. Wheels can be installed with pip3 install --user dist/*.whl.

Building core wheels with CMake

Most of IREE is built/packaged with CMake. For the parts that build with CMake, this is preferred.

Canonical instructions follow:

Linux

export LDFLAGS=-fuse-ld=/usr/bin/ld.lld-10
export PYIREE_CMAKE_BUILD_ROOT=$HOME/build-iree-release
export IREE_SRC=$HOME/src/iree
rm -Rf $CMAKE_BUILD_ROOT; mkdir -p $CMAKE_BUILD_ROOT
cmake -GNinja -B$CMAKE_BUILD_ROOT -H$IREE_SRC \
  -DCMAKE_BUILD_TYPE=Release \
  -DCMAKE_C_COMPILER=clang -DCMAKE_CXX_COMPILER=clang++ \
  -DIREE_BUILD_PYTHON_BINDINGS=ON -DIREE_BUILD_SAMPLES=OFF
(cd $CMAKE_BUILD_ROOT && ninja)
(cd $IREE_SRC/bindings/python/packaging && (
rm -Rf build;
python3 setup_compiler.py bdist_wheel;
rm -Rf build;
python3 setup_rt.py bdist_wheel))

Building IREE/TensorFlow wheels

If building TensorFlow integration wheels, then this must be done via Bazel. In this case, it can be easiest to just package everything from a Bazel build to avoid multiple steps.

Canonical instructions follow:

Env Setup

IREE_SRC=$HOME/src/iree
export PYIREE_BAZEL_BUILD_ROOT="$IREE_SRC/bazel-bin"
if which cygpath; then
  export PYIREE_BAZEL_BUILD_ROOT="$(cygpath -w "$PYIREE_BAZEL_BUILD_ROOT")"
fi

Building:

cd $IREE_SRC
bazel build -c opt \
  //bindings/python/packaging:all_pyiree_packages

Packaging

(cd $IREE_SRC/bindings/python/packaging && (
rm -Rf build;
python3 setup_tf.py bdist_wheel))

(cd $IREE_SRC/bindings/python/packaging && (
rm -Rf build;
python3 setup_compiler.py bdist_wheel))

(cd $IREE_SRC/bindings/python/packaging && (
rm -Rf build;
python3 setup_rt.py bdist_wheel))