intel-extension-for-pytorch

1
#!/bin/bash
2

3
MODE=""
4
LIBTORCH_PATH=""
5
IPEX_VERSION="<IPEX_VERSION>"
6

7
if [ $# == 2 ]; then
8
    MODE=$1
9
    LIBTORCH_PATH=$2
10
else
11
    echo "Usage: bash $0 [install|uninstall] <libtorch_path>"
12
    exit 1
13
fi
14

15
if [[ ! ${MODE} == "install" ]] && [[ ! ${MODE} == "uninstall" ]]; then
16
    echo "Unrecognized execution mode: ${MODE}"
17
    exit 2
18
fi
19

20
if [ ! -f ${LIBTORCH_PATH}/build-version ]; then
21
    echo "${LIBTORCH_PATH} doesn't seem to be directory of libtorch."
22
    exit 3
23
fi
24

25
if [[ ! ${LIBTORCH_PATH:0:1} == '/' ]] && [[ ! ${LIBTORCH_PATH:0:1} == '~' ]]; then
26
    LIBTORCH_PATH=`pwd`/${LIBTORCH_PATH}
27
fi
28

29
mkdir ipex_installer
30
sed -e '1,/^exit$/d' "$0" | tar xzf - -C ipex_installer && cd ipex_installer && bash ./install_c++_sdk.sh ${MODE} ${LIBTORCH_PATH} ${IPEX_VERSION} && cd .. && rm -rf ipex_installer
31
exit
32