diff --git a/hpvm/lib/Transforms/CMakeLists.txt b/hpvm/lib/Transforms/CMakeLists.txt index b18cd4551ba33e0c315a416164b45e6282098aeb..0f8d0c64aa5979d866b35e4e1047e8c496ed3677 100644 --- a/hpvm/lib/Transforms/CMakeLists.txt +++ b/hpvm/lib/Transforms/CMakeLists.txt @@ -8,3 +8,4 @@ add_subdirectory(DFG2LLVM_WrapperAPI) add_subdirectory(DFG2LLVM_CUDNN) add_subdirectory(FuseHPVMTensorNodes) add_subdirectory(InPlaceDFG) +add_subdirectory(DFG2LLVM_EPOCHS) diff --git a/hpvm/set_paths.sh b/hpvm/set_paths.sh index d23cf03046161e4fc2916954202e91a6c1eb0788..b0f8f4c7b386050b43c54dd3757ec0acb17c79cd 100644 --- a/hpvm/set_paths.sh +++ b/hpvm/set_paths.sh @@ -18,3 +18,17 @@ else export LD_LIBRARY_PATH=$CUDA_LIB_PATH:$LD_LIBRARY_PATH echo "Finished setting environment paths!" fi + + +SH="$(readlink -f /proc/$$/exe)" +if [[ "$SH" == "/bin/zsh" ]]; then + DIR="${0:A:h}" +else + DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )" +fi +echo "Adding to PATH: $DIR/build/bin" +export PATH=$DIR/build/bin:$PATH +echo "Exporting HPVM_ROOT = $DIR" +export HPVM_ROOT=$DIR +echo "Exporting LLVM_SRC_ROOT = $HPVM_ROOT/llvm" +export LLVM_SRC_ROOT=$HPVM_ROOT/llvm