From 6fd93375053dfa4c9e4de1ecd37b20f62e987cc5 Mon Sep 17 00:00:00 2001
From: lichun <lichun30@hisilicon.com>
Date: Fri, 25 Jun 2021 13:47:18 +0800
Subject: [PATCH] add atc_params: check_report for ConvertModelToJson

---
 ge/offline/main.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ge/offline/main.cc b/ge/offline/main.cc
index 14db1ded..9ad300fc 100755
--- a/ge/offline/main.cc
+++ b/ge/offline/main.cc
@@ -847,6 +847,7 @@ domi::Status GenerateInfershapeJson() {
   ge::Graph graph;
   std::map<string, string> atc_params;
   atc_params.insert(std::pair<string, string>("input_format", FLAGS_input_format));
+  atc_params.insert(std::pair<string, string>("check_report", FLAGS_check_report));
   ret = ParseGraph(graph, atc_params, FLAGS_om.c_str(), FLAGS_weight.c_str(), (domi::FrameworkType) FLAGS_framework,
                    "", FLAGS_target.c_str(), (ge::RunMode) FLAGS_mode, false);
   if (ret != ge::SUCCESS) {