Dump MLIR only in debug mode (#10422)

可以减少一点点编译时间
Oneflow-Inc · Jan 28, 2024 · 8f055f3 · 8f055f3
1 parent bf9ae02
commit 8f055f3
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/oneflow/ir/oneflow-translate/lib/OneFlow/MLIROneFlowTranslation.cpp b/oneflow/ir/oneflow-translate/lib/OneFlow/MLIROneFlowTranslation.cpp
@@ -876,7 +876,9 @@ LogicalResult ApplyRoundTripPatterns(RoundTripOneFlowJobWrapperInterface& job_wr
   if (shouldPrintGraphviz) {
     job_wrapper.DumpLog("RoundTripOneFlowJob.optimized.mlir.dot", graphviz);
   }
-  DumpMLIR(job_wrapper, module.get(), "RoundTripOneFlowJob.optimized");
+  if (::oneflow::ParseBooleanFromEnv("ONEFLOW_MLIR_DUMPMLIR", false)) {
+    DumpMLIR(job_wrapper, module.get(), "RoundTripOneFlowJob.optimized");
+  }
   return success();
 }
 
@@ -904,7 +906,9 @@ void RoundTripOneFlowJob(
   JobImporter imp(job_wrapper, &context, module.get());
   // TODO: Add flag in job desc to decide whether to run mlir optimizer
   if (succeeded(imp.ProcessJob())) {
-    DumpMLIR(job_wrapper, module.get(), "RoundTripOneFlowJob.imported");
+    if (::oneflow::ParseBooleanFromEnv("ONEFLOW_MLIR_DUMPMLIR", false)) {
+      DumpMLIR(job_wrapper, module.get(), "RoundTripOneFlowJob.imported");
+    }
     if (failed(ApplyRoundTripPatterns(job_wrapper, &context, module))) { exit(EXIT_FAILURE); }
     if (::oneflow::ParseBooleanFromEnv("ONEFLOW_MLIR_STDOUT", false)
         && job_wrapper.IsLastIRPass()) {