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

Update TPU Bridge passes to run breakup-islands again at the end.

Browse Source 
As breaking up islands/populating islands is in the breakup-islands pass, running this pass at the very end again will allow no empty islands, which may be generated from canonicalizations (e.g. island with only a NoOp). This simplifies some handling of islands during export (e.g. empty islands have special handling in the executor to control dialect pass).

PiperOrigin-RevId: 291048345
Change-Id: If4b4135a3a8d2eea60ae725edc2b976eb9473173
This commit is contained in:
Andy Ly 2020-01-22 15:48:12 -08:00 committed by TensorFlower Gardener
parent 8ffef7b095
commit bce01458eb
1 changed files with 1 additions and 0 deletions

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

@ -60,6 +60,7 @@ void CreateTPUBridge(OpPassManager &pm) {
pm.addNestedPass<FuncOp>(TFDevice::CreateReplicateToIslandPass());
pm.addNestedPass<FuncOp>(CreateBreakUpIslandsPass());
pm.addNestedPass<FuncOp>(createCanonicalizerPass());
pm.addNestedPass<FuncOp>(CreateBreakUpIslandsPass());
}
tensorflow::Status TPUBridge(ModuleOp module, bool enable_logging) {