From 8563b2ec26dc123f2940e04ba1de5d92b47e032a Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Thu, 8 Jul 2021 13:21:34 -0500
Subject: [PATCH] No timeout in ssh tests
---
hpvm/test/epoch_dnn/main.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/hpvm/test/epoch_dnn/main.py b/hpvm/test/epoch_dnn/main.py
index 2a66aae200..0397a2d1f9 100644
--- a/hpvm/test/epoch_dnn/main.py
+++ b/hpvm/test/epoch_dnn/main.py
@@ -38,7 +38,7 @@ def run_test_over_ssh(host: str, password: str, working_dir: str, image_dir: Pat
print(f"Running test on remote host {host}...")
args = options.split(" ") + [host]
- child = pexpect.spawn("ssh", args, timeout=60)
+ child = pexpect.spawn("ssh", args, timeout=None)
child.expect(r"password:")
child.sendline(password)
child.expect("# ") # The bash prompt
@@ -49,7 +49,7 @@ def run_test_over_ssh(host: str, password: str, working_dir: str, image_dir: Pat
remote_path = f"{image_dir.name}/{image.name}"
print(f"Sending {image.name} to run")
child.sendline(f"./nvdla_runtime --loadable {BUFFER_NAME} --image {remote_path} --rawdump")
- child.expect("# ")
+ child.expect("Test pass")
child.sendline("cat output.dimg")
child.expect("# ")
result_lines = child.before.decode().splitlines()
--
GitLab