forked from onnx/onnx-coreml
-
Notifications
You must be signed in to change notification settings - Fork 0
/
install-develop.sh
executable file
·60 lines (48 loc) · 1.36 KB
/
install-develop.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#!/bin/bash
set -ex
# realpath might not be available on MacOS
script_path=$(python -c "import os; import sys; print(os.path.realpath(sys.argv[1]))" "${BASH_SOURCE[0]}")
top_dir=$(dirname "$script_path")
REPOS_DIR="$top_dir/third_party"
BUILD_DIR="$top_dir/build"
_check_submodule_present() {
if [ ! -f "$REPOS_DIR/$@/setup.py" ]; then
echo Didn\'t find $@ submodule. Please run: git submodule update --recursive --init
exit 1
fi
}
_check_submodule_present onnx
_check_compilers_use_ccache() {
COMPILERS_WITHOUT_CCACHE=""
for compiler in gcc g++ cc c++; do
if ! readlink $(which $compiler) | grep ccache; then
COMPILERS_WITHOUT_CCACHE="$COMPILERS_WITHOUT_CCACHE $compiler"
fi
done
if [ "$COMPILERS_WITHOUT_CCACHE" != "" ]; then
echo Warning: Compilers not set up for ccache: $COMPILERS_WITHOUT_CCACHE. Incremental builds will be slow.
read -p "Press enter to continue"
fi
}
_check_compilers_use_ccache
mkdir -p "$BUILD_DIR"
_pip_install() {
if [[ -n "$CI" ]]; then
ccache -z
fi
if [[ -n "$CI" ]]; then
time pip install "$@"
else
pip install "$@"
fi
if [[ -n "$CI" ]]; then
ccache -s
fi
}
# Install onnx
# _pip_install -e "$REPOS_DIR/onnx"
cd "$REPOS_DIR/onnx"
python setup.py install
cd -
# Install onnx-coreml
_pip_install -e .[mypy]