From a1a55199d9a2c27bb70127385384d4c8468b940c Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Thu, 8 Jul 2021 19:42:27 -0500
Subject: [PATCH] Adding optional working dir to main.py

---
 hpvm/test/epoch_dnn/main.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/hpvm/test/epoch_dnn/main.py b/hpvm/test/epoch_dnn/main.py
index fe4408ef69..e94b13f48c 100644
--- a/hpvm/test/epoch_dnn/main.py
+++ b/hpvm/test/epoch_dnn/main.py
@@ -70,6 +70,9 @@ SCP_HOST = "root@espgate.cs.columbia.edu"
 SCP_PWD = "openesp"
 SCP_DST = "~/NV_NVDLA"
 
+if len(sys.argv) > 1:
+    WORKING_DIR = Path(sys.argv[1])
+
 # Reproducibility
 np.random.seed(42)
 
-- 
GitLab