install.sh

#!/usr/bin/env bash

export CC=
export CXX=

#compiling option

intel=$1    # compiler icc/gcc
avx512=$2   # AVX512F on/off 
#omp=$3      # intel intel/gnu
use_mklml=$3      # intel intel/gnu
skip=$4     # skip openblas, value = noskip/skip, default is noskip
if [[ $skip == 'skip' ]]; then
   echo "skip openblas check"
else
   #identify user level, and check openblas.For root user, check /opt/OpenBLAS, for none root user, check ~/OpenBLAS
   if [ "$(ls -A /opt/OpenBLAS)" ]; then
      echo "OpenBLAS is installed in /opt"
   else
      if [[ $EUID -ne 0 ]]; then
         #none root user
         RTN=$(echo $LD_LIBRARY_PATH |grep -i openblas)
         if [ -z "$RTN" ]; then
            echo "OpenBLAS is not installed, or environment is not set"
            bash install-openblas.sh
         else
            echo "OpenBLAS installed."
   
         fi
      else
         #root user
         echo "ERROR: please instal OpenBLAS: bash install-deps , or apt-get install -y libopenblas-dev "
      fi
   fi
fi


FORCE_AVX512_v=OFF
if [[ $avx512 == 'avx512' ]]; then
  FORCE_AVX512_v=ON
else
  FORCE_AVX512_v=OFF
fi

USE_MKLML_v=OFF
if [[ $use_mklml == 'mklml' ]]; then
  USE_MKLML_v=ON
else
  USE_MKLML_v=OFF
fi

WITH_IOMP_v=ON
:<<hallo
if [[ $omp == 'iomp' ]]; then
  WITH_IOMP_v=ON
elif [[ $omp == 'gomp' ]]; then 
  WITH_IOMP_v=OFF
fi
hallo

SKIP_R=0
BATCH_INSTALL=0


THIS_DIR=$(cd $(dirname $0); pwd)
if [[ "$THIS_DIR" == *" "* ]]; then
    echo "$THIS_DIR: Torch cannot install to a path containing whitespace.
Please try a different path, one without any spaces."
    exit 1
fi
PREFIX=${PREFIX:-"${THIS_DIR}/install"}
TORCH_LUA_VERSION=${TORCH_LUA_VERSION:-"LUAJIT21"} # by default install LUAJIT21

while getopts 'bsh' x; do
    case "$x" in
        h)
            echo "usage: $0
This script will install Torch and related, useful packages into $PREFIX.

    -b      Run without requesting any user input (will automatically add PATH to shell profile)
    -s      Skip adding the PATH to shell profile
"
            exit 2
            ;;
        b)
            BATCH_INSTALL=1
            ;;
        s)
            SKIP_RC=1
            ;;
    esac
done


# Scrub an anaconda/conda install, if exists, from the PATH.
# It has a malformed MKL library (as of 1/17/2015)
OLDPATH=$PATH
if [[ $(echo $PATH | grep conda) ]]; then
    export PATH=$(echo $PATH | tr ':' '\n' | grep -v "conda[2-9]\?" | uniq | tr '\n' ':')
fi

echo "Prefix set to $PREFIX"

