diff --git a/hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py b/hpvm/projects/keras/keras_frontend/hpvm_dfg_translator.py index 08dd896478684ab0a1955c16463d6b75bb3ae78a..fbe2f41c84c2d018412a682b3c211ad9897dcd5f 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(); }