fix run core dump in the end

Signed-off-by: lijian <lijian6@sugon.com>

Src/bin/decoder_main.cc

@@ -19,16 +19,16 @@ DEFINE_bool(continuous_decoding, false, "continuous decoding mode");
 DEFINE_int32(thread_num, 1, "num of decode thread");
 DEFINE_int32(warmup, 0, "num of warmup decode, 0 means no warmup");
 
-std::shared_ptr<wenet::DecodeOptions> g_decode_config;
-std::shared_ptr<wenet::FeaturePipelineConfig> g_feature_config;
-std::shared_ptr<wenet::DecodeResource> g_decode_resource;
+// std::shared_ptr<wenet::DecodeOptions> g_decode_config;
+// std::shared_ptr<wenet::FeaturePipelineConfig> g_feature_config;
+// std::shared_ptr<wenet::DecodeResource> g_decode_resource;
 
 std::ofstream g_result;
 std::mutex g_mutex;
 int g_total_waves_dur = 0;
 int g_total_decode_time = 0;
 
-void Decode(std::pair<std::string, std::string> wav, bool warmup = false) {
+void Decode(std::pair<std::string, std::string> wav, bool warmup, std::shared_ptr<wenet::DecodeOptions> g_decode_config, std::shared_ptr<wenet::FeaturePipelineConfig> g_feature_config, std::shared_ptr<wenet::DecodeResource> g_decode_resource) {
   wenet::WavReader wav_reader(wav.second);
   int num_samples = wav_reader.num_samples();
   CHECK_EQ(wav_reader.sample_rate(), FLAGS_sample_rate);
@@ -120,9 +120,9 @@ int main(int argc, char* argv[]) {
   gflags::ParseCommandLineFlags(&argc, &argv, false);
   google::InitGoogleLogging(argv[0]);
 
-  g_decode_config = wenet::InitDecodeOptionsFromFlags();
-  g_feature_config = wenet::InitFeaturePipelineConfigFromFlags();
-  g_decode_resource = wenet::InitDecodeResourceFromFlags();
+  std::shared_ptr<wenet::DecodeOptions> g_decode_config = wenet::InitDecodeOptionsFromFlags();
+  std::shared_ptr<wenet::FeaturePipelineConfig> g_feature_config = wenet::InitFeaturePipelineConfigFromFlags();
+  std::shared_ptr<wenet::DecodeResource> g_decode_resource = wenet::InitDecodeResourceFromFlags();
 
   if (FLAGS_wav_path.empty() && FLAGS_wav_scp.empty()) {
     LOG(FATAL) << "Please provide the wave path or the wav scp.";
@@ -156,7 +156,7 @@ int main(int argc, char* argv[]) {
       ThreadPool pool(FLAGS_thread_num);
       auto wav = waves[0];
       for (int i = 0; i < FLAGS_warmup; i++) {
-        pool.enqueue(Decode, wav, true);
+        pool.enqueue(Decode, wav, true, g_decode_config, g_feature_config, g_decode_resource);
       }
     }
     LOG(INFO) << "Warmup done.";
@@ -165,7 +165,7 @@ int main(int argc, char* argv[]) {
   {
     ThreadPool pool(FLAGS_thread_num);  
     for (auto& wav : waves) {
-      pool.enqueue(Decode, wav, false);
+      pool.enqueue(Decode, wav, false, g_decode_config, g_feature_config, g_decode_resource);
     }
   }