if [[ `uname` == 'Linux' ]]; then
    ICC_ON=0
    if [[ $intel == 'icc' ]]; then
        if [[ `which icc` == '' ]]; then
            CC=gcc
        else
            CC=icc
        fi    
    
        if [[ `which icpc` == '' ]]; then
            CXX=g++
        else
            CXX=icpc
        fi   
        if [[ `echo $CC | grep icc` != '' ]]; then
            ICC_ON=1
        fi
    else
        ICC_ON=0
    fi
    
    if [[ $ICC_ON == 1 ]]; then
       echo 'using Intel compiler to compile torch'
    else
       echo 'using GNU compilter to compile torch'
    fi

    #`./prepare_mklml.sh $ICC_ON`
    RETURN_STRING=`./prepare_mklml.sh $ICC_ON $USE_MKLML_v`
    export MKLML_ROOT=`echo $RETURN_STRING | awk '{print $1}'`
    if [ $USE_MKLML_v == ON ]; then
      echo "Path of MKLML_ROOT is:"
    else
      echo "Path of MKL_ROOT is:"
    fi
    echo $MKLML_ROOT
    MKLML_LIBRARY_PATH=$MKLROOT/lib
    MKLML_INCLUDE_PATH=$MKLROOT/include
    
    export CMAKE_LIBRARY_PATH=/opt/OpenBLAS/include:/opt/OpenBLAS/lib:$MKLML_LIBRARY_PATH:$CMAKE_LIBRARY_PATH
    export CMAKE_INCLUDE_PATH=/opt/OpenBLAS/include:$MKLML_INCLUDE_PATH:$CMAKE_INCLUDE_PATH
    export CMAKE_LIBRARY_PATH=$PREFIX/include:/opt/OpenBLAS/include:$PREFIX/lib:/opt/OpenBLAS/lib:$CMAKE_LIBRARY_PATH
    export CMAKE_LIBRARY_PATH=~/OpenBLAS/include:~/OpenBLAS/lib:$CMAKE_LIBRARY_PATH
    export LD_LIBRARY_PATH=~/OpenBLAS/include:~/OpenBLAS/lib:$LD_LIBRARY_PATH

fi

export CMAKE_PREFIX_PATH=$PREFIX
if [[ $intel == 'gcc' ]]; then
  export CC=`which gcc`
  export CXX=`which g++`
fi

git submodule update --init --recursive

# If we're on OS X, use clang
if [[ `uname` == "Darwin" ]]; then
    # make sure that we build with Clang. CUDA's compiler nvcc
    # does not play nice with any recent GCC version.
    export CC=clang
    export CXX=clang++
fi
# If we're on Arch linux, use gcc v5
if [[ `uname -a` == *"ARCH"* ]]; then
    path_to_gcc5=$(which gcc-5)
    if [ -x "$path_to_gcc5" ]; then
      export CC="$path_to_gcc5"
    else
      echo "Warning: GCC v5 not found. CUDA v8 is incompatible with GCC v6, if installation fails, consider running \$ pacman -S gcc5"
    fi
fi

echo "Installing Lua version: ${TORCH_LUA_VERSION}"
mkdir -p install
mkdir -p build
cd build
cmake .. -DCMAKE_INSTALL_PREFIX="${PREFIX}" -DCMAKE_BUILD_TYPE=Release -DWITH_${TORCH_LUA_VERSION}=ON 2>&1 >>$PREFIX/install.log || exit 1
(make 2>&1 >>$PREFIX/install.log  || exit 1) && (make install 2>&1 >>$PREFIX/install.log || exit 1)
cd ..

# Check for a CUDA install (using nvcc instead of nvidia-smi for cross-platform compatibility)
path_to_nvcc=$(which nvcc)
if [ $? == 1 ]; then { # look for it in /usr/local
  if [[ -f /usr/local/cuda/bin/nvcc ]]; then {
    path_to_nvcc=/usr/local/cuda/bin/nvcc
  } 
  elif [[ -f /opt/cuda/bin/nvcc ]]; then { # default path for arch
    path_to_nvcc=/opt/cuda/bin/nvcc
  } fi
} fi

# check if we are on mac and fix RPATH for local install
path_to_install_name_tool=$(which install_name_tool 2>/dev/null)
if [ -x "$path_to_install_name_tool" ]
then
   if [ ${TORCH_LUA_VERSION} == "LUAJIT21" ] || [ ${TORCH_LUA_VERSION} == "LUAJIT20" ] ; then
       install_name_tool -id ${PREFIX}/lib/libluajit.dylib ${PREFIX}/lib/libluajit.dylib
   else
       install_name_tool -id ${PREFIX}/lib/liblua.dylib ${PREFIX}/lib/liblua.dylib
   fi
fi

if [ -x "$path_to_nvcc" ] || [ -x "$path_to_nvidiasmi" ]
then
    echo "Found CUDA on your machine. Installing CMake 3.6 modules to get up-to-date FindCUDA"
    cd ${THIS_DIR}/cmake/3.6 && \
