From 20677b4fb3284d6f709f9c905b0cb57fd298960b Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Wed, 24 Mar 2021 18:01:14 -0500
Subject: [PATCH] Adding -h help for HPVM_binary
---
.../keras/keras_frontend/hpvm_dfg_translator.py | 11 +++++++----
1 file changed, 7 insertions(+), 4 deletions(-)
diff --git a/hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py b/hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py
index 08dd896478..fbe2f41c84 100644
--- a/hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py
+++ b/hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py
@@ -708,7 +708,7 @@ if(config_path != ""){
llvm_hpvm_initializeRuntimeController(config_path.c_str());
}
- """
+ """
main_func_str += "RootIn* args = static_cast<RootIn*>(malloc(sizeof(RootIn))); \n\n"
@@ -767,8 +767,8 @@ if(config_path != ""){
if(config_path != ""){
llvm_hpvm_initializeRuntimeController(config_path.c_str());
}
-
- """
+
+"""
tuner_main_func_str += "std::string input_pth = (ret == 1 ? test_input_path : tune_input_path); \n"
tuner_main_func_str += "std::string labels_pth = (ret == 1 ? test_labels_path : tune_labels_path); \n"
@@ -883,7 +883,7 @@ void printUsage(){
std::string runtype;
std::string config_path = "";
int flag;
- while ( (flag = getopt (argc, argv, "d:c:")) != -1){
+ while ( (flag = getopt (argc, argv, "hd:c:")) != -1){
switch (flag)
{
case 'd':
@@ -894,6 +894,9 @@ void printUsage(){
case 'c':
config_path = std::string(optarg);
break;
+ case 'h':
+ printUsage();
+ break;
default:
printUsage();
}
--
GitLab