llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

revert canonicalizer for PrimListConstructOp (#2408)

pull/2414/head
Jiawei Wu 2023-08-22 09:18:39 +08:00 committed by GitHub
parent 3dd29f9d5d
commit 4c9d234b01
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 0 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
include/torch-mlir/Dialect/Torch/IR/TorchOps.td
View File

@ -413,7 +413,6 @@ def Torch_PrimListConstructOp: Torch_Op<"prim.ListConstruct", [
); );
let hasVerifier = 1; let hasVerifier = 1;
let hasCanonicalizer = 1;
let assemblyFormat = [{ let assemblyFormat = [{
$elements attr-dict `:` functional-type(operands, results) $elements attr-dict `:` functional-type(operands, results)

26
lib/Dialect/Torch/IR/TorchOps.cpp
View File

@ -2099,32 +2099,6 @@ void PrimTupleUnpackOp::getCanonicalizationPatterns(RewritePatternSet &patterns,
}); });
} }
//===----------------------------------------------------------------------===//
// PrimListConstructOp
//===----------------------------------------------------------------------===//
void PrimListConstructOp::getCanonicalizationPatterns(
RewritePatternSet &patterns, MLIRContext *context) {
patterns.add(+[](PrimListConstructOp op, PatternRewriter &rewriter) {
if (isListPotentiallyMutated(op.getResult()))
return failure();
SmallVector<Value> elements = llvm::to_vector<4>(op.getElements());
if (elements.size() == 0)
return failure();
auto listUnpackOp = elements[0].getDefiningOp<PrimListUnpackOp>();
if (!listUnpackOp)
return failure();
if (listUnpackOp.getResults() != elements)
return failure();
if (isListPotentiallyMutated(listUnpackOp.getOperand()))
return failure();
rewriter.replaceOp(op, listUnpackOp.getOperand());
return success();
});
}
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
// PrimListUnpackOp // PrimListUnpackOp
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//

9
test/Dialect/Torch/canonicalize.mlir
View File

@ -965,15 +965,6 @@ func.func @torch.prim.If$fold_same_result$subset_of_results(%arg0: !torch.bool,
return %0, %1: !torch.int, !torch.int return %0, %1: !torch.int, !torch.int
} }
// CHECK-LABEL: func.func @prim.ListConstruct$fold_list(
// CHECK-SAME: %[[ARG0:.*]]: !torch.list<tensor>) -> !torch.list<tensor> {
// CHECK: return %[[ARG0]] : !torch.list<tensor>
func.func @prim.ListConstruct$fold_list(%arg0: !torch.list<tensor>) -> !torch.list<tensor> {
%0:2 = torch.prim.ListUnpack %arg0 : !torch.list<tensor> -> !torch.tensor, !torch.tensor
%1 = torch.prim.ListConstruct %0#0, %0#1 : (!torch.tensor, !torch.tensor) -> !torch.list<tensor>
return %1 : !torch.list<tensor>
}
// CHECK-LABEL: func.func @torch.prim.TupleUnpack( // CHECK-LABEL: func.func @torch.prim.TupleUnpack(
// CHECK-SAME: %[[ARG0:.*]]: !torch.tensor, // CHECK-SAME: %[[ARG0:.*]]: !torch.tensor,
// CHECK-SAME: %[[ARG1:.*]]: !torch.tensor) -> !torch.tensor { // CHECK-SAME: %[[ARG1:.*]]: !torch.tensor) -> !torch.tensor {