From 95255302b65470fdeafc9bd6839591a9e3c68bec Mon Sep 17 00:00:00 2001 From: ZHENG Zhen Date: Wed, 2 Aug 2023 11:30:48 +0800 Subject: [PATCH] Enable the experimental mem-intensive operator optimization by default. --- tao_compiler/mlir/disc/disc_util.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tao_compiler/mlir/disc/disc_util.cc b/tao_compiler/mlir/disc/disc_util.cc index 9ca815e146b..7df10714188 100644 --- 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;