From c1bcf681fd20bbc69dfc75b5c111e1e1a904d8ca Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Wed, 27 Jan 2021 00:58:26 -0600
Subject: [PATCH] Added script to just download weights

---
 hpvm/scripts/download_weights.sh | 10 ++++++++++
 hpvm/scripts/llvm_installer.sh   | 12 +-----------
 2 files changed, 11 insertions(+), 11 deletions(-)
 create mode 100755 hpvm/scripts/download_weights.sh

diff --git a/hpvm/scripts/download_weights.sh b/hpvm/scripts/download_weights.sh
new file mode 100755
index 0000000000..757abbf3b6
--- /dev/null
+++ b/hpvm/scripts/download_weights.sh
@@ -0,0 +1,10 @@
+#!/bin/bash
+# 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
+# Download the zip file from google drive
+./gdown.pl 'https://drive.google.com/file/d/1V_yd9sKcZQ7zhnO5YhRpOsaBPLEEvM9u' model_params.zip
+unzip model_params.zip # should give a "model_params" folder
+# All our benchmarks also know to look for parameters in <build_dir>/model_params.
+# Cleanup:
+rm gdown.pl model_params.zip
\ No newline at end of file
diff --git a/hpvm/scripts/llvm_installer.sh b/hpvm/scripts/llvm_installer.sh
index 21ed6ee6d1..0cbaea8e49 100755
--- a/hpvm/scripts/llvm_installer.sh
+++ b/hpvm/scripts/llvm_installer.sh
@@ -256,17 +256,7 @@ 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
-  # Download the zip file from google drive
-  ./gdown.pl 'https://drive.google.com/file/d/1V_yd9sKcZQ7zhnO5YhRpOsaBPLEEvM9u' model_params.zip
-  unzip model_params.zip # should give a "model_params" folder
-  mv model_params $BUILD_DIR
-  # All our benchmarks also know to look for parameters in <build_dir>/model_params.
-  # Cleanup:
-  rm gdown.pl model_params.zip
+  ../scripts/download_weights.sh
 else
   echo "Skipping weight download"
 fi
-- 
GitLab