diff --git a/mace/benchmark/benchmark_model.cc b/mace/benchmark/benchmark_model.cc index 7317b8bef6447eb64b6c3ddb55ad0e4509abfc29..b6cd2d699ed92c21dd39742cdeb6b5c906159b7f 100644 --- a/mace/benchmark/benchmark_model.cc +++ b/mace/benchmark/benchmark_model.cc @@ -201,7 +201,9 @@ DEFINE_int32(cpu_affinity_policy, 1, int Main(int argc, char **argv) { MACE_CHECK(FLAGS_device != "HEXAGON", "Model benchmark tool do not support DSP."); - gflags::SetUsageMessage("some usage message"); + std::string usage = "benchmark model\nusage: " + std::string(argv[0]) + + " [flags]"; + gflags::SetUsageMessage(usage); gflags::ParseCommandLineFlags(&argc, &argv, true); LOG(INFO) << "Model name: [" << FLAGS_model_name << "]"; diff --git a/mace/benchmark/model_throughput_test.cc b/mace/benchmark/model_throughput_test.cc index fd19be5ca7063d2a7a19e1d61423ef5cb517a1a3..dce0a73616b59fdb1b70cb797bc7bcad26fc3c4c 100644 --- a/mace/benchmark/model_throughput_test.cc +++ b/mace/benchmark/model_throughput_test.cc @@ -163,7 +163,9 @@ DEFINE_string(dsp_model_data_file, "", "dsp model data file name"); DEFINE_int32(run_seconds, 10, "run seconds"); int Main(int argc, char **argv) { - gflags::SetUsageMessage("some usage message"); + std::string usage = "model throughput test\nusage: " + std::string(argv[0]) + + " [flags]"; + gflags::SetUsageMessage(usage); gflags::ParseCommandLineFlags(&argc, &argv, true); LOG(INFO) << "mace version: " << MaceVersion(); diff --git a/mace/core/testing/test_benchmark_main.cc b/mace/core/testing/test_benchmark_main.cc index a2dccb65437b7e3dfde03493e4033af50c8508d4..014a2b826806ec29fccc0e527e4f83975dbfc8b5 100644 --- a/mace/core/testing/test_benchmark_main.cc +++ b/mace/core/testing/test_benchmark_main.cc @@ -27,7 +27,9 @@ DEFINE_int32(cpu_affinity_policy, 1, "0:AFFINITY_NONE/1:AFFINITY_BIG_ONLY/2:AFFINITY_LITTLE_ONLY"); int main(int argc, char **argv) { - gflags::SetUsageMessage("some usage message"); + std::string usage = "run ops benchmark\nusage: " + std::string(argv[0]) + + " [flags]"; + gflags::SetUsageMessage(usage); gflags::ParseCommandLineFlags(&argc, &argv, true); // config runtime diff --git a/mace/examples/example.cc b/mace/examples/example.cc index c54f3c504af23413488a648a0f08534bdf3f32d4..27b7d78708c4094e92a9caf8b533ba1745c515c5 100644 --- a/mace/examples/example.cc +++ b/mace/examples/example.cc @@ -258,7 +258,9 @@ bool RunModel(const std::vector &input_names, } int Main(int argc, char **argv) { - gflags::SetUsageMessage("some usage message"); + std::string usage = "example run\nusage: " + std::string(argv[0]) + + " [flags]"; + gflags::SetUsageMessage(usage); gflags::ParseCommandLineFlags(&argc, &argv, true); LOG(INFO) << "mace version: " << MaceVersion(); diff --git a/mace/tools/validation/mace_run.cc b/mace/tools/validation/mace_run.cc index 17623ab4093152503b35855cb314d6a498083235..ac524e41c164fa47923e66eda1b9727f1b3d5d98 100644 --- a/mace/tools/validation/mace_run.cc +++ b/mace/tools/validation/mace_run.cc @@ -423,7 +423,9 @@ bool RunModel(const std::string &model_name, } int Main(int argc, char **argv) { - gflags::SetUsageMessage("some usage message"); + std::string usage = "mace run model\nusage: " + std::string(argv[0]) + + " [flags]"; + gflags::SetUsageMessage(usage); gflags::ParseCommandLineFlags(&argc, &argv, true); LOG(INFO) << "model name: " << FLAGS_model_name;