mirror of https://github.com/llvm/torch-mlir
34 lines
975 B
Bash
34 lines
975 B
Bash
|
#!/bin/bash
|
||
|
|
set -e
|
||
|
|
|
||
|
|
if [ -z "$PYTHON" ]; then
|
||
|
|
PYTHON="$(which python)"
|
||
|
|
fi
|
||
|
|
version="$("$PYTHON" --version)"
|
||
|
|
echo "Using python: $PYTHON (version $version)"
|
||
|
|
|
||
|
|
repo_root="$(cd $(dirname $0)/.. && pwd)"
|
||
|
|
wheelhouse="$repo_root/wheelhouse"
|
||
|
|
mkdir -p $wheelhouse
|
||
|
|
cd $wheelhouse
|
||
|
|
|
||
|
|
echo "---- BUILDING npcomp-core ----"
|
||
|
|
CMAKE_GENERATOR=Ninja CMAKE_C_COMPILER_LAUNCHER=ccache CMAKE_CXX_COMPILER_LAUNCHER=ccache \
|
||
|
|
$PYTHON -m pip wheel "${repo_root}" \
|
||
|
|
--use-feature=in-tree-build \
|
||
|
|
-w "$wheelhouse" -v
|
||
|
|
|
||
|
|
echo "---- INSTALLING npcomp-core ----"
|
||
|
|
$PYTHON -m pip install -f "$wheelhouse" --force-reinstall npcomp-core
|
||
|
|
|
||
|
|
echo "---- BUILDING npcomp-torch ----"
|
||
|
|
$PYTHON -m pip wheel "${repo_root}/frontends/pytorch" \
|
||
|
|
--use-feature=in-tree-build \
|
||
|
|
-w "$wheelhouse" -v
|
||
|
|
|
||
|
|
echo "---- INSTALLING npcomp-torch ----"
|
||
|
|
$PYTHON -m pip install -f "$wheelhouse" --force-reinstall npcomp-torch
|
||
|
|
|
||
|
|
echo "---- QUICK SMOKE TEST ----"
|
||
|
|
$PYTHON $repo_root/frontends/pytorch/test/torchscript_e2e_test/basic.py
|