(cmake -E make_directory build && cd build && cmake .. -DCMAKE_INSTALL_PREFIX="${PREFIX}" \
        && make install) && echo "FindCuda bits of CMake 3.6 installed" || exit 1
fi

setup_lua_env_cmd=$($PREFIX/bin/luarocks path)
eval "$setup_lua_env_cmd"

echo "$CC"
echo "Installing common Lua packages"
cd ${THIS_DIR}/extra/luafilesystem && $PREFIX/bin/luarocks make rockspecs/luafilesystem-1.6.3-1.rockspec || exit 1
cd ${THIS_DIR}/extra/penlight && $PREFIX/bin/luarocks make penlight-scm-1.rockspec || exit 1
cd ${THIS_DIR}/extra/lua-cjson && $PREFIX/bin/luarocks make lua-cjson-2.1devel-1.rockspec || exit 1

echo "Installing core Torch packages"
cd ${THIS_DIR}/extra/luaffifb && $PREFIX/bin/luarocks make luaffi-scm-1.rockspec       || exit 1
cd ${THIS_DIR}/pkg/sundown   && $PREFIX/bin/luarocks make rocks/sundown-scm-1.rockspec || exit 1
cd ${THIS_DIR}/pkg/cwrap        && $PREFIX/bin/luarocks make rocks/cwrap-scm-1.rockspec   || exit 1
cd ${THIS_DIR}/pkg/paths        && $PREFIX/bin/luarocks make rocks/paths-scm-1.rockspec   || exit 1
cd ${THIS_DIR}/pkg/torch        && $PREFIX/bin/luarocks make WITH_IOMP=$WITH_IOMP_v FORCE_AVX512=$FORCE_AVX512_v USE_MKLML=$USE_MKLML_v rocks/torch-scm-1.rockspec   || exit 1
cd ${THIS_DIR}/pkg/dok          && $PREFIX/bin/luarocks make rocks/dok-scm-1.rockspec     || exit 1
cd ${THIS_DIR}/exe/trepl        && $PREFIX/bin/luarocks make trepl-scm-1.rockspec         || exit 1
cd ${THIS_DIR}/pkg/sys          && $PREFIX/bin/luarocks make sys-1.1-0.rockspec           || exit 1
cd ${THIS_DIR}/pkg/xlua         && $PREFIX/bin/luarocks make xlua-1.0-0.rockspec          || exit 1
cd ${THIS_DIR}/extra/nn         && $PREFIX/bin/luarocks make FORCE_AVX512=$FORCE_AVX512_v rocks/nn-scm-1.rockspec      || exit 1
cd ${THIS_DIR}/extra/mkltorch   && $PREFIX/bin/luarocks make FORCE_AVX512=$FORCE_AVX512_v mkltorch-scm-1.rockspec      || exit 1
cd ${THIS_DIR}/extra/mklnn      && $PREFIX/bin/luarocks make FORCE_AVX512=$FORCE_AVX512_v mklnn-scm-1.rockspec      || exit 1
cd ${THIS_DIR}/extra/graph      && $PREFIX/bin/luarocks make rocks/graph-scm-1.rockspec   || exit 1
cd ${THIS_DIR}/extra/nngraph    && $PREFIX/bin/luarocks make nngraph-scm-1.rockspec       || exit 1
cd ${THIS_DIR}/pkg/image        && $PREFIX/bin/luarocks make image-1.1.alpha-0.rockspec   || exit 1
cd ${THIS_DIR}/pkg/optim        && $PREFIX/bin/luarocks make optim-1.0.5-0.rockspec       || exit 1

if [ -x "$path_to_nvcc" ]
then
    echo "Found CUDA on your machine. Installing CUDA packages"
    cd ${THIS_DIR}/extra/cutorch && $PREFIX/bin/luarocks make rocks/cutorch-scm-1.rockspec || exit 1
    cd ${THIS_DIR}/extra/cunn    && $PREFIX/bin/luarocks make rocks/cunn-scm-1.rockspec    || exit 1
fi

