fix for wrong simplify_preprocess option

Weiming Zhao · tianboh · commit bf4969f61266 · 2022-03-16T01:09:16.000-07:00
(cherry picked from commit ff18c1f)
diff --git a/include/halo/halo.h b/include/halo/halo.h
@@ -75,6 +75,7 @@ struct CXXCodeGenOpts {
   bool enable_ipu_device = false;
   bool use_ipu_model = false;
   bool separate_constants = false;
+  bool simplify_for_preprocess = false;
   bool disable_broadcasting = true;
   bool enable_type_cast = false;
   bool remove_input_transpose = false;
diff --git a/include/halo/utils/passes_helper.h b/include/halo/utils/passes_helper.h
@@ -148,7 +148,7 @@ static void PopulateOptPasses(PassManager* pm, const std::string& target,
     pm->AddInputRewriterPass(inputs);
   }
   pm->AddInstSimplifyPass(
-      target.substr(0, 3) == "cxx", opts.disable_broadcasting,
+      opts.simplify_for_preprocess, opts.disable_broadcasting,
       opts.remove_input_transpose, opts.remove_output_transpose,
       opts.disable_conv_bn, fusion_opts.ConvBias);
   if (opts.channel_order != ChannelOrder::None) {

Original file line number	Diff line number	Diff line change
`@@ -148,7 +148,7 @@ static void PopulateOptPasses(PassManager* pm, const std::string& target,`
`148`	`148`	`pm->AddInputRewriterPass(inputs);`
`149`	`149`	`}`
`150`	`150`	`pm->AddInstSimplifyPass(`
`151`		`- target.substr(0, 3) == "cxx", opts.disable_broadcasting,`
	`151`	`+ opts.simplify_for_preprocess, opts.disable_broadcasting,`
`152`	`152`	`opts.remove_input_transpose, opts.remove_output_transpose,`
`153`	`153`	`opts.disable_conv_bn, fusion_opts.ConvBias);`
`154`	`154`	`if (opts.channel_order != ChannelOrder::None) {`