diff --git a/hpvm/projects/predtuner b/hpvm/projects/predtuner index a149e365170263666db764664ad8ed6b03f258d3..d65bd5c8b774a3b90e983aba7763342e38757421 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 b1ab7ec2ab83487731043a8519c7dd3eb97865f2..644ce0a2176a4e9a257c6618761eb75da34c1697 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(