From 98debb127d3a14e0239a3432461e3876d293b409 Mon Sep 17 00:00:00 2001 From: Mehdi Amini Date: Sat, 20 Mar 2021 11:00:55 -0700 Subject: [PATCH] Update Tensorflow to prepare for changing the default value of `preloadDialectsInContext` This makes tf-opt more strict on the dialect dependencies, and immediately reduces the number of dialects loaded. The canonicalization of TensorFlow graphs showed on the profile that Linalg pattern were dominated the time, which is unexpected since Linalg is not even intended to be used there. PiperOrigin-RevId: 364087027 --- tools/mlir-hlo-opt/mlir-hlo-opt.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/mlir-hlo-opt/mlir-hlo-opt.cpp b/tools/mlir-hlo-opt/mlir-hlo-opt.cpp index ed96dd5..8c4ccb8 100644 --- a/tools/mlir-hlo-opt/mlir-hlo-opt.cpp +++ b/tools/mlir-hlo-opt/mlir-hlo-opt.cpp @@ -34,6 +34,6 @@ int main(int argc, char **argv) { registry.insert(); registry.insert(); - return failed( - mlir::MlirOptMain(argc, argv, "MLIR HLO pass driver\n", registry)); + return failed(mlir::MlirOptMain(argc, argv, "MLIR HLO pass driver\n", + registry, /*preloadDialectsInContext=*/true)); }