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

Fix bug where module passes were nested under function pass manager.

Browse Source 
PiperOrigin-RevId: 276792147
Change-Id: I50de42f33802d5d9ed70a38577475db3dda022fd
This commit is contained in:
River Riddle 2019-10-25 17:33:50 -07:00 committed by TensorFlower Gardener
parent 5aa5839cd7
commit 5f55cc82a9
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc
View File

@ -24,16 +24,16 @@ namespace mlir {
namespace TFTPU { namespace TFTPU {
void createTPUBridge(OpPassManager &pm) { void createTPUBridge(OpPassManager &pm) {
OpPassManager &bridge = pm.nest<FuncOp>(); OpPassManager &func_pm = pm.nest<FuncOp>();
func_pm.addPass(tf_executor::CreateTFExecutorIslandCoarseningPass());
func_pm.addPass(createCanonicalizerPass());
func_pm.addPass(CreateTPUClusterFormationPass());
func_pm.addPass(tf_executor::CreateTFExecutorConstantSinkingPass());
func_pm.addPass(TFDevice::CreateResourceOpLiftingPass());
bridge.addPass(tf_executor::CreateTFExecutorIslandCoarseningPass()); pm.addPass(TFDevice::CreateClusterOutliningPass());
bridge.addPass(createCanonicalizerPass()); pm.addPass(CreateTPURewritePass());
bridge.addPass(CreateTPUClusterFormationPass()); pm.addNestedPass<FuncOp>(createCanonicalizerPass());
bridge.addPass(tf_executor::CreateTFExecutorConstantSinkingPass());
bridge.addPass(TFDevice::CreateResourceOpLiftingPass());
bridge.addPass(TFDevice::CreateClusterOutliningPass());
bridge.addPass(CreateTPURewritePass());
bridge.addPass(createCanonicalizerPass());
} }
tensorflow::Status TPUBridge(ModuleOp module) { tensorflow::Status TPUBridge(ModuleOp module) {