From 4014c9f58af82007b3dbba520cec0b61892c6b07 Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Sun, 4 Apr 2021 00:48:19 -0500
Subject: [PATCH] Bump predtuner version (bugfix) and changes to test script
---
hpvm/projects/predtuner | 2 +-
hpvm/test/dnn_benchmarks/pytorch/test_tuning/test_tuning.py | 5 ++---
2 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/hpvm/projects/predtuner b/hpvm/projects/predtuner
index a149e36517..d65bd5c8b7 160000
--- a/hpvm/projects/predtuner
+++ b/hpvm/projects/predtuner
@@ -1 +1 @@
-Subproject commit a149e365170263666db764664ad8ed6b03f258d3
+Subproject commit d65bd5c8b774a3b90e983aba7763342e38757421
diff --git a/hpvm/test/dnn_benchmarks/pytorch/test_tuning/test_tuning.py b/hpvm/test/dnn_benchmarks/pytorch/test_tuning/test_tuning.py
index b1ab7ec2ab..644ce0a217 100755
--- a/hpvm/test/dnn_benchmarks/pytorch/test_tuning/test_tuning.py
+++ b/hpvm/test/dnn_benchmarks/pytorch/test_tuning/test_tuning.py
@@ -13,8 +13,8 @@ from torch.nn import Module
site.addsitedir(Path(__file__).parent.parent.absolute())
import dnn
-# Set up logger to put log file in /tmp
-msg_logger = config_pylogger(output_dir="/tmp", verbose=True)
+# Set up logger
+msg_logger = config_pylogger(output_dir=".", verbose=True)
benchmarks = {
"lenet_mnist": (dnn.LeNet, 1, 28, 1000),
"alexnet_imagenet": (dnn.AlexNetImageNet, 3, 224, 100),
@@ -59,7 +59,6 @@ def main():
binary_file, metadata_file = generate(
model_cls, nch, img_size, batch_size, netname
)
- print(Path.cwd(), binary_file, metadata_file)
app = PipedBinaryApp("test", binary_file, metadata_file)
tuner = app.get_tuner()
tuner.tune(
--
GitLab