colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
This commit is contained in:
Mehdi Amini 2021-03-20 11:00:55 -07:00 committed by TensorFlow MLIR Team
parent 431be0e9b2
commit 98debb127d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/mlir-hlo-opt/mlir-hlo-opt.cpp
View File

@ -34,6 +34,6 @@ int main(int argc, char **argv) {
registry.insert<mlir::lmhlo::LmhloDialect>();
registry.insert<mlir::lmhlo_gpu::LmhloGpuDialect>();
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));
}