diff --git a/hpvm/scripts/llvm_installer.sh b/hpvm/scripts/llvm_installer.sh index 13184fd7f773fc257e72e250e81b2e2591aaab36..21ed6ee6d13ef83e0cc62f643d8e674e7c0e5a90 100755 --- a/hpvm/scripts/llvm_installer.sh +++ b/hpvm/scripts/llvm_installer.sh @@ -32,6 +32,7 @@ HPVM_RT=hpvm-rt/hpvm-rt.bc TARGET=all TARGET_INPUT=all FLAGGED=false +DOWNLOAD_WEIGHTS=false # Get flags while getopts 'hmj:t:' opt; do @@ -83,6 +84,7 @@ if $FLAGGED; then echo Automated: $AUTOMATE echo Threads: $NUM_THREADS echo Targets: $TARGET + echo Download Weights: $DOWNLOAD_WEIGHTS echo else echo "No Flags found. Using command line prompts." @@ -122,11 +124,22 @@ else TARGET=$TARGET_INPUT fi echo + + read_yn "Download weights necessary to run DNN benchmarks?" LOAD_WEIGHTS + if [[ $LOAD_WEIGHTS == "" ]]; then + echo "No input given. Weights will not be downloaded." + elif [[ $LOAD_WEIGHTS == "n" ]]; then + echo "Weights will not be downloaded." + else + DOWNLOAD_WEIGHTS=$LOAD_WEIGHTS + fi + echo echo "Running with the following options:" echo Automated: $AUTOMATE echo Threads: $NUM_THREADS echo Targets: $TARGET + echo Download Weights: $DOWNLOAD_WEIGHTS echo fi @@ -239,8 +252,11 @@ echo make -j$NUM_THREADS make -j$NUM_THREADS #make install -read_yn "Download weights necessary to run DNN benchmarks?" DOWNLOAD_WEIGHTS if [ $DOWNLOAD_WEIGHTS == "y" ]; then + echo + echo "Downloading weights for DNN benchmarks..." + echo + # First get hands on gdown -- google drive downloader wget https://raw.githubusercontent.com/circulosmeos/gdown.pl/master/gdown.pl -O gdown.pl chmod +x ./gdown.pl