experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[MlirGraphOptimizationPass] fix the indexing issue into the per_pass_state.

Browse Source 
PiperOrigin-RevId: 354125177
Change-Id: Ifcf01771e9c44d330a2abe5059bfec8aa55cadd7
This commit is contained in:
Roman Dzhabarov 2021-01-27 10:52:04 -08:00 committed by TensorFlower Gardener
parent 100e0d8ed0
commit f81438cb02
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc
View File

@ -207,8 +207,7 @@ Status MlirFunctionOptimizationPass::Run(
overall_state == MlirOptimizationPassState::ShadowEnabled)) {
pass_status =
pass_registration.pass->Run(config_proto, *module_ref, **graph);
} else if (per_pass_state[per_pass_state_index] ==
MlirOptimizationPassState::ShadowEnabled) {
} else if (pass_state == MlirOptimizationPassState::ShadowEnabled) {
// Make sure that the pass does not modify MLIR module if it's shadow
// enabled.
auto module_ref_clone = module_ref->clone();