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

Drop OperationFolder usage with Linalg fusion.

Browse Source 
PiperOrigin-RevId: 339653466
Change-Id: Iaf83023f850fdb4ef13a259ab782587aba76eabd
This commit is contained in:
Nicolas Vasilache 2020-10-29 06:09:17 -07:00 committed by TensorFlower Gardener
parent 23c06e9d26
commit 79cdd9533a
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/lhlo_fuse_linalg.cc
View File

@ -26,7 +26,6 @@ limitations under the License.
#include "mlir/Dialect/StandardOps/IR/Ops.h" #include "mlir/Dialect/StandardOps/IR/Ops.h"
#include "mlir/Interfaces/ViewLikeInterface.h" #include "mlir/Interfaces/ViewLikeInterface.h"
#include "mlir/Pass/Pass.h" #include "mlir/Pass/Pass.h"
#include "mlir/Transforms/FoldUtils.h"
#include "mlir/Transforms/GreedyPatternRewriteDriver.h" #include "mlir/Transforms/GreedyPatternRewriteDriver.h"
namespace mlir { namespace mlir {
@ -95,7 +94,6 @@ class LhloFuseLinalgPass
} }
MLIRContext* ctx = func.getContext(); MLIRContext* ctx = func.getContext();
OpBuilder b(func); OpBuilder b(func);
OperationFolder folder(ctx);
func.walk([&](linalg::GenericOp generic_op) { func.walk([&](linalg::GenericOp generic_op) {
SmallVector<int64_t, 2> tile_sizes(tile_sizes_.begin(), SmallVector<int64_t, 2> tile_sizes(tile_sizes_.begin(),
tile_sizes_.end()); tile_sizes_.end());
@ -122,7 +120,7 @@ class LhloFuseLinalgPass
for (unsigned id = 0, e = LinalgOp(op).getNumInputs(); id < e; ++id) { for (unsigned id = 0, e = LinalgOp(op).getNumInputs(); id < e; ++id) {
linalg::Aliases aliases; linalg::Aliases aliases;
linalg::LinalgDependenceGraph graph(aliases, linalg_ops); linalg::LinalgDependenceGraph graph(aliases, linalg_ops);
if (auto info = fuseProducerOfBuffer(b, op, id, graph, &folder)) { if (auto info = fuseProducerOfBuffer(b, op, id, graph)) {
auto originalOp = info->originalProducer.getOperation(); auto originalOp = info->originalProducer.getOperation();
erase_set.insert(originalOp); erase_set.insert(originalOp);
auto originalOpInLinalgOpsVector = std::find_if( auto originalOpInLinalgOpsVector = std::find_if(