Enable the experimental mem-intensive operator optimization by default.

alibaba · Aug 2, 2023 · 9525530 · 9525530
1 parent 738fa33
commit 9525530
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tao_compiler/mlir/disc/disc_util.cc b/tao_compiler/mlir/disc/disc_util.cc
@@ -260,7 +260,8 @@ bool lowerFakeQuantToQuantAndDequant() {
 
 bool isMemIntensiveOptExperimentalEnabled() {
   static bool enabled = []() {
-    bool enabled = false;
+    // Enable mem-intensive operator optimization by default.
+    bool enabled = true;
     tensorflow::ReadBoolFromEnvVar("DISC_MEM_INTENSIVE_OPT_EXPERIMENTAL",
                                    enabled, &enabled);
     return enabled;