From 3b1c6bc450914c09dca3116eabeb54c9a90dd1f0 Mon Sep 17 00:00:00 2001 From: wuchenghui Date: Thu, 31 May 2018 15:11:15 +0800 Subject: [PATCH] fix usage message --- mace/benchmark/benchmark_model.cc | 4 +++- mace/benchmark/model_throughput_test.cc | 4 +++- mace/core/testing/test_benchmark_main.cc | 4 +++- mace/examples/example.cc | 4 +++- mace/tools/validation/mace_run.cc | 4 +++- 5 files changed, 15 insertions(+), 5 deletions(-) diff --git a/mace/benchmark/benchmark_model.cc b/mace/benchmark/benchmark_model.cc index 7317b8be..b6cd2d69 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 fd19be5c..dce0a736 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 a2dccb65..014a2b82 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 c54f3c50..27b7d787 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 17623ab4..ac524e41 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; -- GitLab