# Optional packages
echo "Installing optional Torch packages"
cd ${THIS_DIR}/pkg/gnuplot          && $PREFIX/bin/luarocks make rocks/gnuplot-scm-1.rockspec
cd ${THIS_DIR}/exe/env              && $PREFIX/bin/luarocks make env-scm-1.rockspec
cd ${THIS_DIR}/extra/nnx            && $PREFIX/bin/luarocks make nnx-0.1-1.rockspec
cd ${THIS_DIR}/exe/qtlua            && $PREFIX/bin/luarocks make rocks/qtlua-scm-1.rockspec
cd ${THIS_DIR}/pkg/qttorch          && $PREFIX/bin/luarocks make rocks/qttorch-scm-1.rockspec
cd ${THIS_DIR}/extra/threads        && $PREFIX/bin/luarocks make rocks/threads-scm-1.rockspec
cd ${THIS_DIR}/extra/argcheck       && $PREFIX/bin/luarocks make rocks/argcheck-scm-1.rockspec

# Optional CUDA packages
if [ -x "$path_to_nvcc" ]
then
    echo "Found CUDA on your machine. Installing optional CUDA packages"
    cd ${THIS_DIR}/extra/cudnn   && $PREFIX/bin/luarocks make cudnn-scm-1.rockspec
fi

export PATH=$OLDPATH # Restore anaconda distribution if we took it out.

# Add C libs to LUA_CPATH
if [[ `uname` == "Darwin" ]]; then
    CLIB_LUA_CPATH=$PREFIX/lib/?.dylib
else
    CLIB_LUA_CPATH=$PREFIX/lib/?.so
fi

cat <<EOF >$PREFIX/bin/torch-activate
$setup_lua_env_cmd
export PATH=$PREFIX/bin:\$PATH
export LD_LIBRARY_PATH=$PREFIX/lib:\$LD_LIBRARY_PATH
export DYLD_LIBRARY_PATH=$PREFIX/lib:\$DYLD_LIBRARY_PATH
export LUA_CPATH='$CLIB_LUA_CPATH;'\$LUA_CPATH
EOF
chmod +x $PREFIX/bin/torch-activate

if [[ $SKIP_RC == 1 ]]; then
  exit 0
fi

RC_FILE=0
DEFAULT=yes
if [[ $(echo $SHELL | grep bash) ]]; then
    RC_FILE=$HOME/.bashrc
elif [[ $(echo $SHELL | grep zsh) ]]; then
    RC_FILE=$HOME/.zshrc
else
    echo "

Non-standard shell $SHELL detected. You might want to
add the following lines to your shell profile:

. $PREFIX/bin/torch-activate
"
fi

if [[ $skip == 'skip' ]]; then
   BATCH_INSTALL=1
   RC_FILE=
fi

WRITE_PATH_TO_PROFILE=0
if [[ $BATCH_INSTALL == 0 ]]; then
    if [ -f "$RC_FILE" ]; then
        echo "

Do you want to automatically prepend the Torch install location
to PATH and LD_LIBRARY_PATH in your $RC_FILE? (yes/no)
[$DEFAULT] >>> "
        read input
        if [[ $input == "" ]]; then
            input=$DEFAULT
        fi

        is_yes() {
            yesses={y,Y,yes,Yes,YES}
            if [[ $yesses =~ $1 ]]; then
                echo 1
            fi
        }

        if [[ $(is_yes $input) ]]; then
            WRITE_PATH_TO_PROFILE=1
        fi
    fi
else
    if [[ "$RC_FILE" ]]; then
        WRITE_PATH_TO_PROFILE=1
    fi
fi

WRITE_PATH_TO_PROFILE=0
if [[ $WRITE_PATH_TO_PROFILE == 1 ]]; then
    echo "

. $PREFIX/bin/torch-activate" >> "$RC_FILE"
    echo "

. $PREFIX/bin/torch-activate" >> "$HOME"/.profile

else
    echo "

Not updating your shell profile.
You might want to
add the following lines to your shell profile:

. $PREFIX/bin/torch-activate
"
fi